Start of hopper management command
This commit is contained in:
parent
ff49d5cba2
commit
403d1483f6
3 changed files with 33 additions and 0 deletions
0
todo/management/__init__.py
Normal file
0
todo/management/__init__.py
Normal file
0
todo/management/commands/__init__.py
Normal file
0
todo/management/commands/__init__.py
Normal file
33
todo/management/commands/hopper.py
Normal file
33
todo/management/commands/hopper.py
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
from django.core.management.base import BaseCommand
|
||||||
|
from django.contrib.auth.models import Group
|
||||||
|
from todo.models import Task, TaskList
|
||||||
|
from django.contrib.auth import get_user_model
|
||||||
|
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
help = """Create random list and task data for a few fake users."""
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
|
||||||
|
# Create users and groups, add different users to different groups. Staff user is in both groups.
|
||||||
|
bw_group = Group.objects.create(name='Basket Weavers')
|
||||||
|
sd_group = Group.objects.create(name='Scuba Divers')
|
||||||
|
usernames = ['user1', 'user2', 'staff_user']
|
||||||
|
|
||||||
|
for username in usernames:
|
||||||
|
get_user_model().objects.create_user(username=username, password="todo")
|
||||||
|
|
||||||
|
if username == 'user1':
|
||||||
|
u1 = get_user_model().objects.get(username=username)
|
||||||
|
u1.groups.add(bw_group)
|
||||||
|
|
||||||
|
if username == 'user2':
|
||||||
|
u2 = get_user_model().objects.get(username=username)
|
||||||
|
u2.groups.add(sd_group)
|
||||||
|
|
||||||
|
if username == 'staff_user':
|
||||||
|
staffer = get_user_model().objects.get(username=username, is_staff=True,)
|
||||||
|
staffer.groups.add(bw_group)
|
||||||
|
staffer.groups.add(sd_group)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue