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

Add invitations and empty projects to the sample_data command

remotes/origin/enhancement/email-actions
David Barragán Merino 2014-07-23 12:40:06 +02:00
parent 75cecbed70
commit b6079a6ff9
1 changed files with 59 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -81,7 +81,9 @@ SUBJECT_CHOICES = [
"Migrate to Python 3 and milk a beautiful cow"]
NUM_USERS = 10
NUM_INVITATIONS = 2
NUM_PROJECTS = 4
NUM_EMPTY_PROJECTS = 2
NUM_MILESTONES = (1, 5)
NUM_USS = (3, 7)
NUM_TASKS_FINISHED = (1, 5)
@ -103,7 +105,7 @@ class Command(BaseCommand):
self.users.append(self.create_user(x))
# create project
for x in range(NUM_PROJECTS):
for x in range(NUM_PROJECTS + NUM_EMPTY_PROJECTS):
project = self.create_project(x)
# added memberships
@ -123,6 +125,20 @@ class Command(BaseCommand):
if role.computable:
computable_project_roles.add(role)
# added invitations
for x in range(NUM_INVITATIONS):
role = self.sd.db_object_from_queryset(Role.objects.all())
Membership.objects.create(email=self.sd.email(),
project=project,
role=role,
is_owner=self.sd.boolean(),
token=''.join(random.sample('abcdef0123456789', 10)))
if role.computable:
computable_project_roles.add(role)
if x < NUM_PROJECTS:
start_date = now() - datetime.timedelta(55)
# create milestones
@ -153,11 +169,6 @@ class Command(BaseCommand):
for y in range(self.sd.int(*NUM_USS_BACK)):
us = self.create_us(project, None, computable_project_roles)
# Set a value to total_story_points to show the deadline in the backlog
defined_points = sum(project.defined_points.values())
project.total_story_points = int(defined_points * self.sd.int(5,12) / 10)
project.save()
# create bugs.
for y in range(self.sd.int(*NUM_ISSUES)):
bug = self.create_bug(project)
@ -165,6 +176,12 @@ class Command(BaseCommand):
# create a wiki page
wiki_page = self.create_wiki(project, "home")
# Set a value to total_story_points to show the deadline in the backlog
defined_points = sum(project.defined_points.values())
project.total_story_points = int(defined_points * self.sd.int(5,12) / 10)
project.save()
def create_attachment(self, object):
attachment = Attachment.objects.create(
project=object.project,
@ -172,7 +189,7 @@ class Command(BaseCommand):
content_object=object,
object_id=object.id,
owner=self.sd.db_object_from_queryset(
object.project.memberships.all()).user,
object.project.memberships.filter(user__isnull=False)).user,
attached_file=self.sd.image_from_directory(*ATTACHMENT_SAMPLE_DATA))
return attachment
@ -182,7 +199,7 @@ class Command(BaseCommand):
project=project,
slug=slug,
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.filter(user__isnull=False)).user)
for i in range(self.sd.int(*NUM_ATTACHMENTS)):
attachment = self.create_attachment(wiki_page)
@ -194,7 +211,7 @@ class Command(BaseCommand):
project=project,
subject=self.sd.choice(SUBJECT_CHOICES),
description=self.sd.paragraph(),
owner=self.sd.db_object_from_queryset(project.memberships.all()).user,
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user,
severity=self.sd.db_object_from_queryset(Severity.objects.filter(
project=project)),
status=self.sd.db_object_from_queryset(IssueStatus.objects.filter(
@ -209,7 +226,7 @@ class Command(BaseCommand):
attachment = self.create_attachment(bug)
if bug.status.order != 1:
bug.assigned_to = self.sd.db_object_from_queryset(project.memberships.all()).user
bug.assigned_to = self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user
bug.save()
return bug
@ -219,11 +236,11 @@ class Command(BaseCommand):
subject=self.sd.choice(SUBJECT_CHOICES),
description=self.sd.paragraph(),
project=project,
owner=self.sd.db_object_from_queryset(project.memberships.all()).user,
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user,
milestone=milestone,
user_story=us,
finished_date=None,
assigned_to = self.sd.db_object_from_queryset(project.memberships.all()).user)
assigned_to = self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user)
if closed:
task.status = project.task_statuses.get(order=4)
@ -244,7 +261,7 @@ class Command(BaseCommand):
us = UserStory.objects.create(
subject=self.sd.choice(SUBJECT_CHOICES),
project=project,
owner=self.sd.db_object_from_queryset(project.memberships.all()).user,
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user,
description=self.sd.paragraph(),
milestone=milestone,
status=self.sd.db_object_from_queryset(project.us_statuses.filter(
@ -266,7 +283,7 @@ class Command(BaseCommand):
attachment = self.create_attachment(us)
if self.sd.choice([True, True, False, True, True]):
us.assigned_to = self.sd.db_object_from_queryset(project.memberships.all()).user
us.assigned_to = self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user
us.save()
return us
@ -277,7 +294,7 @@ class Command(BaseCommand):
name='Sprint {0}-{1}-{2}'.format(start_date.year,
start_date.month,
start_date.day),
owner=self.sd.db_object_from_queryset(project.memberships.all()).user,
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user,
created_date=start_date,
modified_date=start_date,
estimated_start=start_date,
@ -299,13 +316,13 @@ class Command(BaseCommand):
def create_user(self, counter):
user = User.objects.create(
username='user-{0}'.format(counter),
username='user{0}'.format(counter),
full_name="{} {}".format(self.sd.name('es'), self.sd.surname('es', number=1)),
email=self.sd.email(),
token=''.join(random.sample('abcdef0123456789', 10)),
color=self.sd.choice(COLOR_CHOICES))
user.set_password('user{0}'.format(counter))
user.set_password('123123')
user.save()
return user