mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-09-14 22:45:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Make migrate command recognise prefixes and 'zero'.

Browse source
This commit is contained in:
Andrew Godwin 2013-07-22 19:43:58 +01:00
parent 52eb19b545
commit 162f7b938f
4 changed files with 95 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

24
django/db/migrations/loader.py
View file

@ -79,6 +79,23 @@ class MigrationLoader(object):
raise BadMigrationError("Migration %s in app %s has no Migration class" % (migration_name, app_label))
self.disk_migrations[app_label, migration_name] = migration_module.Migration(migration_name, app_label)
def get_migration_by_prefix(self, app_label, name_prefix):
"Returns the migration(s) which match the given app label and name _prefix_"
# Make sure we have the disk data
if self.disk_migrations is None:
self.load_disk()
# Do the search
results = []
for l, n in self.disk_migrations:
if l == app_label and n.startswith(name_prefix):
results.append((l, n))
if len(results) > 1:
raise AmbiguityError("There is more than one migration for '%s' with the prefix '%s'" % (app_label, name_prefix))
elif len(results) == 0:
raise KeyError("There no migrations for '%s' with the prefix '%s'" % (app_label, name_prefix))
else:
return self.disk_migrations[results[0]]
@cached_property
def graph(self):
"""
@ -141,3 +158,10 @@ class BadMigrationError(Exception):
Raised when there's a bad migration (unreadable/bad format/etc.)
"""
pass
class AmbiguityError(Exception):
"""
Raised when more than one migration matches a name prefix
"""
pass