41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
|
|
# Common framework for syncdb actions
|
|
|
|
import copy
|
|
|
|
from django.core import management
|
|
from django.conf import settings
|
|
|
|
# Make sure the template loader cache is fixed _now_ (#448)
|
|
import django.template.loaders.app_directories
|
|
|
|
from south.hacks import hacks
|
|
from south.management.commands.syncdb import Command as SyncCommand
|
|
|
|
class MigrateAndSyncCommand(SyncCommand):
|
|
"""Used for situations where "syncdb" is called by test frameworks."""
|
|
|
|
option_list = copy.deepcopy(SyncCommand.option_list)
|
|
|
|
for opt in option_list:
|
|
if "--migrate" == opt.get_opt_string():
|
|
opt.default = True
|
|
break
|
|
|
|
def patch_for_test_db_setup():
|
|
# Load the commands cache
|
|
management.get_commands()
|
|
# Repoint to the correct version of syncdb
|
|
if hasattr(settings, "SOUTH_TESTS_MIGRATE") and not settings.SOUTH_TESTS_MIGRATE:
|
|
# point at the core syncdb command when creating tests
|
|
# tests should always be up to date with the most recent model structure
|
|
management._commands['syncdb'] = 'django.core'
|
|
else:
|
|
management._commands['syncdb'] = MigrateAndSyncCommand()
|
|
# Avoid flushing data migrations.
|
|
# http://code.djangoproject.com/ticket/14661 introduced change that flushed custom
|
|
# sql during the test database creation (thus flushing the data migrations).
|
|
# we patch flush to be no-op during create_test_db, but still allow flushing
|
|
# after each test for non-transactional backends.
|
|
hacks.patch_flush_during_test_db_creation()
|