sf_user/management/commands/clean_guest_data.py

   1 from django.core.management.base import BaseCommand, CommandError
   2 from sf_user.models import CustomSession, CustomUser
   3
   4
   5 class Command(BaseCommand):
   6     help = 'Clean Guest Users Sessions'
   7
   8     #def add_arguments(self, parser):
   9     #    parser.add_argument('poll_id', nargs='+', type=int)
  10
  11     def handle(self, *args, **options):
  12         #for guest in CustomUser.objects.filter(is_guest_user="True"):
  13         #    self.stdout.write(self.style.SUCCESS(guest.username))
  14             try:
  15                 CustomUser.objects.filter(is_guest_user="True").delete()
  16             except Exception:
  17                 raise CommandError('Error unable to clean guest users sessions')
  18
  19             self.stdout.write(self.style.SUCCESS('Successfully cleaned guest users sessions'))