mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 13:32:15 +00:00
27 lines
1.2 KiB
Python
27 lines
1.2 KiB
Python
from datetime import datetime, timedelta
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils import timezone
|
|
|
|
from RIGS.models import Profile
|
|
from training.models import TrainingLevel
|
|
|
|
|
|
# This is triggered nightly by Heroku Scheduler
|
|
class Command(BaseCommand):
|
|
help = 'Performs perodic user maintenance tasks'
|
|
|
|
def handle(self, *args, **options):
|
|
for person in Profile.objects.all():
|
|
# Inactivate users that have not logged in for a year
|
|
if person.last_login is not None and (timezone.now() - person.last_login).days > 365:
|
|
person.is_active = False
|
|
person.is_approved = False
|
|
person.save()
|
|
# Ensure everyone with a supervisor level has the flag correctly set in the database
|
|
if person.level_qualifications.exclude(confirmed_on=None).select_related('level') \
|
|
.filter(level__level__gte=TrainingLevel.SUPERVISOR) \
|
|
.exclude(level__department=TrainingLevel.HAULAGE) \
|
|
.exclude(level__department__isnull=True).exists():
|
|
person.is_supervisor = True
|
|
person.save()
|