router.py 658 Bytes
Newer Older
Vasyl Bodnaruk's avatar
Vasyl Bodnaruk committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23

class WebappDatabaseRouter:

    def db_for_read(self, model, **hints):
        if model._meta.app_label == 'webapp':
            return 'esi'
        return None

    def db_for_write(self, model, **hints):
        if model._meta.app_label == 'webapp':
            return 'esi'
        return None

    def allow_relation(self, obj1, obj2, **hints):
        if obj1._meta.app_label == 'webapp' or \
                        obj2._meta.app_label == 'webapp':
            return True
        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        if app_label == 'webapp':
            return db == 'esi'
        return None