新增資料庫路由檔案
要在 Django 中使用多個資料庫,只需在 settings.py
中指定每個資料庫:
DATABASES = {
'default': {
'NAME': 'app_data',
'ENGINE': 'django.db.backends.postgresql',
'USER': 'django_db_user',
'PASSWORD': os.environ['LOCAL_DB_PASSWORD']
},
'users': {
'NAME': 'remote_data',
'ENGINE': 'django.db.backends.mysql',
'HOST': 'remote.host.db',
'USER': 'remote_user',
'PASSWORD': os.environ['REMOTE_DB_PASSWORD']
}
}
使用 dbrouters.py
檔案指定哪些模型應該針對每類資料庫操作操作哪些資料庫,例如對於儲存在 remote_data
中的遠端資料,你可能需要以下內容:
class DbRouter(object):
"""
A router to control all database operations on models in the
auth application.
"""
def db_for_read(self, model, **hints):
"""
Attempts to read remote models go to remote database.
"""
if model._meta.app_label == 'remote':
return 'remote_data'
return 'app_data'
def db_for_write(self, model, **hints):
"""
Attempts to write remote models go to the remote database.
"""
if model._meta.app_label == 'remote':
return 'remote_data'
return 'app_data'
def allow_relation(self, obj1, obj2, **hints):
"""
Do not allow relations involving the remote database
"""
if obj1._meta.app_label == 'remote' or \
obj2._meta.app_label == 'remote':
return False
return None
def allow_migrate(self, db, app_label, model_name=None, **hints):
"""
Do not allow migrations on the remote database
"""
if model._meta.app_label == 'remote':
return False
return True
最後,將你的 dbrouter.py
新增到 settings.py
:
DATABASE_ROUTERS = ['path.to.DbRouter', ]