Django多資料庫聯用實現方法解析

阿新 • • 發佈：2020-11-13

本文講述在一個 django project 中使用多個數據庫的方法,多個數據庫的聯用以及多資料庫時資料匯入匯出的方法。

直接給出一種簡單的方法吧，想了解更多的到官方教程，點選此處

程式碼

1. 每個app都可以單獨設定一個數據庫

settings.py中有資料庫的相關設定，有一個預設的資料庫 default,我們可以再加一些其它的，比如：

# Database
# https://docs.djangoproject.com/en/1.8/ref/settings/#databases
DATABASES = {
  'default': {
    'ENGINE': 'django.db.backends.sqlite3','NAME': os.path.join(BASE_DIR,'db.sqlite3'),},'db1': {
    'ENGINE': 'django.db.backends.mysql','NAME': 'dbname1','USER': 'your_db_user_name','PASSWORD': 'yourpassword',"HOST": "localhost",'db2': {
    'ENGINE': 'django.db.backends.mysql','NAME': 'dbname2',}
 
# use multi-database in django
# add by WeizhongTu
DATABASE_ROUTERS = ['project_name.database_router.DatabaseAppsRouter']
DATABASE_APPS_MAPPING = {
  # example:
  #'app_name':'database_name','app1': 'db1','app2': 'db2',}

在project_name資料夾中存放 database_router.py 檔案，內容如下：

# -*- coding: utf-8 -*-
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,**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
 
  # for Django 1.4 - Django 1.6
  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
 
  # Django 1.7 - Django 1.11
  def allow_migrate(self,app_label,model_name=None,**hints):
    print db,model_name,hints
    if db in DATABASE_MAPPING.values():
      return DATABASE_MAPPING.get(app_label) == db
    elif app_label in DATABASE_MAPPING:
      return False
    return None

這樣就實現了指定的 app 使用指定的資料庫了,當然你也可以多個sqlite3一起使用，相當於可以給每個app都可以單獨設定一個數據庫！如果不設定或者沒有設定的app就會自動使用預設的資料庫。

2.使用指定的資料庫來執行操作

在查詢的語句後面用 using(dbname) 來指定要操作的資料庫即可

# 查詢
YourModel.objects.using('db1').all()
或者 YourModel.objects.using('db2').all()
# 儲存或刪除
user_obj.save(using='new_users')
user_obj.delete(using='legacy_users')

3.多個數據庫聯用時資料匯入匯出

使用的時候和一個數據庫的區別是:

如果不是defalut(預設資料庫）要在命令後邊加 --database=資料庫對應的settings.py中的名稱如： --database=db1 或 --database=db2

資料庫同步（建立表）

# Django 1.6及以下版本
python manage.py syncdb #同步預設的資料庫，和原來的沒有區別

# 同步資料庫 db1 (注意：不是資料庫名是db1,是settings.py中的那個db1，不過你可以使這兩個名稱相同，容易使用)
python manage.py syncdb --database=db1

# Django 1.7 及以上版本
python manage.py migrate --database=db1

資料匯出

python manage.py dumpdata app1 --database=db1 > app1_fixture.json
python manage.py dumpdata app2 --database=db2 > app2_fixture.json
python manage.py dumpdata auth > auth_fixture.json

資料庫匯入