From d68b4c3e9278a8702bc1afb3c18b1e2e11eb31df Mon Sep 17 00:00:00 2001 From: Andrey Antukh Date: Mon, 1 Apr 2013 12:22:00 +0200 Subject: [PATCH] Now sample_data add members to a created projects. --- greenmine/scrum/management/commands/sample_data.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/greenmine/scrum/management/commands/sample_data.py b/greenmine/scrum/management/commands/sample_data.py index bae8a828..d2ee11f3 100644 --- a/greenmine/scrum/management/commands/sample_data.py +++ b/greenmine/scrum/management/commands/sample_data.py @@ -9,7 +9,7 @@ from django.utils.timezone import now from django.contrib.webdesign import lorem_ipsum -from greenmine.base.models import User +from greenmine.base.models import User, Role from greenmine.scrum.models import * subjects = [ @@ -42,10 +42,12 @@ class Command(BaseCommand): @transaction.commit_on_success def handle(self, *args, **options): - users = [] + users = [User.objects.get(is_superuser=True)] for x in range(10): users.append(self.create_user(x)) + role = Role.objects.all()[0] + # projects for x in xrange(3): # create project @@ -58,6 +60,9 @@ class Command(BaseCommand): project.save() + for user in users: + Membership.objects.create(project=project, role=role, user=user) + now_date = now() - datetime.timedelta(30) # create random milestones