import sys from datetime import datetime from django.core.management.base import NoArgsCommand from django.contrib.auth.models import User from pytask.taskapp.events import task as taskEvents from pytask.taskapp.events import user as userEvents def seed_db(): """ a method to seed the database with random data """ defaultMentor = userEvents.createSuUser("admin", "admin@example.com", "123456", datetime.now(), "M") mentor_profile = defaultMentor.get_profile() userEvents.updateProfile(mentor_profile, {'rights':"AD"}) for i in range(1,10): username = 'user'+str(i) email = username+'@example.com' password = '123456' dob = datetime.now() gender = "M" userEvents.createUser(username,email,password,dob,gender) for i in range(1,6): title = "Task "+str(i) desc = "I am "+title created_by = defaultMentor credits = 20 task = taskEvents.createTask(title,desc,created_by,credits) if task: taskEvents.addMentor(task, defaultMentor) if i%2==0:taskEvents.publishTask(task) class Command(NoArgsCommand): def handle_noargs(self, **options): """ Just copied the code from seed_db.py """ seed_db()