pi-ux-ceandClaude Sonnet 4.6
b4545d6c99
Gitea Actions Demo / Explore-Gitea-Actions (push) Successful in 3s
- Register TelegramCommunicationConfig in INSTALLED_APPS and add telegram_db to DATABASES - Define TelegramData model mapped to the messages table - Fix allow_migrate in ProjectDatabaseRouter: redirect and telegram entries were returning the DB name string (always truthy) instead of a boolean comparison, causing their migrations to bleed into every database - Add telegram routing to db_for_read, db_for_write, allow_relation, and allow_migrate Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
# routers.py
|
|
|
|
class ProjectDatabaseRouter:
|
|
"""
|
|
A router to control all database operations on models in the
|
|
simpsons and cloudmanager applications.
|
|
"""
|
|
|
|
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 == 'telegram':
|
|
return 'telegram_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 == 'redirect':
|
|
return 'redirect_db'
|
|
if model._meta.app_label == 'telegram':
|
|
return 'telegram_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']
|
|
):
|
|
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 == 'redirect':
|
|
return db == 'redirect_db'
|
|
if app_label == 'telegram':
|
|
return db == 'telegram_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']:
|
|
return False
|
|
|
|
return None |