21 lines
874 B
Python
21 lines
874 B
Python
from django.conf import settings
|
|
from django.core.management.base import CommandError, BaseCommand
|
|
|
|
from tenant_schemas.management.commands.migrate_schemas import Command as MigrateSchemasCommand
|
|
from tenant_schemas.utils import django_is_in_test_mode
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
def handle(self, *args, **options):
|
|
database = options.get('database', 'default')
|
|
if (settings.DATABASES[database]['ENGINE'] == 'tenant_schemas.postgresql_backend'):
|
|
raise CommandError("migrate has been disabled, for database '{0}'. Use migrate_schemas "
|
|
"instead. Please read the documentation if you don't know why you "
|
|
"shouldn't call migrate directly!".format(database))
|
|
super(Command, self).handle(*args, **options)
|
|
|
|
|
|
if django_is_in_test_mode():
|
|
Command = MigrateSchemasCommand
|