south / south / management / commands /

Overridden syncdb command

import sys
from optparse import make_option

from import NoArgsCommand, BaseCommand 
from import no_style
from django.utils.datastructures import SortedDict
from import syncdb
from django.conf import settings
from django.db import models
from django.db.models.loading import cache
from django.core import management

from south.db import dbs
from south import migration
from south.exceptions import NoMigrations

def get_app_label(app):
    return '.'.join( app.__name__.split('.')[0:-1] )

class Command(NoArgsCommand):
    option_list = syncdb.Command.option_list + ( 
        make_option('--migrate', action='store_true', dest='migrate', default=False,
            help='Tells South to also perform migrations after the sync. Default for during testing, and other internal calls.'),
        make_option('--all', action='store_true', dest='migrate_all', default=False,
            help='Makes syncdb work on all apps, even migrated ones. Be careful!'),
    if '--verbosity' not in [opt.get_opt_string() for opt in syncdb.Command.option_list]:
        option_list += (
            make_option('--verbosity', action='store', dest='verbosity', default='1',
            type='choice', choices=['0', '1', '2'],
            help='Verbosity level; 0=minimal output, 1=normal output, 2=all output'),
    help = "Create the database tables for all apps in INSTALLED_APPS whose tables haven't already been created, except those which use migrations."

    def handle_noargs(self, migrate_all=False, **options):
        # Work out what uses migrations and so doesn't need syncing
        apps_needing_sync = []
        apps_migrated = []
        for app in models.get_apps():
            app_label = get_app_label(app)
            if migrate_all:
                    migrations = migration.Migrations(app_label)
                except NoMigrations:
                    # It needs syncing
                    # This is a migrated app, leave it
        verbosity = int(options.get('verbosity', 0))
        # Run syncdb on only the ones needed
        if verbosity:
            print "Syncing..."
        old_installed, settings.INSTALLED_APPS = settings.INSTALLED_APPS, apps_needing_sync
        old_app_store, cache.app_store = cache.app_store, SortedDict([
            (k, v) for (k, v) in cache.app_store.items()
            if get_app_label(k) in apps_needing_sync
        # This will allow the setting of the MySQL storage engine, for example.
        for db in dbs.values(): 
        # OK, run the actual syncdb
        settings.INSTALLED_APPS = old_installed
        cache.app_store = old_app_store
        # Migrate if needed
        if options.get('migrate', True):
            if verbosity:
                print "Migrating..."
            management.call_command('migrate', **options)
        # Be obvious about what we did
        if verbosity:
            print "\nSynced:\n > %s" % "\n > ".join(apps_needing_sync)
        if options.get('migrate', True):
            if verbosity:
                print "\nMigrated:\n - %s" % "\n - ".join(apps_migrated)
            if verbosity:
                print "\nNot synced (use migrations):\n - %s" % "\n - ".join(apps_migrated)
                print "(use ./ migrate to migrate these)"

class MigrateAndSyncCommand(Command):
    """Used for situations where "syncdb" is called by test frameworks."""
    option_list = Command.option_list
    for opt in option_list:
        if "--migrate" == opt.get_opt_string():
            opt.default = True

def patch_for_test_db_setup():
        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'
            management._commands['syncdb'] = MigrateAndSyncCommand()