Updated sample data creation

This commit is contained in:
Harry Bridge
2019-01-08 00:22:46 +00:00
parent f24a510a8e
commit 905eea7792
2 changed files with 8 additions and 10 deletions

View File

@@ -15,7 +15,8 @@ class Command(BaseCommand):
self.create_user_object('staff', True)
self.create_user_object('basic')
def create_user_object(self, name, staff=False, superuser=False):
@staticmethod
def create_user_object(name, staff=False, superuser=False):
user, created = User.objects.get_or_create(
username=name, defaults={'email': '{}@{}.com'.format(name, name),
'first_name': name.title(), 'last_name': 'User', 'is_superuser': superuser,
@@ -23,4 +24,4 @@ class Command(BaseCommand):
if created:
user.set_password(name)
user.save()
user.save()