Python Entrouvert

from django.core.management.base import NoArgsCommand

from django.db import connections, router, models

from south.models import MigrationHistory

from django.core.management.commands import syncdb as django_syncdb

class Command(NoArgsCommand):

    option_list = django_syncdb.Command.option_list

    help = '''list apps with new initial migrations (for "migrate --fake <app> 0001")'''

    def handle_noargs(self, *args, **options):

        # detect installed models

        # (code borrowed from django syncdb command)

        db = options.get('database')

        connection = connections[db]

        cursor = connection.cursor()

        tables = connection.introspection.table_names()

        def model_in_database(model):

            opts = model._meta

            converter = connection.introspection.table_name_converter

            return (converter(opts.db_table) in tables) or \

                (opts.auto_created and converter(opts.auto_created._meta.db_table) in tables)

        # list all applications with south migration

        # (code borrowed from south migrate command)

        from south import migration

        apps = list(migration.all_migrations())

        applied_migrations = MigrationHistory.objects.filter(app_name__in=[app.app_label() for app in apps])

        applied_migrations_lookup = dict(('%s.%s' % (mi.app_name, mi.migration), mi) for mi in applied_migrations)

        apps_with_new_initial_migrations = []

        for app in apps:

            # for each app with migrations, list already applied migrations

            applied_migrations = []

            for migration in app:

                full_name = migration.app_label() + "." + migration.name()

                if full_name in applied_migrations_lookup:

                    applied_migration = applied_migrations_lookup[full_name]

                    if applied_migration.applied:

                        applied_migrations.append(migration.name())

            # try all models in database, if there none, the application is new

            # (because south syncdb does not create any tables)

            new = True

            for m in models.get_models(app.get_application().models):

                if model_in_database(m):

                    new = False

                    break

            if new:

                status = 'application-is-new'

            elif not applied_migrations:

                status = 'migration-is-new'

            else:

                status = 'normal'

            if status == 'migration-is-new':

                # The app was installed without migrations, we suppose

                # it is the first version 0001.

                apps_with_new_initial_migrations.append(app.app_label())

        if apps_with_new_initial_migrations:

            print ' '.join(apps_with_new_initial_migrations)