From 11028f44cb96fd88313ac4d5ae5926337cd472cf Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Thu, 31 Mar 2016 18:23:36 +0200 Subject: [PATCH] add router --- app/sandbox/routers.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 app/sandbox/routers.py diff --git a/app/sandbox/routers.py b/app/sandbox/routers.py new file mode 100644 index 00000000..7766b6d7 --- /dev/null +++ b/app/sandbox/routers.py @@ -0,0 +1,35 @@ + + +class Router(object): + """ + A router to control all database operations between the 2 apps + """ + def db_for_read(self, model, **hints): + if model._meta.app_label == 'eve': + return 'eve' + if model._meta.app_label == 'presta': + return 'presta' + return None + + def db_for_write(self, model, **hints): + if model._meta.app_label == 'eve': + return 'eve' + if model._meta.app_label == 'presta': + return 'presta' + return None + + def allow_relation(self, obj1, obj2, **hints): + if obj1._meta.app_label == 'eve' or \ + obj2._meta.app_label == 'eve': + return True + if obj1._meta.app_label == 'presta' or \ + obj2._meta.app_label == 'presta': + return True + return None + + def allow_migrate(self, db, app_label, model=None, **hints): + if app_label == 'eve': + return db == 'eve' + if app_label == 'presta': + return db == 'presta' + return None -- 2.39.5