From d7ef0fcd3ead51105b1e9b182eb2b9832bf4163e Mon Sep 17 00:00:00 2001 From: pierre geeraert Date: Thu, 2 Jul 2026 22:25:35 +0100 Subject: [PATCH] create an equivalent of router, need to find a data type for it --- .../management/commands/new_urls.py | 5 +- django_code/templates/helloWorld.jinja | 70 ++++++++++++++++++- 2 files changed, 72 insertions(+), 3 deletions(-) diff --git a/django_code/apps/telegram_communication/management/commands/new_urls.py b/django_code/apps/telegram_communication/management/commands/new_urls.py index 2344f68..f7c0232 100644 --- a/django_code/apps/telegram_communication/management/commands/new_urls.py +++ b/django_code/apps/telegram_communication/management/commands/new_urls.py @@ -4,6 +4,7 @@ import os #import jinja2 from jinja2 import Environment, FileSystemLoader +router_path = os.path.join(settings.BASE_DIR,'core','routers_jinja.py') class Command(BaseCommand): args = '' @@ -12,9 +13,9 @@ class Command(BaseCommand): template_dir = os.path.join(settings.BASE_DIR, 'templates') env = Environment(loader=FileSystemLoader(template_dir)) template = env.get_template('helloWorld.jinja') - output = template.render() + output = template.render(name = 'Geeks') - with open("outputFileName.txt", 'w') as f: + with open(router_path, 'w') as f: print(output, file = f) print(output) diff --git a/django_code/templates/helloWorld.jinja b/django_code/templates/helloWorld.jinja index b6fc4c6..230acb4 100644 --- a/django_code/templates/helloWorld.jinja +++ b/django_code/templates/helloWorld.jinja @@ -1 +1,69 @@ -hello \ No newline at end of file +# 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' + 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 == 'redirect': + 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','podcast'] or + obj2._meta.app_label in ['simpsons', 'cloudmanager','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 == 'redirect': + 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','podcast_db']: + return False + + return None \ No newline at end of file