diff --git a/taiga/projects/management/commands/sample_data.py b/taiga/projects/management/commands/sample_data.py index 937b11d7..23bbf598 100644 --- a/taiga/projects/management/commands/sample_data.py +++ b/taiga/projects/management/commands/sample_data.py @@ -19,6 +19,7 @@ import random import datetime from os import path +from hashlib import sha1 from django.core.management.base import BaseCommand @@ -256,16 +257,21 @@ class Command(BaseCommand): self.create_wiki_page(project, wiki_link.href) + project.refresh_from_db() + + # Set color for some tags: + for tag in project.tags_colors: + if self.sd.boolean(): + tag[1] = self.generate_color(tag[0]) + # Set a value to total_story_points to show the deadline in the backlog project_stats = get_stats_for_project(project) defined_points = project_stats["defined_points"] project.total_story_points = int(defined_points * self.sd.int(5,12) / 10) - project.refresh_from_db() project.save() self.create_likes(project) - def create_attachment(self, obj, order): attached_file = self.sd.file_from_directory(*ATTACHMENT_SAMPLE_DATA) membership = self.sd.db_object_from_queryset(obj.project.memberships @@ -552,3 +558,8 @@ class Command(BaseCommand): obj.add_watcher(user) else: obj.add_watcher(user, notify_level) + + def generate_color(self, tag): + color = sha1(tag.encode("utf-8")).hexdigest()[0:6] + return "#{}".format(color) +