diff options
Diffstat (limited to 'taskapp/management/commands/seed_db.py')
| -rw-r--r-- | taskapp/management/commands/seed_db.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/taskapp/management/commands/seed_db.py b/taskapp/management/commands/seed_db.py index 6e57338..c4ed25e 100644 --- a/taskapp/management/commands/seed_db.py +++ b/taskapp/management/commands/seed_db.py @@ -7,6 +7,8 @@ from django.contrib.auth.models import User from pytask.taskapp.events import task as taskEvents from pytask.taskapp.events import user as userEvents +from pytask.taskapp.utilities.request import create_request + def seed_db(): """ a method to seed the database with random data """ @@ -15,14 +17,24 @@ def seed_db(): mentor_profile = defaultMentor.get_profile() userEvents.updateProfile(mentor_profile, {'rights':"AD"}) - for i in range(1,10): + for i in range(1,21): username = 'user'+str(i) email = username+'@example.com' password = '123456' dob = datetime.now() gender = "M" - userEvents.createUser(username,email,password,dob,gender) + user = userEvents.createUser(username,email,password,dob,gender) + + if i%4==0: + create_request(defaultMentor, "MG", user) + elif i%3==0: + create_request(defaultMentor, "DV", user) + elif i%2==0: + create_request(defaultMentor, "AD", user) + elif i in ["7", "13"]: + user.is_active = False + user.save() for i in range(1,21): |