Commit 42051058 authored by Flavio TARSETTI's avatar Flavio TARSETTI

[accounts] patch database and postpone date of users validation

parent 7bdcac62
Pipeline #35937 passed with stage
in 13 minutes and 13 seconds
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
# encoding: utf-8
###############################################################################
# #
# Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/ #
# Contact: beat.support@idiap.ch #
# #
# This file is part of the beat.web module of the BEAT platform. #
# #
# Commercial License Usage #
# Licensees holding valid commercial BEAT licenses may use this file in #
# accordance with the terms contained in a written agreement between you #
# and Idiap. For further information contact tto@idiap.ch #
# #
# Alternatively, this file may be used under the terms of the GNU Affero #
# Public License version 3 as published by the Free Software and appearing #
# in the file LICENSE.AGPL included in the packaging of this file. #
# The BEAT platform is distributed in the hope that it will be useful, but #
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY #
# or FITNESS FOR A PARTICULAR PURPOSE. #
# #
# You should have received a copy of the GNU Affero Public License along #
# with the BEAT platform. If not, see http://www.gnu.org/licenses/. #
# #
###############################################################################
from django.core.management.base import BaseCommand, CommandError
from datetime import date
import datetime
from django.contrib.auth.models import User
from django.conf import settings
from ...models import SupervisionTrack
from ...models import Profile
from ....ui.registration.models import RegistrationProfile
import sys
import random
class Command(BaseCommand):
help = 'Cleanup outdated invalid users'
def add_arguments(self, parser):
parser.add_argument('--noinput', action='store_false', dest='interactive', default=False,
help=('Tells Django to NOT prompt the user for input of any kind.'))
def handle(self, *args, **options):
clean = True
if options['interactive']:
try:
answer = self.get_input_data('Postpone user(s) validation process? (y/n)? ', 'y').lower()
except KeyboardInterrupt:
self.stderr.write("\nOperation canceled.")
sys.exit(1)
if answer != 'y':
self.stdout.write('Postpone canceled')
sys.exit(1)
if clean:
invalid_userprofiles_new_users = Profile.objects.filter(status=Profile.NEWUSER)
invalid_userprofiles_waiting_validation = Profile.objects.filter(status=Profile.WAITINGVALIDATION)
count = 0
postpone_count = 0
now = datetime.datetime.now()
for invalid_profile in invalid_userprofiles_new_users:
user = invalid_profile.user
supervisiontracks = SupervisionTrack.objects.filter(supervision_key=invalid_profile.supervision_key)
registration_profiles = RegistrationProfile.objects.filter(user=invalid_profile.user)
expiration_date = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS)
if user.profile.registration_date + expiration_date <= now:
count += 1
user.delete()
invalid_profile.delete()
for _track in supervisiontracks:
_track.delete()
for _profile in registration_profiles:
_profile.delete()
for invalid_profile in invalid_userprofiles_waiting_validation:
user = invalid_profile.user
supervisiontracks = SupervisionTrack.objects.filter(supervision_key=invalid_profile.supervision_key)
registration_profiles = RegistrationProfile.objects.filter(user=invalid_profile.user)
expiration_date = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS_FROM_SUPERVISOR)
if user.profile.registration_date + expiration_date <= now:
count += 1
user.delete()
invalid_profile.delete()
for _track in supervisiontracks:
_track.delete()
for _profile in registration_profiles:
_profile.delete()
supervisiontracks = SupervisionTrack.objects.all()
for supervisiontrack in supervisiontracks:
expiration_date_delta = datetime.timedelta(days=settings.ACCOUNT_BLOCKAGE_AFTER_FIRST_REJECTION_DAYS)
if supervisiontrack.is_valid and supervisiontrack.expiration_date < now + expiration_date_delta:
postpone_count += 1
extra_days_delta = datetime.timedelta(days=(settings.ACCOUNT_BLOCKAGE_AFTER_FIRST_REJECTION_DAYS + 1))
postpone_date = now + extra_days_delta
supervisiontrack.expiration_date = postpone_date
supervisiontrack.save()
supervisee_profile = Profile.objects.get(user=supervisiontrack.supervisee)
if supervisee_profile.status == Profile.YEARREVALIDATION:
supervisee_profile.status = Profile.ACCEPTED
supervisee_profile.save()
self.stdout.write('{} Invalid user(s) successfully cleaned/'.format(count) + '{} Total user(s) checked'.format(invalid_userprofiles_new_users.count()+invalid_userprofiles_waiting_validation.count()))
self.stdout.write('{} Valid user(s) postponed/'.format(postpone_count) + '{} Total supervision track(s) checked'.format(supervisiontracks.count()))
def get_input_data(self, message, default=None):
"""
Override this method if you want to customize data inputs or
validation exceptions.
"""
raw_value = raw_input(message)
if default and raw_value == '':
raw_value = default
return raw_value
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment