From d16f0c21937e359f91935cf615bab70a9b318a9b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20Barrag=C3=A1n=20Merino?= Date: Tue, 21 Oct 2014 13:53:05 +0200 Subject: [PATCH] Fix some errors in sample_data command --- taiga/projects/management/commands/sample_data.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/taiga/projects/management/commands/sample_data.py b/taiga/projects/management/commands/sample_data.py index bc798a9b..cd64d577 100644 --- a/taiga/projects/management/commands/sample_data.py +++ b/taiga/projects/management/commands/sample_data.py @@ -126,7 +126,7 @@ class Command(BaseCommand): if user == project.owner: continue - role = self.sd.db_object_from_queryset(Role.objects.all()) + role = self.sd.db_object_from_queryset(project.roles.all()) Membership.objects.create(email=user.email, project=project, @@ -138,8 +138,8 @@ class Command(BaseCommand): computable_project_roles.add(role) # added invitations - for x in range(NUM_INVITATIONS): - role = self.sd.db_object_from_queryset(Role.objects.all()) + for i in range(NUM_INVITATIONS): + role = self.sd.db_object_from_queryset(project.roles.all()) Membership.objects.create(email=self.sd.email(), project=project, @@ -286,7 +286,7 @@ class Command(BaseCommand): return task - def create_us(self, project, milestone=None, computable_project_roles=list(Role.objects.all())): + def create_us(self, project, milestone=None, computable_project_roles=[]): us = UserStory.objects.create(subject=self.sd.choice(SUBJECT_CHOICES), project=project, owner=self.sd.db_object_from_queryset(