Tooltron

# 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()