如何在Django项目中根据登录用户动态设置默认Postgres数据库

嘿，针对你这个需求——根据登录用户自动切换默认数据库（主库给正式用户，测试库给实习生），不用每次查询都手动加using(db_name)，我给你整理了一套落地性很强的方案，完全能满足给客户提供独立测试版本的需求！

首先，先确认你的settings.py里已经配置好两个数据库了，大概是这样：

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.postgresql',
        'NAME': 'prod_main_db',
        'USER': 'prod_db_user',
        'PASSWORD': 'your_prod_pass',
        'HOST': 'prod_db_host',
        'PORT': '5432',
    },
    'intern_test_db': {
        'ENGINE': 'django.db.backends.postgresql',
        'NAME': 'intern_test_db',
        'USER': 'test_db_user',
        'PASSWORD': 'your_test_pass',
        'HOST': 'test_db_host',
        'PORT': '5432',
    }
}

接下来核心思路是：在用户登录后，通过中间件动态修改当前请求的数据库连接配置，让整个请求周期里的所有数据库操作都自动用对应的库，不用手动指定。

步骤1：编写自定义中间件 #

在你的业务app里新建一个middleware.py文件，代码如下：

from django.conf import settings
from django.contrib.auth import get_user_model
from django.db import connections

User = get_user_model()

class DynamicDBMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response
        # 提前保存原始默认数据库配置，防止后续覆盖丢失
        if 'original_default' not in settings.DATABASES:
            settings.DATABASES['original_default'] = settings.DATABASES['default'].copy()

    def __call__(self, request):
        # 只对已登录用户做切换
        if request.user.is_authenticated:
            # 这里根据你的业务规则判断用户用哪个库，比如检查用户是否是实习生
            # 假设你的User模型有个is_intern的布尔字段
            if request.user.is_intern:
                # 切换默认数据库为测试库
                connections['default'].settings_dict = settings.DATABASES['intern_test_db']
            else:
                # 恢复为原始主库
                connections['default'].settings_dict = settings.DATABASES['original_default']
        else:
            # 未登录用户默认用主库
            connections['default'].settings_dict = settings.DATABASES['original_default']

        response = self.get_response(request)
        return response

步骤2：注册中间件 #

把这个中间件加到settings.py的MIDDLEWARE列表里，一定要放在认证中间件之后，这样才能拿到已登录的用户信息：

MIDDLEWARE = [
    # ... 其他系统中间件
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    # 你的自定义中间件，替换成实际的app名称
    'your_app_name.middleware.DynamicDBMiddleware',
    # ... 其他中间件
]

步骤3：添加用户区分字段 #

你需要在User模型里加一个标识字段，用来区分普通用户和实习生，比如：

from django.contrib.auth.models import AbstractUser
from django.db import models

class CustomUser(AbstractUser):
    is_intern = models.BooleanField(default=False, verbose_name='是否为实习生用户')

如果是新增的字段，别忘了生成迁移并执行：

python manage.py makemigrations
python manage.py migrate

关键注意事项（避坑指南） #

• 线程/进程安全：直接修改全局settings可能在多线程环境出问题，所以上面的代码用了connections['default'].settings_dict，针对每个请求的数据库连接单独修改，更安全。如果你的服务器是多进程模式（比如Gunicorn多worker），每个worker有独立的settings副本，也不用担心冲突。
• 连接池问题：如果你的项目用了数据库连接池（比如django-db-connection-pool），要确保连接池能正确识别动态切换的配置，避免复用错误的连接。建议测试时重点验证这一点。
• 测试场景验证：写测试用例时，分别用普通用户和实习生用户登录，检查数据库操作是否真的切换到了对应的库，比如创建一条记录，然后去对应的数据库里查是否存在。
• 缓存与会话隔离：如果用了缓存，建议给实习生用户的缓存加前缀，避免正式数据和测试数据混淆。

这样配置完之后，用户登录后，整个请求里的所有ORM操作都会自动用对应的数据库，完全不用手动写using()，完美实现你要的独立测试数据库版本的需求！

内容的提问来源于stack exchange，提问作者Harshit verma