new_url command will automatically gather and replace the router file
Gitea Actions Demo / Explore-Gitea-Actions (push) Successful in 4s
Gitea Actions Demo / Explore-Gitea-Actions (push) Successful in 4s
This commit is contained in:
+40
-19
@@ -8,62 +8,83 @@ class ProjectDatabaseRouter:
|
||||
|
||||
def db_for_read(self, model, **hints):
|
||||
"""Attempts to read simpsons and cloudmanager models go to their specific dbs."""
|
||||
if model._meta.app_label == 'simpsons':
|
||||
return 'simpsons_db'
|
||||
if model._meta.app_label == 'job':
|
||||
return 'job_db'
|
||||
if model._meta.app_label == 'redirect':
|
||||
return 'redirect_db'
|
||||
|
||||
if model._meta.app_label == 'cloudmanager':
|
||||
return 'cloudmanager_db'
|
||||
|
||||
if model._meta.app_label == 'simpsons':
|
||||
return 'simpsons_db'
|
||||
|
||||
if model._meta.app_label == 'redirect':
|
||||
return 'redirect_db'
|
||||
|
||||
if model._meta.app_label == 'job':
|
||||
return 'job_db'
|
||||
|
||||
if model._meta.app_label == 'telegram':
|
||||
return 'telegram_db'
|
||||
|
||||
if model._meta.app_label == 'podcast':
|
||||
return 'podcast_db'
|
||||
|
||||
return 'default'
|
||||
|
||||
def db_for_write(self, model, **hints):
|
||||
"""Attempts to write simpsons and cloudmanager models go to their specific dbs."""
|
||||
if model._meta.app_label == 'simpsons':
|
||||
return 'simpsons_db'
|
||||
if model._meta.app_label == 'job':
|
||||
return 'job_db'
|
||||
|
||||
if model._meta.app_label == 'cloudmanager':
|
||||
return 'cloudmanager_db'
|
||||
|
||||
if model._meta.app_label == 'simpsons':
|
||||
return 'simpsons_db'
|
||||
|
||||
if model._meta.app_label == 'redirect':
|
||||
return 'redirect_db'
|
||||
|
||||
if model._meta.app_label == 'job':
|
||||
return 'job_db'
|
||||
|
||||
if model._meta.app_label == 'telegram':
|
||||
return 'telegram_db'
|
||||
|
||||
if model._meta.app_label == 'podcast':
|
||||
return 'podcast_db'
|
||||
|
||||
return 'default'
|
||||
|
||||
def allow_relation(self, obj1, obj2, **hints):
|
||||
"""Allow relations if both models are in the same database."""
|
||||
# Prevents accidental foreign key relationships across different physical DBs
|
||||
if (
|
||||
obj1._meta.app_label in ['simpsons', 'cloudmanager','redirect','job','telegram','podcast'] or
|
||||
obj2._meta.app_label in ['simpsons', 'cloudmanager','redirect','job','telegram','podcast']
|
||||
if (
|
||||
obj1._meta.app_label in ['cloudmanager', 'simpsons', 'redirect', 'job', 'telegram', 'podcast'] or
|
||||
obj2._meta.app_label in ['cloudmanager', 'simpsons', 'redirect', 'job', 'telegram', 'podcast']
|
||||
):
|
||||
return obj1._meta.app_label == obj2._meta.app_label
|
||||
return None
|
||||
|
||||
def allow_migrate(self, db, app_label, model_name=None, **hints):
|
||||
"""Ensure the app's models only show up in the designated database."""
|
||||
if app_label == 'simpsons':
|
||||
return db == 'simpsons_db'
|
||||
if app_label == 'job':
|
||||
return db == 'job_db'
|
||||
|
||||
if app_label == 'cloudmanager':
|
||||
return db == 'cloudmanager_db'
|
||||
|
||||
if app_label == 'simpsons':
|
||||
return db == 'simpsons_db'
|
||||
|
||||
if app_label == 'redirect':
|
||||
return db == 'redirect_db'
|
||||
|
||||
if app_label == 'job':
|
||||
return db == 'job_db'
|
||||
|
||||
if app_label == 'telegram':
|
||||
return db == 'telegram_db'
|
||||
|
||||
if app_label == 'podcast':
|
||||
return db == 'podcast_db'
|
||||
|
||||
# Ensure default Django apps (auth, sessions, etc.) don't migrate into simpsons/cloudmanager
|
||||
if db in ['simpsons_db', 'cloudmanager_db','redirect_db','job_db','telegram_db','podcast_db']:
|
||||
if db in ['cloudmanager_db', 'simpsons_db', 'redirect_db', 'job_db', 'telegram_db', 'podcast_db']:
|
||||
return False
|
||||
|
||||
return None
|
||||
return None
|
||||
|
||||
Reference in New Issue
Block a user