from textwrap import dedent from django.contrib.auth.hashers import make_password from django.core.management.base import BaseCommand, CommandError from django.db import transaction from django.db.utils import IntegrityError from django.utils.crypto import get_random_string from openslides.agenda.models import Item from openslides.assignments.models import Assignment from openslides.motions.models import Motion from openslides.topics.models import Topic from openslides.users.models import Group, User MOTION_NUMBER_OF_PARAGRAPHS = 4 LOREM_IPSUM = [ """\

Lorem ipsum dolor sit amet, consectetur adipisici elit, sed eiusmod tempor incidunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquid ex ea commodi consequat. Quis aute iure reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint obcaecat cupiditat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.

""", """\

Sed ut perspiciatis, unde omnis iste natus error sit voluptatem accusantium doloremque laudantium, totam rem aperiam eaque ipsa, quae ab illo inventore veritatis et quasi architecto beatae vitae dicta sunt, explicabo. Nemo enim ipsam voluptatem, quia voluptas sit, aspernatur aut odit aut fugit, sed quia consequuntur magni dolores eos, qui ratione voluptatem sequi nesciunt, neque porro quisquam est, qui dolorem ipsum, quia dolor sit amet consectetur adipisci[ng] velit, sed quia non numquam [do] eius modi tempora inci[di]dunt, ut labore et dolore magnam aliquam quaerat voluptatem. Ut enim ad minima veniam, quis nostrum exercitationem ullam corporis suscipit laboriosam, nisi ut aliquid ex ea commodi consequatur? Quis autem vel eum iure reprehenderit, qui in ea voluptate velit esse, quam nihil molestiae consequatur, vel illum, qui dolorem eum fugiat, quo voluptas nulla pariatur?

""", """\

At vero eos et accusamus et iusto odio dignissimos ducimus, qui blanditiis praesentium voluptatum deleniti atque corrupti, quos dolores et quas molestias excepturi sint, obcaecati cupiditate non provident, similique sunt in culpa, qui officia deserunt mollitia animi, id est laborum et dolorum fuga. Et harum quidem rerum facilis est et expedita distinctio. Nam libero tempore, cum soluta nobis est eligendi optio, cumque nihil impedit, quo minus id, quod maxime placeat, facere possimus, omnis voluptas assumenda est, omnis dolor repellendus. Temporibus autem quibusdam et aut officiis debitis aut rerum necessitatibus saepe eveniet, ut et voluptates repudiandae sint et molestiae non recusandae. Itaque earum rerum hic tenetur a sapiente delectus, ut aut reiciendis voluptatibus maiores alias consequatur aut perferendis doloribus asperiores repellat…

""", ] DEFAULT_NUMBER = 100 STAFF_USER_USERNAME = 'admin{}' DEFAULT_USER_USERNAME = 'user{}' PASSWORD = 'password' class Command(BaseCommand): """ Command to create example data for OpenSlides. """ help = 'Create example data for OpenSlides.' chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789' def add_arguments(self, parser): """ Adds arguments to the command parser. The default values for the apps are set by DEFAULT_NUMBER. """ parser.add_argument( '--only', action='store_true', help='Only the given objects are created i. e. all defaults are ' 'set to 0.', ) parser.add_argument( '-t', '--topics', type=int, help='Number of topics to be created (default {}).'.format( DEFAULT_NUMBER), ) parser.add_argument( '-m', '--motions', type=int, help='Number of motions to be created (default {}).'.format( DEFAULT_NUMBER), ) parser.add_argument( '-a', '--assignments', type=int, help='Number of assignments to be created (default {}).'.format( DEFAULT_NUMBER), ) parser.add_argument( '-u', '--users', nargs=2, type=int, help='Number of users to be created. The first number of users is ' 'added to the group "Staff" (default {}). The second number ' 'of users is not added to any group (default {}).'.format( DEFAULT_NUMBER, DEFAULT_NUMBER), ) def handle(self, *args, **options): self.create_topics(options) self.create_motions(options) self.create_assignments(options) self.create_users(options) @transaction.atomic def create_topics(self, options): number_of_topics = options['topics'] if number_of_topics is None and not options['only']: number_of_topics = DEFAULT_NUMBER if number_of_topics is not None and number_of_topics > 0: self.stdout.write('Start creating {} topcis ...'.format(number_of_topics)) current_topics = list(Topic.objects.values_list('id', flat=True)) new_topics = [] for i in range(number_of_topics): new_topics.append(Topic(title=get_random_string(20, self.chars))) Topic.objects.bulk_create(new_topics) items = [] for topic in Topic.objects.exclude(pk__in=current_topics): items.append(Item(content_object=topic, type=Item.AGENDA_ITEM)) Item.objects.bulk_create(items) self.stdout.write(self.style.SUCCESS('{} topcis successfully created.'.format(number_of_topics))) elif number_of_topics is not None and number_of_topics < 0: raise CommandError('Number for topics must not be negative.') @transaction.atomic def create_motions(self, options): number_of_motions = options['motions'] if number_of_motions is None and not options['only']: number_of_motions = DEFAULT_NUMBER if number_of_motions is not None and number_of_motions > 0: self.stdout.write('Start creating {} motions ...'.format(number_of_motions)) text = '' for i in range(MOTION_NUMBER_OF_PARAGRAPHS): text += dedent(LOREM_IPSUM[i % 3]) for i in range(number_of_motions): motion = Motion( title=get_random_string(20, self.chars), text=text, ) motion.save(skip_autoupdate=True) self.stdout.write(self.style.SUCCESS('{} motions successfully created.'.format(number_of_motions))) elif number_of_motions is not None and number_of_motions < 0: raise CommandError('Number for motions must not be negative.') @transaction.atomic def create_assignments(self, options): number_of_assignments = options['assignments'] if number_of_assignments is None and not options['only']: number_of_assignments = DEFAULT_NUMBER if number_of_assignments is not None and number_of_assignments > 0: self.stdout.write('Start creating {} assignments ...'.format(number_of_assignments)) current_assignments = list(Assignment.objects.values_list('id', flat=True)) new_assignments = [] for i in range(number_of_assignments): new_assignments.append(Assignment(title=get_random_string(20, self.chars), open_posts=1)) Assignment.objects.bulk_create(new_assignments) items = [] for assignment in Assignment.objects.exclude(pk__in=current_assignments): items.append(Item(content_object=assignment)) Item.objects.bulk_create(items) self.stdout.write(self.style.SUCCESS('{} assignments successfully created.'.format(number_of_assignments))) elif number_of_assignments is not None and number_of_assignments < 0: raise CommandError('Number for assignments must not be negative.') def create_users(self, options): self.create_staff_users(options) self.create_default_users(options) @transaction.atomic def create_staff_users(self, options): if options['users'] is None and not options['only']: staff_users = DEFAULT_NUMBER elif options['users'] is None: staff_users = None else: staff_users = options['users'][0] if staff_users is not None and staff_users > 0: self.stdout.write('Start creating {} staff users ...'.format(staff_users)) group_staff = Group.objects.get(name='Staff') hashed_password = make_password(PASSWORD, '', 'md5') current_users = list(User.objects.values_list('id', flat=True)) new_users = [] for i in range(staff_users): new_users.append(User( username=STAFF_USER_USERNAME.format(i), default_password=PASSWORD, password=hashed_password )) try: User.objects.bulk_create(new_users) except IntegrityError: self.stdout.write('FAILED: The requested staff users to create are already existing...') else: for user in User.objects.exclude(pk__in=current_users): user.groups.add(group_staff) self.stdout.write(self.style.SUCCESS('{} staff users successfully created.'.format(staff_users))) elif staff_users is not None and staff_users < 0: raise CommandError('Number for staff users must not be negative.') @transaction.atomic def create_default_users(self, options): if options['users'] is None and not options['only']: default_users = DEFAULT_NUMBER elif options['users'] is None: default_users = None else: default_users = options['users'][1] if default_users is not None and default_users > 0: self.stdout.write('Start creating {} default users ...'.format(default_users)) hashed_password = make_password(PASSWORD, '', 'md5') new_users = [] for i in range(default_users): new_users.append(User( username=DEFAULT_USER_USERNAME.format(i), default_password=PASSWORD, password=hashed_password )) try: User.objects.bulk_create(new_users) except IntegrityError: self.stdout.write('FAILED: The requested staff users to create are already existing...') else: self.stdout.write(self.style.SUCCESS('{} default users successfully created.'.format(default_users))) elif default_users is not None and default_users < 0: raise CommandError('Number for default users must not be negative.')