From eb3c03f2ae2f4cff8cf5146980bc512297492bb6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20Barrag=C3=A1n=20Merino?= Date: Thu, 24 Jul 2014 13:28:19 +0200 Subject: [PATCH] Minor fixes in sample_data command --- .../management/commands/sample_data.py | 23 ++++++++++--------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/taiga/projects/management/commands/sample_data.py b/taiga/projects/management/commands/sample_data.py index f60253f4..307a6c76 100644 --- a/taiga/projects/management/commands/sample_data.py +++ b/taiga/projects/management/commands/sample_data.py @@ -17,6 +17,7 @@ from django.core.management.base import BaseCommand from django.db import transaction from django.utils.timezone import now +from django.conf import settings from django.contrib.webdesign import lorem_ipsum from django.contrib.contenttypes.models import ContentType @@ -80,17 +81,17 @@ SUBJECT_CHOICES = [ "Support for bulk actions", "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) -NUM_TASKS = (0, 4) -NUM_USS_BACK = (8, 20) -NUM_ISSUES = (12, 25) -NUM_ATTACHMENTS = (0, 4) +NUM_USERS = getattr(settings, "SAMPLE_DATA_NUM_USERS", 10) +NUM_INVITATIONS =getattr(settings, "SAMPLE_DATA_NUM_INVITATIONS", 2) +NUM_PROJECTS =getattr(settings, "SAMPLE_DATA_NUM_PROJECTS", 4) +NUM_EMPTY_PROJECTS = getattr(settings, "SAMPLE_DATA_NUM_EMPTY_PROJECTS", 2) +NUM_MILESTONES = getattr(settings, "SAMPLE_DATA_NUM_MILESTONES", (1, 5)) +NUM_USS = getattr(settings, "SAMPLE_DATA_NUM_USS", (3, 7)) +NUM_TASKS_FINISHED = getattr(settings, "SAMPLE_DATA_NUM_TASKS_FINISHED", (1, 5)) +NUM_TASKS = getattr(settings, "SAMPLE_DATA_NUM_TASKS", (0, 4)) +NUM_USS_BACK = getattr(settings, "SAMPLE_DATA_NUM_USS_BACK", (8, 20)) +NUM_ISSUES = getattr(settings, "SAMPLE_DATA_NUM_ISSUES", (12, 25)) +NUM_ATTACHMENTS = getattr(settings, "SAMPLE_DATA_NUM_ATTACHMENTS", (0, 4)) class Command(BaseCommand):