Tooltron

from django.db import models

from south.db import DEFAULT_DB_ALIAS

class MigrationHistory(models.Model):

    app_name = models.CharField(max_length=255)

    migration = models.CharField(max_length=255)

    applied = models.DateTimeField(blank=True)

    @classmethod

    def for_migration(cls, migration, database):

        try:

            # Switch on multi-db-ness

            if database != DEFAULT_DB_ALIAS:

                # Django 1.2

                objects = cls.objects.using(database)

            else:

                # Django <= 1.1

                objects = cls.objects

            return objects.get(

                app_name=migration.app_label(),

                migration=migration.name(),

            )

        except cls.DoesNotExist:

            return cls(

                app_name=migration.app_label(),

                migration=migration.name(),

            )

    def get_migrations(self):

        from south.migration.base import Migrations

        return Migrations(self.app_name)

    def get_migration(self):

        return self.get_migrations().migration(self.migration)

    def __unicode__(self):

        return "<%s: %s>" % (self.app_name, self.migration)