diff --git a/django/contrib/gis/db/backends/spatialite/creation.py b/django/contrib/gis/db/backends/spatialite/creation.py index d299208d79..41baf53e7f 100644 --- a/django/contrib/gis/db/backends/spatialite/creation.py +++ b/django/contrib/gis/db/backends/spatialite/creation.py @@ -33,7 +33,9 @@ class SpatiaLiteCreation(DatabaseCreation): for cache_alias in settings.CACHES: cache = get_cache(cache_alias) if isinstance(cache, BaseDatabaseCache): - call_command('createcachetable', cache._table, database=self.connection.alias) + from django.db import router + if router.allow_syncdb(self.connection.alias, cache.cache_model_class): + call_command('createcachetable', cache._table, database=self.connection.alias) # Get a cursor (even though we don't need one yet). This has # the side effect of initializing the test database. cursor = self.connection.cursor() diff --git a/django/core/management/commands/createcachetable.py b/django/core/management/commands/createcachetable.py index e2f17abdc5..90172b6ca3 100644 --- a/django/core/management/commands/createcachetable.py +++ b/django/core/management/commands/createcachetable.py @@ -1,8 +1,7 @@ from optparse import make_option from django.core.management.base import LabelCommand -from django.core.cache.backends.db import BaseDatabaseCache -from django.db import connections, router, transaction, models, DEFAULT_DB_ALIAS +from django.db import connections, transaction, models, DEFAULT_DB_ALIAS class Command(LabelCommand): help = "Creates the table needed to use the SQL cache backend." @@ -19,11 +18,8 @@ class Command(LabelCommand): requires_model_validation = False def handle_label(self, tablename, **options): - db = options.get('database', DEFAULT_DB_ALIAS) - cache = BaseDatabaseCache(tablename, {}) - if not router.allow_syncdb(db, cache.cache_model_class): - return - connection = connections[db] + alias = options.get('database', DEFAULT_DB_ALIAS) + connection = connections[alias] fields = ( # "key" is a reserved word in MySQL, so use "cache_key" instead. models.CharField(name='cache_key', max_length=255, unique=True, primary_key=True), @@ -54,4 +50,4 @@ class Command(LabelCommand): curs.execute("\n".join(full_statement)) for statement in index_output: curs.execute(statement) - transaction.commit_unless_managed(using=db) + transaction.commit_unless_managed(using=alias) diff --git a/django/db/backends/creation.py b/django/db/backends/creation.py index 6170bf6b97..437a4297af 100644 --- a/django/db/backends/creation.py +++ b/django/db/backends/creation.py @@ -261,7 +261,9 @@ class BaseDatabaseCreation(object): for cache_alias in settings.CACHES: cache = get_cache(cache_alias) if isinstance(cache, BaseDatabaseCache): - call_command('createcachetable', cache._table, database=self.connection.alias) + from django.db import router + if router.allow_syncdb(self.connection.alias, cache.cache_model_class): + call_command('createcachetable', cache._table, database=self.connection.alias) # Get a cursor (even though we don't need one yet). This has # the side effect of initializing the test database. diff --git a/tests/regressiontests/cache/tests.py b/tests/regressiontests/cache/tests.py index 958dcfb6a4..d206b462bc 100644 --- a/tests/regressiontests/cache/tests.py +++ b/tests/regressiontests/cache/tests.py @@ -14,11 +14,10 @@ from django.conf import settings from django.core import management from django.core.cache import get_cache, DEFAULT_CACHE_ALIAS from django.core.cache.backends.base import CacheKeyWarning, InvalidCacheBackendError -from django.db import router from django.http import HttpResponse, HttpRequest, QueryDict from django.middleware.cache import FetchFromCacheMiddleware, UpdateCacheMiddleware, CacheMiddleware -from django.test import RequestFactory, TestCase -from django.test.utils import get_warnings_state, restore_warnings_state, override_settings +from django.test import RequestFactory +from django.test.utils import get_warnings_state, restore_warnings_state from django.utils import translation from django.utils import unittest from django.utils.cache import patch_vary_headers, get_cache_key, learn_cache_key @@ -759,42 +758,6 @@ class DBCacheTests(unittest.TestCase, BaseCacheTests): self.cache = get_cache('db://%s?max_entries=30&cull_frequency=0' % self._table_name) self.perform_cull_test(50, 18) -class DBCacheRouter(object): - """A router that puts the cache table on the 'other' database.""" - - def db_for_read(self, model, **hints): - if model._meta.app_label == 'django_cache': - return 'other' - - def db_for_write(self, model, **hints): - if model._meta.app_label == 'django_cache': - return 'other' - - def allow_syncdb(self, db, model): - if model._meta.app_label == 'django_cache': - return db == 'other' - - -class CreateCacheTableForDBCacheTests(TestCase): - multi_db = True - - def test_createcachetable_observes_database_router(self): - old_routers = router.routers - try: - router.routers = [DBCacheRouter()] - # cache table should not be created on 'default' - with self.assertNumQueries(0): - management.call_command('createcachetable', 'cache_table', - database='default', - verbosity=0, interactive=False) - # cache table should be created on 'other' - with self.assertNumQueries(1): - management.call_command('createcachetable', 'cache_table', - database='other', - verbosity=0, interactive=False) - finally: - router.routers = old_routers - class LocMemCacheTests(unittest.TestCase, BaseCacheTests): backend_name = 'django.core.cache.backends.locmem.LocMemCache'