cmsmanage/paperwork/routers.py

23 lines
654 B
Python
Raw Normal View History

class PaperworkRouter:
2022-02-11 13:48:47 -05:00
app_label = "paperwork"
related_app_labels = {"paperwork", "membershipworks"}
db = "cms"
def db_for_read(self, model, **hints):
if model._meta.app_label == self.app_label:
return self.db
return None
def db_for_write(self, model, **hints):
if model._meta.app_label == self.app_label:
return self.db
return None
def allow_relation(self, obj1, obj2, **hints):
2022-02-11 13:48:47 -05:00
if (
obj1._meta.app_label in self.related_app_labels
or obj2._meta.app_label in self.related_app_labels
):
return True
return None