""" Creates the default Organization object. """ from django.apps import apps from django.core.management.color import no_style from django.db import DEFAULT_DB_ALIAS, connections, router from django.db.models import signals def create_default_organization(app_config, verbosity=2, interactive=True, using=DEFAULT_DB_ALIAS, **kwargs): try: Organization = apps.get_model('organizations', 'Organization') except LookupError: return if not router.allow_migrate(using, Organization): return if not Organization.objects.using(using).exists(): # The default settings set ORGANIZATION_ID = 1, and some tests in Django's test # suite rely on this value. However, if database sequences are reused # (e.g. in the test suite after flush/syncdb), it isn't guaranteed that # the next id will be 1, so we coerce it. See #15573 and #16353. This # can also crop up outside of tests - see #15346. if verbosity >= 2: print("Creating default Organization object") Organization(pk=1, label='Default').save(using=using) # We set an explicit pk instead of relying on auto-incrementation, # so we need to reset the database sequence. See #17415. sequence_sql = connections[using].ops.sequence_reset_sql(no_style(), [Organization]) if sequence_sql: if verbosity >= 2: print('Resetting sequence') with connections[using].cursor() as cursor: for command in sequence_sql: cursor.execute(command) Organization.objects.clear_cache() signals.post_migrate.connect(create_default_organization, sender=apps.get_app_config('organizations'))