Ticket #16039: issue-16039.diff

django/contrib/auth/management/__init__.py

@@ -7 +7 @@
 from django.contrib.auth import models as auth_app
 from django.db.models import get_models, signals
 from django.contrib.auth.models import User
+from django.db import DEFAULT_DB_ALIAS
 
 
 def _get_permission_codename(action, opts):
@@ -21 +22 @@
     return perms + list(opts.permissions)
 
 
-def create_permissions(app, created_models, verbosity, **kwargs):
+def create_permissions(app, created_models, verbosity, db=DEFAULT_DB_ALIAS, **kwargs):
     from django.contrib.contenttypes.models import ContentType
 
     app_models = get_models(app)
@@ -32 +33 @@
     # The codenames and ctypes that should exist.
     ctypes = set()
     for klass in app_models:
-        ctype = ContentType.objects.get_for_model(klass)
+        ctype = ContentType.objects.db_manager(db).get_for_model(klass)
         ctypes.add(ctype)
         for perm in _get_all_permissions(klass._meta):
             searched_perms.append((ctype, perm))
@@ -40 +41 @@
     # Find all the Permissions that have a context_type for a model we're
     # looking for.  We don't need to check for codenames since we already have
     # a list of the ones we're going to create.
-    all_perms = set(auth_app.Permission.objects.filter(
+    all_perms = set(auth_app.Permission.objects.using(db).filter(
         content_type__in=ctypes,
     ).values_list(
         "content_type", "codename"
@@ -51 +52 @@
         for ctype, (codename, name) in searched_perms
         if (ctype.pk, codename) not in all_perms
     ]
-    auth_app.Permission.objects.bulk_create(objs)
+    auth_app.Permission.objects.using(db).bulk_create(objs)
     if verbosity >= 2:
         for obj in objs:
             print "Adding permission '%s'" % obj

django/contrib/contenttypes/management.py

@@ -1 +1 @@
 from django.contrib.contenttypes.models import ContentType
 from django.db.models import get_apps, get_models, signals
 from django.utils.encoding import smart_unicode
+from django.db import DEFAULT_DB_ALIAS
 
-def update_contenttypes(app, created_models, verbosity=2, **kwargs):
+def update_contenttypes(app, created_models, verbosity=2, db=DEFAULT_DB_ALIAS, **kwargs):
     """
     Creates content types for models in the given app, removing any model
     entries that no longer have a matching model class.
@@ -20 +21 @@
     # Get all the content types
     content_types = dict(
         (ct.model, ct)
-        for ct in ContentType.objects.filter(app_label=app_label)
+        for ct in ContentType.objects.using(db).filter(app_label=app_label)
     )
     to_remove = [
         ct
@@ -28 +29 @@
         if model_name not in app_models
     ]
 
-    cts = ContentType.objects.bulk_create([
+    cts = ContentType.objects.using(db).bulk_create([
         ContentType(
             name=smart_unicode(model._meta.verbose_name_raw),
             app_label=app_label,