Fix sampledata command to generate tags with and without color

remotes/origin/issue/4795/notification_even_they_are_disabled
David Barragán Merino 2016-06-13 18:02:27 +02:00
parent 13ef1b9af5
commit 93e30ceffa
1 changed files with 13 additions and 2 deletions

View File

@ -19,6 +19,7 @@
import random import random
import datetime import datetime
from os import path from os import path
from hashlib import sha1
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
@ -256,16 +257,21 @@ class Command(BaseCommand):
self.create_wiki_page(project, wiki_link.href) 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 # Set a value to total_story_points to show the deadline in the backlog
project_stats = get_stats_for_project(project) project_stats = get_stats_for_project(project)
defined_points = project_stats["defined_points"] defined_points = project_stats["defined_points"]
project.total_story_points = int(defined_points * self.sd.int(5,12) / 10) project.total_story_points = int(defined_points * self.sd.int(5,12) / 10)
project.refresh_from_db()
project.save() project.save()
self.create_likes(project) self.create_likes(project)
def create_attachment(self, obj, order): def create_attachment(self, obj, order):
attached_file = self.sd.file_from_directory(*ATTACHMENT_SAMPLE_DATA) attached_file = self.sd.file_from_directory(*ATTACHMENT_SAMPLE_DATA)
membership = self.sd.db_object_from_queryset(obj.project.memberships membership = self.sd.db_object_from_queryset(obj.project.memberships
@ -552,3 +558,8 @@ class Command(BaseCommand):
obj.add_watcher(user) obj.add_watcher(user)
else: else:
obj.add_watcher(user, notify_level) obj.add_watcher(user, notify_level)
def generate_color(self, tag):
color = sha1(tag.encode("utf-8")).hexdigest()[0:6]
return "#{}".format(color)