Commit 53ebb713 authored by Flavio TARSETTI's avatar Flavio TARSETTI

[utils/accounts][management][commands] add expired temporary urls cleanup

parent 64987863
Pipeline #39652 canceled with stage
in 1 minute and 28 seconds
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
# encoding: utf-8
###############################################################################
# #
# Copyright (c) 2020 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
import datetime
import sys
from ...models import TemporaryUrl
class Command(BaseCommand):
help = 'Clean expired temporary urls'
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):
if options['interactive']:
try:
answer = self.get_input_data('Clean expired temporary url(s) that have not been used? (y/n)? ', 'y').lower()
except KeyboardInterrupt:
self.stderr.write("\nOperation canceled.")
sys.exit(1)
if answer != 'y':
self.stdout.write('Clean expired temporary urls operation canceled')
sys.exit(1)
temporary_urls = TemporaryUrl.objects.all()
count = 0
for temporary_url in temporary_urls:
now = datetime.datetime.now()
if temporary_url.expires < now:
count+=1
temporary_url.delete()
self.stdout.write('{} Expired temporary url(s) successfully cleaned/'.format(count) + '{} Total temporary url(s) checked'.format(temporary_urls.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
......@@ -46,7 +46,7 @@ class Command(BaseCommand):
help = 'Daily CRON actions'
def handle(self, *args, **options):
# Send attestations cleanup warnings and cleanup attestations
call_command('send_attestation_cleanup_warning')
......@@ -64,3 +64,6 @@ class Command(BaseCommand):
# Yearly revalidation process with warnings, status change and blockage
call_command('year_revalidation_users')
# Clean expired temporary urls for account activation/re-activation
call_command('clean_expired_temporary_urls')
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