misc: style, PEP8 (#41284)

This commit is contained in:
Benjamin Dauvergne 2020-04-02 22:03:17 +02:00
parent a8bd2006cf
commit f2f80d5f30
1 changed files with 7 additions and 16 deletions

View File

@ -1,5 +1,5 @@
# authentic2 - versatile identity manager
# Copyright (C) 2010-2019 Entr'ouvert
# Copyright (C) 2010-2020 Entr'ouvert
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
@ -16,16 +16,14 @@
from __future__ import print_function
import json
import logging
import smtplib
from datetime import timedelta
from django.contrib.auth import get_user_model
from django.core.management.base import BaseCommand, CommandError
from django.core.management.base import BaseCommand
from django.utils import timezone
from django.utils.six.moves.urllib import parse as urlparse
from django.template.loader import render_to_string
from django_rbac.utils import get_ou_model
from authentic2.models import DeletedUser
@ -38,13 +36,6 @@ logger = logging.getLogger(__name__)
User = get_user_model()
def print_table(table):
col_width = [max(len(x) for x in col) for col in zip(*table)]
for line in table:
line = u"| " + u" | ".join(u"{0:>{1}}".format(x, col_width[i]) for i, x in enumerate(line)) + u" |"
print(line)
class Command(BaseCommand):
help = '''Clean unused accounts'''
@ -72,11 +63,10 @@ class Command(BaseCommand):
for ou in get_ou_model().objects.filter(clean_unused_accounts_alert__isnull=False):
alert_delay = timedelta(days=ou.clean_unused_accounts_alert)
deletion_delay = timedelta(days=ou.clean_unused_accounts_deletion)
users = User.objects.filter(ou=ou, last_login__lte=now-alert_delay)
users = User.objects.filter(ou=ou, last_login__lte=now - alert_delay)
for user in users.filter(last_account_deletion_alert__isnull=True):
logger.info('%s last login %d days ago, sending alert', user,
ou.clean_unused_accounts_alert)
logger.info('%s last login %d days ago, sending alert', user, ou.clean_unused_accounts_alert)
self.send_alert(user)
to_delete = users.filter(
@ -85,8 +75,9 @@ class Command(BaseCommand):
last_account_deletion_alert__lte=now - (deletion_delay - alert_delay)
)
for user in to_delete:
logger.info('%s last login more than %d days ago, deleting user',
user, ou.clean_unused_accounts_deletion)
logger.info(
'%s last login more than %d days ago, deleting user', user,
ou.clean_unused_accounts_deletion)
self.delete_user(user)
def send_alert(self, user):