summaryrefslogtreecommitdiff
path: root/tests/multiple_database/routers.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/multiple_database/routers.py
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/multiple_database/routers.py')
-rw-r--r--tests/multiple_database/routers.py25
1 files changed, 14 insertions, 11 deletions
diff --git a/tests/multiple_database/routers.py b/tests/multiple_database/routers.py
index 7891e22969..0cc7f1729c 100644
--- a/tests/multiple_database/routers.py
+++ b/tests/multiple_database/routers.py
@@ -9,14 +9,17 @@ class TestRouter:
def db_for_read(self, model, instance=None, **hints):
if instance:
- return instance._state.db or 'other'
- return 'other'
+ return instance._state.db or "other"
+ return "other"
def db_for_write(self, model, **hints):
return DEFAULT_DB_ALIAS
def allow_relation(self, obj1, obj2, **hints):
- return obj1._state.db in ('default', 'other') and obj2._state.db in ('default', 'other')
+ return obj1._state.db in ("default", "other") and obj2._state.db in (
+ "default",
+ "other",
+ )
def allow_migrate(self, db, app_label, **hints):
return True
@@ -29,30 +32,30 @@ class AuthRouter:
def db_for_read(self, model, **hints):
"Point all read operations on auth models to 'default'"
- if model._meta.app_label == 'auth':
+ if model._meta.app_label == "auth":
# We use default here to ensure we can tell the difference
# between a read request and a write request for Auth objects
- return 'default'
+ return "default"
return None
def db_for_write(self, model, **hints):
"Point all operations on auth models to 'other'"
- if model._meta.app_label == 'auth':
- return 'other'
+ if model._meta.app_label == "auth":
+ return "other"
return None
def allow_relation(self, obj1, obj2, **hints):
"Allow any relation if a model in Auth is involved"
- return obj1._meta.app_label == 'auth' or obj2._meta.app_label == 'auth' or None
+ return obj1._meta.app_label == "auth" or obj2._meta.app_label == "auth" or None
def allow_migrate(self, db, app_label, **hints):
"Make sure the auth app only appears on the 'other' db"
- if app_label == 'auth':
- return db == 'other'
+ if app_label == "auth":
+ return db == "other"
return None
class WriteRouter:
# A router that only expresses an opinion on writes
def db_for_write(self, model, **hints):
- return 'writer'
+ return "writer"