Set dafault values for attachments in sample data and generate first history entries
David Barragán Merino
parent
b0bc546706
commit
7df2b50d90
|
|
@ -33,6 +33,8 @@ from taiga.projects.issues.models import *
|
|||
from taiga.projects.wiki.models import *
|
||||
from taiga.projects.attachments.models import *
|
||||
|
||||
from taiga.projects.history.services import take_snapshot
|
||||
|
||||
import random
|
||||
import datetime
|
||||
|
||||
|
|
@ -188,36 +190,43 @@ class Command(BaseCommand):
|
|||
project.save()
|
||||
|
||||
|
||||
def create_attachment(self, object):
|
||||
attachment = Attachment.objects.create(
|
||||
project=object.project,
|
||||
def create_attachment(self, object, order):
|
||||
attachment = Attachment.objects.create(project=object.project,
|
||||
content_type=ContentType.objects.get_for_model(object.__class__),
|
||||
content_object=object,
|
||||
order=order,
|
||||
is_deprecated=self.sd.boolean(),
|
||||
description=self.sd.words(3, 12).split(" "),
|
||||
object_id=object.id,
|
||||
owner=self.sd.db_object_from_queryset(
|
||||
object.project.memberships.filter(user__isnull=False)).user,
|
||||
attached_file=self.sd.image_from_directory(*ATTACHMENT_SAMPLE_DATA))
|
||||
attached_file=self.sd.image_from_directory(
|
||||
*ATTACHMENT_SAMPLE_DATA))
|
||||
|
||||
return attachment
|
||||
|
||||
def create_wiki(self, project, slug):
|
||||
wiki_page = WikiPage.objects.create(
|
||||
project=project,
|
||||
wiki_page = WikiPage.objects.create(project=project,
|
||||
slug=slug,
|
||||
content=self.sd.paragraphs(3,15),
|
||||
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).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)
|
||||
attachment = self.create_attachment(wiki_page, i)
|
||||
|
||||
take_snapshot(wiki_page,
|
||||
comment=self.sd.paragraph(),
|
||||
user=wiki_page.owner)
|
||||
|
||||
return wiki_page
|
||||
|
||||
def create_bug(self, project):
|
||||
bug = Issue.objects.create(
|
||||
project=project,
|
||||
bug = Issue.objects.create(project=project,
|
||||
subject=self.sd.choice(SUBJECT_CHOICES),
|
||||
description=self.sd.paragraph(),
|
||||
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).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(
|
||||
|
|
@ -229,24 +238,29 @@ class Command(BaseCommand):
|
|||
tags=self.sd.words(1, 5).split(" "))
|
||||
|
||||
for i in range(self.sd.int(*NUM_ATTACHMENTS)):
|
||||
attachment = self.create_attachment(bug)
|
||||
attachment = self.create_attachment(bug, i)
|
||||
|
||||
if bug.status.order != 1:
|
||||
bug.assigned_to = self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user
|
||||
bug.assigned_to = self.sd.db_object_from_queryset(project.memberships.filter(
|
||||
user__isnull=False)).user
|
||||
bug.save()
|
||||
|
||||
take_snapshot(bug,
|
||||
comment=self.sd.paragraph(),
|
||||
user=bug.owner)
|
||||
|
||||
return bug
|
||||
|
||||
def create_task(self, project, milestone, us, min_date, max_date, closed=False):
|
||||
task = Task(
|
||||
subject=self.sd.choice(SUBJECT_CHOICES),
|
||||
task = Task(subject=self.sd.choice(SUBJECT_CHOICES),
|
||||
description=self.sd.paragraph(),
|
||||
project=project,
|
||||
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.filter(user__isnull=False)).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)
|
||||
|
|
@ -259,21 +273,24 @@ class Command(BaseCommand):
|
|||
task.save()
|
||||
|
||||
for i in range(self.sd.int(*NUM_ATTACHMENTS)):
|
||||
attachment = self.create_attachment(task)
|
||||
attachment = self.create_attachment(task, i)
|
||||
|
||||
take_snapshot(task,
|
||||
comment=self.sd.paragraph(),
|
||||
user=task.owner)
|
||||
|
||||
return task
|
||||
|
||||
def create_us(self, project, milestone=None, computable_project_roles=list(Role.objects.all())):
|
||||
us = UserStory.objects.create(
|
||||
subject=self.sd.choice(SUBJECT_CHOICES),
|
||||
us = UserStory.objects.create(subject=self.sd.choice(SUBJECT_CHOICES),
|
||||
project=project,
|
||||
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).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(
|
||||
is_closed=False)),
|
||||
tags=self.sd.words(1, 3).split(" ")
|
||||
)
|
||||
tags=self.sd.words(1, 3).split(" "))
|
||||
|
||||
for role_points in us.role_points.filter(role__in=computable_project_roles):
|
||||
if milestone:
|
||||
|
|
@ -286,21 +303,25 @@ class Command(BaseCommand):
|
|||
role_points.save()
|
||||
|
||||
for i in range(self.sd.int(*NUM_ATTACHMENTS)):
|
||||
attachment = self.create_attachment(us)
|
||||
attachment = self.create_attachment(us, i)
|
||||
|
||||
if self.sd.choice([True, True, False, True, True]):
|
||||
us.assigned_to = self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).user
|
||||
us.save()
|
||||
|
||||
take_snapshot(us,
|
||||
comment=self.sd.paragraph(),
|
||||
user=us.owner)
|
||||
|
||||
return us
|
||||
|
||||
def create_milestone(self, project, start_date, end_date):
|
||||
milestone = Milestone.objects.create(
|
||||
project=project,
|
||||
milestone = Milestone.objects.create(project=project,
|
||||
name='Sprint {0}-{1}-{2}'.format(start_date.year,
|
||||
start_date.month,
|
||||
start_date.day),
|
||||
owner=self.sd.db_object_from_queryset(project.memberships.filter(user__isnull=False)).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,
|
||||
|
|
@ -310,8 +331,7 @@ class Command(BaseCommand):
|
|||
return milestone
|
||||
|
||||
def create_project(self, counter):
|
||||
project = Project.objects.create(
|
||||
name='Project Example {0}'.format(counter),
|
||||
project = Project.objects.create(name='Project Example {0}'.format(counter),
|
||||
description='Project example {0} description'.format(counter),
|
||||
owner=random.choice(self.users),
|
||||
public=True,
|
||||
|
|
@ -326,8 +346,7 @@ class Command(BaseCommand):
|
|||
full_name = full_name or "{} {}".format(self.sd.name('es'), self.sd.surname('es', number=1))
|
||||
email = email or self.sd.email()
|
||||
|
||||
user = User.objects.create(
|
||||
username=username,
|
||||
user = User.objects.create(username=username,
|
||||
full_name=full_name,
|
||||
email=email,
|
||||
token=''.join(random.sample('abcdef0123456789', 10)),
|
||||
|
|
|
|||
Loading…
Reference in New Issue