taskapp/management/commands/seed_db.py
changeset 37 40651a873f44
child 111 c272d4c601cd
equal deleted inserted replaced
36:0f10deac0a9b 37:40651a873f44
       
     1 import sys
       
     2 from datetime import datetime
       
     3 from django.core.management.base import NoArgsCommand
       
     4 
       
     5 from django.contrib.auth.models import User
       
     6 
       
     7 from pytask.taskapp.events import task as taskEvents
       
     8 from pytask.taskapp.events import user as userEvents
       
     9 
       
    10 
       
    11 def seed_db():
       
    12     """ a method to seed the database with random data """
       
    13     
       
    14     defaultMentor = userEvents.createSuUser("admin", "admin@example.com", "123456", datetime.now(), "M")
       
    15     
       
    16     for i in range(1,10):
       
    17         
       
    18         username = 'user'+str(i)
       
    19         email = username+'@example.com'
       
    20         password = '123456'
       
    21         dob = datetime.now()
       
    22         gender = "M"
       
    23         userEvents.createUser(username,email,password,dob,gender)
       
    24 
       
    25     for i in range(1,21):
       
    26         
       
    27         title = "Task "+str(i)
       
    28         desc = "I am "+title
       
    29         created_by = defaultMentor
       
    30         credits = 20
       
    31         
       
    32         task = taskEvents.createTask(title,desc,created_by,credits)
       
    33         if task:
       
    34             taskEvents.addMentor(task, defaultMentor)
       
    35             taskEvents.publishTask(task)
       
    36 
       
    37 class Command(NoArgsCommand):
       
    38     
       
    39     def handle_noargs(self, **options):
       
    40         """ Just copied the code from seed_db.py """
       
    41         
       
    42         seed_db()