25 lines
793 B
Python
25 lines
793 B
Python
|
"""
|
||
|
|
reinitialize the test user account (and clean up)
|
||
|
|
"""
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
from django.contrib.auth.models import User
|
||
|
|
from django.core.management.base import NoArgsCommand
|
||
|
|
|
||
|
|
|
||
|
|
class Command(NoArgsCommand):
|
||
|
|
help = 'reinitialize the test user'
|
||
|
|
|
||
|
|
def handle_noargs(self, **options):
|
||
|
|
try:
|
||
|
|
u = User.objects.get(username='test')
|
||
|
|
# delete test user and (via CASCADE behaviour) everything that
|
||
|
|
# points to it (has user as ForeignKey), e.g. via created_by.
|
||
|
|
u.delete()
|
||
|
|
except User.DoesNotExist:
|
||
|
|
pass
|
||
|
|
# create a fresh test user
|
||
|
|
u = User.objects.create_user('test', settings.DEFAULT_FROM_EMAIL, 'test')
|
||
|
|
u.save()
|
||
|
|
self.stdout.write('Successfully reinitialized the test user')
|