mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-29 03:12:15 +00:00
Fix sample data generator
This commit is contained in:
@@ -1,27 +0,0 @@
|
|||||||
from django.core.management.base import BaseCommand, CommandError
|
|
||||||
from django.contrib.auth.models import User
|
|
||||||
|
|
||||||
from django.conf import settings
|
|
||||||
|
|
||||||
|
|
||||||
class Command(BaseCommand):
|
|
||||||
help = 'Creates a super user'
|
|
||||||
|
|
||||||
def handle(self, *args, **options):
|
|
||||||
if not (settings.DEBUG or settings.STAGING):
|
|
||||||
raise CommandError('You cannot run this command in production')
|
|
||||||
|
|
||||||
self.create_user_object('super', True, True)
|
|
||||||
self.create_user_object('staff', True)
|
|
||||||
self.create_user_object('basic')
|
|
||||||
|
|
||||||
@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,
|
|
||||||
'is_staff': staff})
|
|
||||||
|
|
||||||
if created:
|
|
||||||
user.set_password(name)
|
|
||||||
user.save()
|
|
||||||
@@ -17,8 +17,6 @@ class Command(BaseCommand):
|
|||||||
|
|
||||||
random.seed('Some object to see the random number generator')
|
random.seed('Some object to see the random number generator')
|
||||||
|
|
||||||
call_command('createBaseUsers')
|
|
||||||
|
|
||||||
self.create_categories()
|
self.create_categories()
|
||||||
self.create_statuses()
|
self.create_statuses()
|
||||||
self.create_suppliers()
|
self.create_suppliers()
|
||||||
|
|||||||
Reference in New Issue
Block a user