【Django】django 多資料庫配置
在django專案中, 一個工程中存在多個APP應用很常見. 有時候希望不同的APP連線不同的資料庫，這個時候需要建立多個數據庫連線。
1. 修改專案的 settings 配置
在 settings.py 中配置需要連線的多個數據庫連線串
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
DATABASES = {
'default': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': os.path.join(BASE_DIR, 'sqlite3'),
},
'db01': { 'ENGINE': 'django.db.backends.sqlite3',
'NAME': os.path.join(BASE_DIR, 'db_01'),
},
'db02': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': os.path.join(BASE_DIR, 'db_02'),
},
}
|
假設現在我們用到3個數據庫，一個default預設庫，一個 db01 和 db02
2. 設定資料庫的路由規則方法
在 settings.py 中配置 DATABASE_ROUTERS
| 1 |
DATABASE_ROUTERS = ['Prject.database_router.DatabaseAppsRouter']
|
Project: 建立的django專案名稱(project_name)
database_router: 定義路由規則database_router.py 檔名稱, 這個檔名可以自己定義
DatabaseAppsRouter: 路由規則的類名稱，這個類是在database_router.py 檔案中定義
3. 設定APP對應的資料庫路由表
每個APP要連線哪個資料庫，需要在做匹配設定，在 settings.py 檔案中做如下配置：
| 1 2 3 4 5 6 7 8 9 10 |
DATABASE_APPS_MAPPING = {
# example:
# 'app_name':'database_name',
'app02': 'db02',
'app01': 'db01',
'admin': 'db01',
'auth': 'db01',
'contenttypes': 'db01',
'sessions': 'db01',
}
|
以上的app01， app02是專案中的 APP名，分別指定到 db01， db02 的資料庫。
為了使django自己的表也建立到你自己定義的資料庫中，你可以指定 : admin, auth, contenttypes, sessions 到設定的資料庫中，如果不指定則會自動建立到預設（default）的資料庫中
4. 建立資料庫路由規則
在專案工程根路徑下(與 settings.py 檔案一級）建立 database_router.py 檔案:
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from django.conf import settings
DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING
class DatabaseAppsRouter(object):
"""
A router to control all database operations on models for different
databases.
In case an app is not set in settings.DATABASE_APPS_MAPPING, the router
will fallback to the `default` database.
Settings example:
DATABASE_APPS_MAPPING = {'app1': 'db1', 'app2': 'db2'}
"""
def db_for_read(self, model, **hints):
""""Point all read operations to the specific database."""
if model._meta.app_label in DATABASE_MAPPING:
return DATABASE_MAPPING[model._meta.app_label]
return None
def db_for_write(self, model, **hints):
"""Point all write operations to the specific database."""
if model._meta.app_label in DATABASE_MAPPING:
return DATABASE_MAPPING[model._meta.app_label]
return None
def allow_relation(self, obj1, obj2, **hints):
"""Allow any relation between apps that use the same database."""
db_obj1 = DATABASE_MAPPING.get(obj1._meta.app_label)
db_obj2 = DATABASE_MAPPING.get(obj2._meta.app_label)
if db_obj1 and db_obj2:
if db_obj1 == db_obj2:
return True
else:
return False
return None
def allow_syncdb(self, db, model):
"""Make sure that apps only appear in the related database."""
if db in DATABASE_MAPPING.values():
return DATABASE_MAPPING.get(model._meta.app_label) == db
elif model._meta.app_label in DATABASE_MAPPING:
return False
return None
def allow_migrate(self, db, app_label, model=None, **hints):
"""
Make sure the auth app only appears in the 'auth_db'
database.
"""
if db in DATABASE_MAPPING.values():
return DATABASE_MAPPING.get(app_label) == db
elif app_label in DATABASE_MAPPING:
return False
return None
|
5. Models建立樣例
在各自的 APP 中建立資料表的models時，必須要指定表的 app_label 名字，如果不指定則會建立到 default 中配置的資料庫名下，
如下：
在app01下建立models
| 1 2 3 4 5 6 7 |
class Users(models.Model):
name = models.CharField(max_length=50)
passwd = models.CharField(max_length=100)
def __str__(self):
return "app01 %s " % self.name
class Meta:
app_label = "app01"
|
在app02下建立models
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
class Users(models.Model):
username = models.CharField(max_length=100)
password = models.CharField(max_length=50)
age = models.IntegerField()
def __str__(self):
return "app02 %s" % self.username
class Meta:
app_label = "app02"
class Book(models.Model):
user = models.ForeignKey("Users", on_delete=models.CASCADE)
bookname = models.CharField(max_length=100)
def __str__(self):
return "%s: %s" % (self.user.username, self.bookname)
class Meta:
app_label = "app02"
|
在 app03建立models 未指定 app_label，建立到default下
| 1 2 |
class Users(models.Model):
username = models.CharField(max_length=100)
|
6. 生成資料表
在使用django的 migrate 建立生成表的時候，需要加上 –database 引數，如果不加則將 未 指定 app_label 的 APP的models中的表建立到default指定的資料庫中,如:
將app01下models中的表建立到db01的資料庫”db_01”中
| 1 |
./ manage.py migrate --database=db01
|
將app02下models中的表建立到db02的資料庫”db_02”中
| 1 |
./ manage.py migrate --database=db02
|
將app03下models中的表建立到default的資料庫”sqlite3”中
| 1 |
./ manage.py migrate
|
以上建立完成後,其它所有的建立、查詢、刪除等操作就和普通一樣操作就可以了,無需再使用類似
| 1 |
models.User.objects.using(dbname).all(）
|
這樣的方式來操作。