don't delete tokens generated by installer during update

This commit is contained in:
wh1te909 2020-08-18 21:54:21 +00:00
parent 8cf26c8ed0
commit 5244064805
1 changed files with 8 additions and 3 deletions

View File

@ -1,10 +1,15 @@
from django.utils import timezone as djangotime
from django.core.management.base import BaseCommand
from knox.models import AuthToken
class Command(BaseCommand):
help = "Deletes all knox tokens"
help = "Deletes all knox web tokens"
def handle(self, *args, **kwargs):
AuthToken.objects.all().delete()
self.stdout.write("All tokens have been deleted!")
# only delete web tokens, not any generated by the installer
dont_delete = djangotime.now() + djangotime.timedelta(hours=23)
tokens = AuthToken.objects.filter(expiry__lt=dont_delete)
tokens.delete()
self.stdout.write("All web tokens have been deleted!")