【Django】连接使用多个数据库 -- 2019-08-08 18:03:24

原文:

@

第一步： 配置你要使用的数据库

---

在 setting.py 配置文件中的 DATABASES 字典中指定。
注意：默认的 default 数据库别名不可删除，如果不使用，可留空 {} 。

# 先定义好数据库别名
DB01 = 'db01'  # 第一个数据库别名
DB02 = 'db02'  # 第二个数据库别名

DATABASES = {
    'default': {},  # 不可删除，留空 {}
    DB01: {  # 第一个数据库
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'db01',
        'HOST': 'localhost',
        'POST': '3306',
        'USER': 'u1',
        'PASSWORD': 'user@u1',
    },
    DB02: {  # 第二个数据库
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'db02',
        'HOST': 'localhost',
        'POST': '3306',
        'USER': 'u2',
        'PASSWORD': 'user@u2',
    }
}

第二步： 定义数据库路由方法类

---

使用多个数据库时，最简单的方法就是设置数据库路由方案，以保证对象对原始数据库的“粘性”，默认所有的查询都会返回到 default 数据库中。

------------↓
数据库路由类方法详述：

1. __db_for_read(model, **hints)__：应用于读取类型对象的数据库模型，如果数据库提供附加信息会在hints字典中提供，最后如果没有则返回None。
2. __db_for_write(model, **hints)__：应用于写入类型对象的数据库模型，hints字典提供附加信息，如果没有则返回None。
3. __allow_relation(obj1, obj2, **hints)__：外键操作，判断两个对象之间是否是应该允许关系，是返回True,否则返回False，如果路由允许返回None。
4. __allow_migrate(db, app_label, model_name=None, **hints)__：db确定是否允许在具有别名的数据库上运行迁移操作，操作运行返回True，否则返回False，或者返回None，如果路由器没有意见。
   
   

位置参数解释：

• app_label：正在迁移的应用程序的标签。
• model_name：多个迁移操作设置模型的值，如：model._meta.app_label

------------↑

实现数据库路由的具体步骤如下

1.在 settings.py 文件中配置app与数据库匹配路由表，采用字典形式，app名对应数据库别名：

DB01 = 'db01'  # 第一个数据库别名
DB02 = 'db02'  # 第二个数据库别名

# 数据库路由表
DATABASE_APPS_MAPPING = {
    'app1': DB01,
    'app2': DB02,
}

2.在与 setting.py 同级的目录下创建路由表，app应用会根据上面指定的路由选择数据库。

# db_router文件内容如下

from django.conf import settings

DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING  # 在setting中定义的路由表


class DatabaseAppsRouter(object):
    def db_for_read(self, model, **hints):
        if model._meta.app_label in DATABASE_MAPPING:
            return DATABASE_MAPPING[model._meta.app_label]
        return None

    def db_for_write(self, model, **hints):
        if model._meta.app_label in DATABASE_MAPPING:
            return DATABASE_MAPPING[model._meta.app_label]
        return None

    def allow_relation(self, obj1, obj2, **hints):
        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):
        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_name=None, **hints):
        if db in DATABASE_MAPPING.values():
            return DATABASE_MAPPING.get(app_label) == db
        elif app_label in DATABASE_MAPPING:
            return False
        return None

注：
这里的 db_for_read 与 db_for_write 方法中的逻辑是根据app进行分库分表的，即不同的app使用不同的数据库。
如果想要读写分离的逻辑，可见文章补充部分。

3.在 settings.py 文件中配置 DATABASE_ROUTERS 来指定路由表：

# 指定数据库路由表
DATABASE_ROUTERS = ['blog.db_router.DatabaseAppsRouter']
# blog：项目名称
# db_router：路由文件名称
# DatabaseAppRouter：路由文件中的路由类名称