author | Nishanth Amuluru <nishanth@fossee.in> |
Sat, 08 Jan 2011 01:52:17 +0530 | |
changeset 65 | 41fcf2273003 |
parent 43 | 20f411241654 |
permissions | -rw-r--r-- |
25 | 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 |
||
41
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
7 |
from pytask.profile.models import Profile, Notification |
43
20f411241654
added key to the notification
Nishanth Amuluru <nishanth@fossee.in>
parents:
41
diff
changeset
|
8 |
from pytask.utils import make_key |
25 | 9 |
|
10 |
def seed_db(): |
|
11 |
""" a method to seed the database with random data """ |
|
12 |
||
13 |
||
41
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
14 |
for i in range(21,1,-1): |
25 | 15 |
|
16 |
username = 'user'+str(i) |
|
17 |
email = username+'@example.com' |
|
18 |
password = '123456' |
|
65
41fcf2273003
Used the full_name in seed_db
Nishanth Amuluru <nishanth@fossee.in>
parents:
43
diff
changeset
|
19 |
full_name = "User "+str(i) |
25 | 20 |
dob = datetime.now() |
21 |
gender = "M" |
|
22 |
aboutme = "I am User"+str(i) |
|
23 |
address = "I live in street"+str(i) |
|
24 |
phonenum = "1234567890" |
|
25 |
||
26 |
new_user = User.objects.create_user(username=username, |
|
27 |
email=email, |
|
28 |
password=password) |
|
29 |
||
30 |
new_profile = Profile() |
|
31 |
new_profile.user = new_user |
|
65
41fcf2273003
Used the full_name in seed_db
Nishanth Amuluru <nishanth@fossee.in>
parents:
43
diff
changeset
|
32 |
new_profile.full_name = full_name |
25 | 33 |
new_profile.dob = dob |
34 |
new_profile.aboutme = aboutme |
|
35 |
new_profile.gender = gender |
|
36 |
new_profile.address = address |
|
37 |
new_profile.phonenum = phonenum |
|
38 |
if i%2 == 0: |
|
39 |
new_profile.rights = "CT" |
|
40 |
elif i%3 == 0: |
|
41 |
new_profile.rights = "CR" |
|
42 |
new_profile.save() |
|
43 |
||
41
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
44 |
new_user.is_superuser = True |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
45 |
new_user.is_staff = True |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
46 |
new_user.save() |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
47 |
|
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
48 |
for i in range(10): |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
49 |
Notification(sent_to=new_user, sent_date=datetime.now(), |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
50 |
subject="A subject here for"+str(i), |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
51 |
message="A message with mess"+str(i)+" html inside.\ |
43
20f411241654
added key to the notification
Nishanth Amuluru <nishanth@fossee.in>
parents:
41
diff
changeset
|
52 |
<br /><b>a bold text</b>", |
20f411241654
added key to the notification
Nishanth Amuluru <nishanth@fossee.in>
parents:
41
diff
changeset
|
53 |
uniq_key=make_key(Notification), |
41
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
54 |
).save() |
d84a3781b979
now seed_db created notifications also
Nishanth Amuluru <nishanth@fossee.in>
parents:
25
diff
changeset
|
55 |
|
25 | 56 |
class Command(NoArgsCommand): |
57 |
||
58 |
def handle_noargs(self, **options): |
|
59 |
""" Just copied the code from seed_db.py """ |
|
60 |
||
61 |
seed_db() |