public/coins-demo
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Require -d to delete old content on hopper run, randomize num tasks per list

Browse source
This commit is contained in:
Scot Hacker 2018-04-05 22:32:31 -07:00
parent 1f27688aa4
commit 97bcfd96c2
1 changed files with 15 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

24
todo/management/commands/hopper.py
View file

@ -12,7 +12,6 @@ from todo.models import Task, TaskList
num_lists = 5 num_lists = 5
num_tasks_per_list = 10
def gen_title(tc=True): def gen_title(tc=True):
@ -39,12 +38,20 @@ def gen_content():
class Command(BaseCommand): class Command(BaseCommand):
help = """Create random list and task data for a few fake users.""" help = """Create random list and task data for a few fake users."""
def add_arguments(self, parser):
parser.add_argument(
"-d",
"--delete",
help="Wipe out existing content before generating new.",
action="store_true")
def handle(self, *args, **options): def handle(self, *args, **options):
# Wipe out previous contents. Cascade deletes the Tasks from the TaskLists. if options.get('delete'):
TaskList.objects.all().delete() # Wipe out previous contents? Cascade deletes the Tasks from the TaskLists.
print("Content from previous run deleted.") TaskList.objects.all().delete()
print("Working...") print("Content from previous run deleted.")
print("Working...")
fake = Faker() # Use to create user's names fake = Faker() # Use to create user's names
@ -83,9 +90,7 @@ class Command(BaseCommand):
TaskListFactory.create_batch(5, group=bw_group) TaskListFactory.create_batch(5, group=bw_group)
TaskListFactory.create_batch(5, group=sd_group) TaskListFactory.create_batch(5, group=sd_group)
print("For each of two groups, created {} fake tasks in each of {} fake lists.".format( print("For each of two groups, created fake tasks in each of {} fake lists.".format(num_lists))
num_lists, num_tasks_per_list)
)
class TaskListFactory(factory.django.DjangoModelFactory): class TaskListFactory(factory.django.DjangoModelFactory):
@ -100,7 +105,8 @@ class TaskListFactory(factory.django.DjangoModelFactory):
@factory.post_generation @factory.post_generation
def add_tasks(self, build, extracted, **kwargs): def add_tasks(self, build, extracted, **kwargs):
TaskFactory.create_batch(num_tasks_per_list, task_list=self) num = random.randint(5, 25)
TaskFactory.create_batch(num, task_list=self)
class TaskFactory(factory.django.DjangoModelFactory): class TaskFactory(factory.django.DjangoModelFactory):