- from datetime import timedelta
-
- from django.contrib.auth.models import User
- from django.core.management.base import BaseCommand
- from django.db.models import Count
- from django.utils import timezone
-
-
- class Command(BaseCommand):
- help = """Prune old, inactive user accounts.
-
- Conditions for removing an user account:
- - created 6 months ago and never logged in. Does not belong
- to any team.
- Use case: visitor types in their email at the website but
- never follows through with login.
-
- - not logged in for 6 months, and has no checks. Does not
- belong to any team.
- Use case: user wants to remove their account. So they
- remove all checks and leave the account at that.
-
- """
-
- def handle(self, *args, **options):
- cutoff = timezone.now() - timedelta(days=180)
-
- # Old accounts, never logged in, no team memberships
- q = User.objects
- q = q.annotate(n_teams=Count("memberships"))
- q = q.filter(date_joined__lt=cutoff, last_login=None, n_teams=0)
- n1, _ = q.delete()
-
- # Not logged in for 1 month, 0 checks, no team memberships
- q = User.objects
- q = q.annotate(n_checks=Count("check"))
- q = q.annotate(n_teams=Count("memberships"))
- q = q.filter(last_login__lt=cutoff, n_checks=0, n_teams=0)
- n2, _ = q.delete()
-
- return "Done! Pruned %d user accounts." % (n1 + n2)
|
