2021-03-23 17:08:00 -04:00
|
|
|
class PaperworkRouter:
|
|
|
|
app_label = 'paperwork'
|
2022-02-03 13:45:58 -05:00
|
|
|
related_app_labels = {'paperwork', 'membershipworks'}
|
2021-03-23 17:08:00 -04:00
|
|
|
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
|
|
|
|
|
2021-03-23 17:11:11 -04:00
|
|
|
def allow_relation(self, obj1, obj2, **hints):
|
2022-02-03 13:45:58 -05:00
|
|
|
if obj1._meta.app_label in self.related_app_labels or \
|
|
|
|
obj2._meta.app_label in self.related_app_labels:
|
2021-03-23 17:11:11 -04:00
|
|
|
return True
|
|
|
|
return None
|