taskapp/management/commands/seed_db.py
author nishanth
Fri, 26 Feb 2010 16:02:32 +0530
changeset 119 39ab7c460143
parent 114 38793914921b
child 121 0c568378a4bc
permissions -rw-r--r--
did lots and now id field in task is a random key
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
37
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     1
import sys
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     2
from datetime import datetime
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     3
from django.core.management.base import NoArgsCommand
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     4
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     5
from django.contrib.auth.models import User
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     6
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     7
from pytask.taskapp.events import task as taskEvents
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     8
from pytask.taskapp.events import user as userEvents
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
     9
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    10
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    11
def seed_db():
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    12
    """ a method to seed the database with random data """
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    13
    
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    14
    defaultMentor = userEvents.createSuUser("admin", "admin@example.com", "123456", datetime.now(), "M")
114
38793914921b mark task as complete functionality is added.
nishanth
parents: 111
diff changeset
    15
    mentor_profile = defaultMentor.get_profile()
38793914921b mark task as complete functionality is added.
nishanth
parents: 111
diff changeset
    16
    userEvents.updateProfile(mentor_profile, {'rights':"AD"})
37
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    17
    
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    18
    for i in range(1,10):
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    19
        
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    20
        username = 'user'+str(i)
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    21
        email = username+'@example.com'
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    22
        password = '123456'
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    23
        dob = datetime.now()
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    24
        gender = "M"
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    25
        userEvents.createUser(username,email,password,dob,gender)
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    26
119
39ab7c460143 did lots and now id field in task is a random key
nishanth
parents: 114
diff changeset
    27
    for i in range(1,6):
37
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    28
        
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    29
        title = "Task "+str(i)
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    30
        desc = "I am "+title
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    31
        created_by = defaultMentor
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    32
        credits = 20
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    33
        
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    34
        task = taskEvents.createTask(title,desc,created_by,credits)
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    35
        if task:
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    36
            taskEvents.addMentor(task, defaultMentor)
111
c272d4c601cd added the functionality to publish a task.
nishanth
parents: 37
diff changeset
    37
            if i%2==0:taskEvents.publishTask(task)
37
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    38
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    39
class Command(NoArgsCommand):
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    40
    
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    41
    def handle_noargs(self, **options):
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    42
        """ Just copied the code from seed_db.py """
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    43
        
40651a873f44 removed seed_db from utils and added it as a command for manage.py and removed url corresponding to seed_db .
nishanth
parents:
diff changeset
    44
        seed_db()