taiga
/
taiga-back
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Sample data command parameterized

remotes/origin/enhancement/email-actions
David Barragán Merino 2014-07-22 17:21:24 +02:00
parent 953e1e9f55
commit add0e458cf
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
taiga/projects/management/commands/sample_data.py
View File

@ -80,6 +80,16 @@ SUBJECT_CHOICES = [
"Support for bulk actions", "Support for bulk actions",
"Migrate to Python 3 and milk a beautiful cow"] "Migrate to Python 3 and milk a beautiful cow"]
NUM_USERS = 10
NUM_PROJECTS = 4
NUM_MILESTONES = ( 1, 5)
NUM_USS = ( 3, 7)
NUM_TASKS_FINISHED = ( 1, 5)
NUM_TASKS = ( 0, 4)
NUM_USS_BACK = ( 8, 20)
NUM_ISSUES = ( 12, 25)
NUM_ATTACHMENTS = ( 0, 4)
class Command(BaseCommand): class Command(BaseCommand):
sd = SampleDataHelper(seed=12345678901) sd = SampleDataHelper(seed=12345678901)
@ -89,11 +99,11 @@ class Command(BaseCommand):
self.users = [User.objects.get(is_superuser=True)] self.users = [User.objects.get(is_superuser=True)]
# create users # create users
for x in range(10): for x in range(NUM_USERS):
self.users.append(self.create_user(x)) self.users.append(self.create_user(x))
# create project # create project
for x in range(4): for x in range(NUM_PROJECTS):
project = self.create_project(x) project = self.create_project(x)
# added memberships # added memberships
@ -116,16 +126,16 @@ class Command(BaseCommand):
start_date = now() - datetime.timedelta(55) start_date = now() - datetime.timedelta(55)
# create milestones # create milestones
for y in range(self.sd.int(1, 5)): for y in range(self.sd.int(*NUM_MILESTONES)):
end_date = start_date + datetime.timedelta(15) end_date = start_date + datetime.timedelta(15)
milestone = self.create_milestone(project, start_date, end_date) milestone = self.create_milestone(project, start_date, end_date)
# create uss asociated to milestones # create uss asociated to milestones
for z in range(self.sd.int(3, 7)): for z in range(self.sd.int(*NUM_USS)):
us = self.create_us(project, milestone, computable_project_roles) us = self.create_us(project, milestone, computable_project_roles)
# create tasks # create tasks
rang = (1, 4) if start_date <= now() and end_date <= now() else (0, 6) rang = NUM_TASKS_FINISHED if start_date <= now() and end_date <= now() else NUM_TASKS
for w in range(self.sd.int(*rang)): for w in range(self.sd.int(*rang)):
if start_date <= now() and end_date <= now(): if start_date <= now() and end_date <= now():
task = self.create_task(project, milestone, us, start_date, task = self.create_task(project, milestone, us, start_date,
@ -140,7 +150,7 @@ class Command(BaseCommand):
start_date = end_date start_date = end_date
# created unassociated uss. # created unassociated uss.
for y in range(self.sd.int(8,20)): for y in range(self.sd.int(*NUM_USS_BACK)):
us = self.create_us(project, None, computable_project_roles) us = self.create_us(project, None, computable_project_roles)
# Set a value to total_story_points to show the deadline in the backlog # Set a value to total_story_points to show the deadline in the backlog
@ -149,7 +159,7 @@ class Command(BaseCommand):
project.save() project.save()
# create bugs. # create bugs.
for y in range(self.sd.int(15,25)): for y in range(self.sd.int(*NUM_ISSUES)):
bug = self.create_bug(project) bug = self.create_bug(project)
# create a wiki page # create a wiki page
@ -174,7 +184,7 @@ class Command(BaseCommand):
content=self.sd.paragraphs(3,15), content=self.sd.paragraphs(3,15),
owner=self.sd.db_object_from_queryset(project.memberships.all()).user) owner=self.sd.db_object_from_queryset(project.memberships.all()).user)
for i in range(self.sd.int(0, 4)): for i in range(self.sd.int(*NUM_ATTACHMENTS)):
attachment = self.create_attachment(wiki_page) attachment = self.create_attachment(wiki_page)
return wiki_page return wiki_page
@ -195,7 +205,7 @@ class Command(BaseCommand):
project=project)), project=project)),
tags=self.sd.words(1, 5).split(" ")) tags=self.sd.words(1, 5).split(" "))
for i in range(self.sd.int(0, 4)): for i in range(self.sd.int(*NUM_ATTACHMENTS)):
attachment = self.create_attachment(bug) attachment = self.create_attachment(bug)
if bug.status.order != 1: if bug.status.order != 1:
@ -225,7 +235,7 @@ class Command(BaseCommand):
task.save() task.save()
for i in range(self.sd.int(0, 4)): for i in range(self.sd.int(*NUM_ATTACHMENTS)):
attachment = self.create_attachment(task) attachment = self.create_attachment(task)
return task return task
@ -252,7 +262,7 @@ class Command(BaseCommand):
role_points.save() role_points.save()
for i in range(self.sd.int(0, 4)): for i in range(self.sd.int(*NUM_ATTACHMENTS)):
attachment = self.create_attachment(us) attachment = self.create_attachment(us)
if self.sd.choice([True, True, False, True, True]): if self.sd.choice([True, True, False, True, True]):