mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2024-09-21 16:33:22 +00:00
30 lines
785 B
Python
30 lines
785 B
Python
|
# coding=utf-8
|
||
|
|
|
||
|
|
|
||
|
|
class DBRouter(object):
|
||
|
|
def db_for_read(self, model, **hints):
|
||
|
|
if model._meta.app_label == 'submission':
|
||
|
|
return 'submission'
|
||
|
|
return "default"
|
||
|
|
|
||
|
|
def db_for_write(self, model, **hints):
|
||
|
|
if model._meta.app_label == 'submission':
|
||
|
|
return 'submission'
|
||
|
|
return "default"
|
||
|
|
|
||
|
|
def allow_relation(self, obj1, obj2, **hints):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def allow_migrate(self, db, app_label, model=None, **hints):
|
||
|
|
if app_label == "submission":
|
||
|
|
if db == "submission":
|
||
|
|
r = True
|
||
|
|
else:
|
||
|
|
r = False
|
||
|
|
else:
|
||
|
|
if db == "default":
|
||
|
|
r = True
|
||
|
|
else:
|
||
|
|
r = False
|
||
|
|
print db, app_label, r
|
||
|
|
return r
|