setup new model

This commit is contained in:
pierre geeraert
2026-06-30 23:55:10 +01:00
parent f68b0ad00f
commit a44ca7e825
2 changed files with 18 additions and 3 deletions
+9 -3
View File
@@ -18,6 +18,8 @@ class ProjectDatabaseRouter:
return 'cloudmanager_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):
@@ -32,14 +34,16 @@ class ProjectDatabaseRouter:
return 'redirect_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'] or
obj2._meta.app_label in ['simpsons', 'cloudmanager','redirect','job','telegram']
obj1._meta.app_label in ['simpsons', 'cloudmanager','redirect','job','telegram','podcast'] or
obj2._meta.app_label in ['simpsons', 'cloudmanager','redirect','job','telegram','podcast']
):
return obj1._meta.app_label == obj2._meta.app_label
return None
@@ -56,8 +60,10 @@ class ProjectDatabaseRouter:
return db == 'redirect_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']:
if db in ['simpsons_db', 'cloudmanager_db','redirect_db','job_db','telegram_db','podcast_db']:
return False
return None