Fixed #22474 -- Made migration recorder aware of multiple databases
Thanks Tim Graham for the review.
This commit is contained in:
parent
7fd1b35ed7
commit
7c54f8cced
|
@ -1,5 +1,6 @@
|
||||||
from django.apps.registry import Apps
|
from django.apps.registry import Apps
|
||||||
from django.db import models
|
from django.db import models
|
||||||
|
from django.utils.encoding import python_2_unicode_compatible
|
||||||
from django.utils.timezone import now
|
from django.utils.timezone import now
|
||||||
|
|
||||||
|
|
||||||
|
@ -16,6 +17,7 @@ class MigrationRecorder(object):
|
||||||
a row in the table always means a migration is applied.
|
a row in the table always means a migration is applied.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
@python_2_unicode_compatible
|
||||||
class Migration(models.Model):
|
class Migration(models.Model):
|
||||||
app = models.CharField(max_length=255)
|
app = models.CharField(max_length=255)
|
||||||
name = models.CharField(max_length=255)
|
name = models.CharField(max_length=255)
|
||||||
|
@ -26,9 +28,16 @@ class MigrationRecorder(object):
|
||||||
app_label = "migrations"
|
app_label = "migrations"
|
||||||
db_table = "django_migrations"
|
db_table = "django_migrations"
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return "Migration %s for %s" % (self.name, self.app)
|
||||||
|
|
||||||
def __init__(self, connection):
|
def __init__(self, connection):
|
||||||
self.connection = connection
|
self.connection = connection
|
||||||
|
|
||||||
|
@property
|
||||||
|
def migration_qs(self):
|
||||||
|
return self.Migration.objects.using(self.connection.alias)
|
||||||
|
|
||||||
def ensure_schema(self):
|
def ensure_schema(self):
|
||||||
"""
|
"""
|
||||||
Ensures the table exists and has the correct schema.
|
Ensures the table exists and has the correct schema.
|
||||||
|
@ -46,25 +55,24 @@ class MigrationRecorder(object):
|
||||||
Returns a set of (app, name) of applied migrations.
|
Returns a set of (app, name) of applied migrations.
|
||||||
"""
|
"""
|
||||||
self.ensure_schema()
|
self.ensure_schema()
|
||||||
return set(tuple(x) for x in self.Migration.objects.values_list("app", "name"))
|
return set(tuple(x) for x in self.migration_qs.values_list("app", "name"))
|
||||||
|
|
||||||
def record_applied(self, app, name):
|
def record_applied(self, app, name):
|
||||||
"""
|
"""
|
||||||
Records that a migration was applied.
|
Records that a migration was applied.
|
||||||
"""
|
"""
|
||||||
self.ensure_schema()
|
self.ensure_schema()
|
||||||
self.Migration.objects.create(app=app, name=name)
|
self.migration_qs.create(app=app, name=name)
|
||||||
|
|
||||||
def record_unapplied(self, app, name):
|
def record_unapplied(self, app, name):
|
||||||
"""
|
"""
|
||||||
Records that a migration was unapplied.
|
Records that a migration was unapplied.
|
||||||
"""
|
"""
|
||||||
self.ensure_schema()
|
self.ensure_schema()
|
||||||
self.Migration.objects.filter(app=app, name=name).delete()
|
self.migration_qs.filter(app=app, name=name).delete()
|
||||||
|
|
||||||
@classmethod
|
def flush(self):
|
||||||
def flush(cls):
|
|
||||||
"""
|
"""
|
||||||
Deletes all migration records. Useful if you're testing migrations.
|
Deletes all migration records. Useful if you're testing migrations.
|
||||||
"""
|
"""
|
||||||
cls.Migration.objects.all().delete()
|
self.migration_qs.all().delete()
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
from unittest import skipIf
|
from unittest import skipIf
|
||||||
|
|
||||||
from django.test import TestCase, override_settings
|
from django.test import TestCase, override_settings
|
||||||
from django.db import connection
|
from django.db import connection, connections
|
||||||
from django.db.migrations.loader import MigrationLoader, AmbiguityError
|
from django.db.migrations.loader import MigrationLoader, AmbiguityError
|
||||||
from django.db.migrations.recorder import MigrationRecorder
|
from django.db.migrations.recorder import MigrationRecorder
|
||||||
from django.utils import six
|
from django.utils import six
|
||||||
|
@ -26,6 +26,12 @@ class RecorderTests(TestCase):
|
||||||
recorder.applied_migrations(),
|
recorder.applied_migrations(),
|
||||||
set([("myapp", "0432_ponies")]),
|
set([("myapp", "0432_ponies")]),
|
||||||
)
|
)
|
||||||
|
# That should not affect records of another database
|
||||||
|
recorder_other = MigrationRecorder(connections['other'])
|
||||||
|
self.assertEqual(
|
||||||
|
recorder_other.applied_migrations(),
|
||||||
|
set(),
|
||||||
|
)
|
||||||
recorder.record_unapplied("myapp", "0432_ponies")
|
recorder.record_unapplied("myapp", "0432_ponies")
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
recorder.applied_migrations(),
|
recorder.applied_migrations(),
|
||||||
|
|
Loading…
Reference in New Issue