Django多数据库切换后用户自动登出问题的解决方法求助

问题根源分析 #

你遇到的自动登出问题，本质是两个核心错误导致的：

1. 直接修改connections['default'].alias是错误操作——这个属性是数据库连接初始化时的固定标识，动态修改会导致线程内数据库连接逻辑混乱。
2. Session数据默认存储在default数据库，当你的路由把Session读写导向用户选择的其他数据库时，因目标库没有该用户的Session记录，Django会判定用户未登录，触发自动登出。

正确实现方案 #

我们需要用线程本地存储安全传递当前请求的数据库选择，同时强制Session始终使用default数据库，避免读取不到Session的问题。

1. 创建线程本地存储工具

新建db_utils.py，用于在当前请求线程内存储数据库别名：

import threading

_thread_locals = threading.local()

def set_current_db(db_alias):
    setattr(_thread_locals, 'current_db', db_alias)

def get_current_db():
    return getattr(_thread_locals, 'current_db', 'default')

2. 编写数据库选择中间件

新建middleware.py，替代原装饰器，在请求生命周期内处理登录检查和数据库标识存储：

from django.shortcuts import redirect
from django.urls import reverse, urlencode
from .db_utils import set_current_db

class DatabaseMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        # 未登录用户跳转至登录页（排除登录页本身）
        if not request.user.is_authenticated:
            if not request.path.startswith(reverse('login')):
                next_url = request.get_full_path()
                return redirect(f'{reverse("login")}?{urlencode({"next": next_url})}')
        else:
            # 从Session取出用户选择的数据库，存入线程本地
            db_alias = request.session.get('database', 'default')
            set_current_db(db_alias)
        
        response = self.get_response(request)
        return response

3. 修改数据库路由

更新routers.py，从线程本地获取数据库标识，并强制Session使用default数据库：

from .db_utils import get_current_db

class SessionDatabaseRouter(object):
    def db_for_read(self, model, **hints):
        # Session模型强制使用default数据库
        if model._meta.app_label == 'sessions' and model._meta.model_name == 'session':
            return 'default'
        return get_current_db()

    def db_for_write(self, model, **hints):
        # Session模型强制使用default数据库
        if model._meta.app_label == 'sessions' and model._meta.model_name == 'session':
            return 'default'
        return get_current_db()

    def allow_relation(self, obj1, obj2, **hints):
        return True

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        # Session表仅在default数据库迁移
        if app_label == 'sessions':
            return db == 'default'
        # 其他模型允许在两个数据库迁移（因结构一致）
        return db in ['default', 'db_1402']

4. 配置Settings

在settings.py中注册中间件和路由：

# 中间件列表（注意放在AuthenticationMiddleware之后）
MIDDLEWARE = [
    # ... 其他默认中间件
    'your_app_name.middleware.DatabaseMiddleware',
]

# 数据库路由
DATABASE_ROUTERS = ['your_app_name.routers.SessionDatabaseRouter']

5. 登录视图补充逻辑

确保登录时将用户选择的数据库存入Session：

def login_view(request):
    if request.method == 'POST':
        # ... 你的用户认证逻辑
        # 获取用户选择的数据库并写入Session
        database_choice = request.POST.get('database', 'default')
        request.session['database'] = database_choice
        # ... 后续登录跳转逻辑

原装饰器的处理 #

直接删除原db_login_required装饰器，功能已由中间件替代。

内容的提问来源于stack exchange，提问作者Mohammad Karimi