ag-admin/OpenSlides
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed the example data generator

Browse Source
This commit is contained in:
FinnStutzenstein 2019-08-08 12:47:47 +02:00
parent f25a8aefb2
commit a8946cfe99
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/example_data_generator/management/commands/create-example-data.py
View File

@ -7,7 +7,7 @@ from django.db import transaction
from django.db.utils import IntegrityError from django.db.utils import IntegrityError
from django.utils.crypto import get_random_string from django.utils.crypto import get_random_string
from openslides.agenda.models import Item from openslides.agenda.models import Item, ListOfSpeakers
from openslides.assignments.models import Assignment from openslides.assignments.models import Assignment
from openslides.core.apps import startup from openslides.core.apps import startup
from openslides.motions.models import Motion from openslides.motions.models import Motion
@ -137,9 +137,12 @@ class Command(BaseCommand):
new_topics.append(Topic(title=get_random_string(20, self.chars))) new_topics.append(Topic(title=get_random_string(20, self.chars)))
Topic.objects.bulk_create(new_topics) Topic.objects.bulk_create(new_topics)
items = [] items = []
lists_of_speakers = []
for topic in Topic.objects.exclude(pk__in=current_topics): for topic in Topic.objects.exclude(pk__in=current_topics):
items.append(Item(content_object=topic, type=Item.AGENDA_ITEM)) items.append(Item(content_object=topic, type=Item.AGENDA_ITEM))
lists_of_speakers.append(ListOfSpeakers(content_object=topic))
Item.objects.bulk_create(items) Item.objects.bulk_create(items)
ListOfSpeakers.objects.bulk_create(lists_of_speakers)
self.stdout.write( self.stdout.write(
self.style.SUCCESS(f"{number_of_topics} topcis successfully created.") self.style.SUCCESS(f"{number_of_topics} topcis successfully created.")
) )
@ -180,9 +183,12 @@ class Command(BaseCommand):
) )
Assignment.objects.bulk_create(new_assignments) Assignment.objects.bulk_create(new_assignments)
items = [] items = []
lists_of_speakers = []
for assignment in Assignment.objects.exclude(pk__in=current_assignments): for assignment in Assignment.objects.exclude(pk__in=current_assignments):
items.append(Item(content_object=assignment)) items.append(Item(content_object=assignment))
lists_of_speakers.append(ListOfSpeakers(content_object=assignment))
Item.objects.bulk_create(items) Item.objects.bulk_create(items)
ListOfSpeakers.objects.bulk_create(lists_of_speakers)
self.stdout.write( self.stdout.write(
self.style.SUCCESS( self.style.SUCCESS(
f"{number_of_assignments} assignments successfully created." f"{number_of_assignments} assignments successfully created."