taskapp/management/commands/seed_db.py
changeset 218 59107ce0a618
parent 183 c088c79a225c
child 219 f04a1ec7a07f
equal deleted inserted replaced
217:307f699e6102 218:59107ce0a618
    12 
    12 
    13 
    13 
    14 def seed_db():
    14 def seed_db():
    15     """ a method to seed the database with random data """
    15     """ a method to seed the database with random data """
    16     
    16     
    17     defaultMentor = userEvents.createSuUser("admin", "admin@example.com", "123456", datetime.now(), "M")
    17     defaultReviewer = userEvents.createSuUser("admin", "admin@example.com", "123456", datetime.now(), "M")
    18     mentor_profile = defaultMentor.get_profile()
    18     reviewer_profile = defaultReviewer.get_profile()
    19     userEvents.updateProfile(mentor_profile, {'rights':"AD"})
    19     userEvents.updateProfile(reviewer_profile, {'rights':"AD"})
    20     
    20     
    21     for i in range(1,21):
    21     for i in range(1,21):
    22         
    22         
    23         username = 'user'+str(i)
    23         username = 'user'+str(i)
    24         email = username+'@example.com'
    24         email = username+'@example.com'
    27         gender = "M"
    27         gender = "M"
    28         user = userEvents.createUser(username,email,password,dob,gender)
    28         user = userEvents.createUser(username,email,password,dob,gender)
    29         create_notification("NU", user)
    29         create_notification("NU", user)
    30 
    30 
    31         if i%4==0:
    31         if i%4==0:
    32             create_request(defaultMentor, "MG", user)
    32             create_request(defaultReviewer, "MG", user)
    33         elif i%3==0:
    33         elif i%3==0:
    34             create_request(defaultMentor, "DV", user)
    34             create_request(defaultReviewer, "DV", user)
    35         elif i%2==0:
    35         elif i%2==0:
    36             create_request(defaultMentor, "AD", user)
    36             create_request(defaultReviewer, "AD", user)
    37         elif i in [7, 13]:
    37         elif i in [7, 13]:
    38             user.is_active = False
    38             user.is_active = False
    39             user.save()
    39             user.save()
    40 
    40 
    41     for i in range(1,21):
    41     for i in range(1,21):
    42         
    42         
    43         title = "Task "+str(i)
    43         title = "Task "+str(i)
    44         desc = "I am "+title
    44         desc = "I am "+title
    45         created_by = defaultMentor
    45         created_by = defaultReviewer
    46         credits = 20
    46         credits = 20
    47         
    47         
    48         task = taskEvents.createTask(title,desc,created_by,credits)
    48         task = taskEvents.createTask(title,desc,created_by,credits)
    49         if task:
    49         if task:
    50             taskEvents.addMentor(task, defaultMentor)
    50             taskEvents.addReviewer(task, defaultReviewer)
    51             if i%2==0:taskEvents.publishTask(task)
    51             if i%2==0:taskEvents.publishTask(task)
    52 
    52 
    53 class Command(NoArgsCommand):
    53 class Command(NoArgsCommand):
    54     
    54     
    55     def handle_noargs(self, **options):
    55     def handle_noargs(self, **options):