taskapp/management/commands/seed_db.py
author nishanth
Fri, 26 Feb 2010 02:52:32 +0530
changeset 111 c272d4c601cd
parent 37 40651a873f44
child 114 38793914921b
permissions -rw-r--r--
added the functionality to publish a task.
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")
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
    15
    
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
    16
    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
    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
        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
    19
        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
    20
        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
    21
        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
    22
        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
    23
        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
    24
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
    for i in range(1,21):
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
        
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
    27
        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
    28
        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
    29
        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
    30
        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
    31
        
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
        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
    33
        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
    34
            taskEvents.addMentor(task, defaultMentor)
111
c272d4c601cd added the functionality to publish a task.
nishanth
parents: 37
diff changeset
    35
            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
    36
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
    37
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
    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
    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
    40
        """ 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
    41
        
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
        seed_db()