如何修改Django用户会话键字段且不破坏活跃会话？

步骤1：添加新的用户唯一标识符字段 #

首先，给自定义User模型添加一个用于替代用户名的自增字段（比如internal_id），先允许为空以便填充现有数据：

# models.py
from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager

class CustomUserManager(BaseUserManager):
    # 你的用户管理器逻辑...

class CustomUser(AbstractBaseUser):
    username = models.CharField(max_length=150, primary_key=True, unique=True)
    # 添加新的唯一标识符字段
    internal_id = models.BigAutoField(unique=True, null=True)
    # 其他字段...

    objects = CustomUserManager()
    USERNAME_FIELD = 'username'

生成并执行第一次迁移：

python manage.py makemigrations
python manage.py migrate

给所有现有用户填充internal_id的值，可在Django Shell中执行：

python manage.py shell

from yourapp.models import CustomUser

# 让AutoField自动为未设置值的用户生成ID
for user in CustomUser.objects.filter(internal_id__isnull=True):
    user.save()

修改模型将internal_id设为非空：

internal_id = models.BigAutoField(unique=True, null=False)

再次生成并执行迁移：

python manage.py makemigrations
python manage.py migrate

步骤2：自定义认证后端兼容两种会话键 #

创建自定义认证后端，让系统同时识别旧会话的用户名和新会话的internal_id：

# backends.py
from django.contrib.auth.backends import ModelBackend
from django.contrib.auth import get_user_model

User = get_user_model()

class DualKeyAuthBackend(ModelBackend):
    def get_user(self, user_id):
        # 优先尝试用新的internal_id查找用户
        try:
            return User.objects.get(internal_id=user_id)
        except User.DoesNotExist:
            # 失败则回退到旧的用户名主键查找
            try:
                return User.objects.get(username=user_id)
            except User.DoesNotExist:
                return None

在settings.py中替换默认认证后端：

AUTHENTICATION_BACKENDS = [
    'yourapp.backends.DualKeyAuthBackend',
    # 保留其他需要的后端（如admin后端）
]

步骤3：让新登录会话使用新标识符 #

修改登录逻辑，让新登录用户的会话存储internal_id而非用户名，这里用登录信号实现更通用：

# signals.py
from django.contrib.auth.signals import user_logged_in
from django.contrib.auth import SESSION_KEY
from django.dispatch import receiver

@receiver(user_logged_in)
def update_session_key(sender, request, user, **kwargs):
    # 将SESSION_KEY替换为internal_id的字符串形式
    request.session[SESSION_KEY] = str(user.internal_id)

在apps.py中注册信号：

# apps.py
from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'yourapp'

    def ready(self):
        import yourapp.signals

步骤4：逐步迁移并最终切换主键 #

• 部署代码后，新登录用户的会话将使用internal_id，旧活跃会话仍能通过用户名正常工作，无中断。
• 等待所有旧会话自然过期（根据SESSION_COOKIE_AGE设置，默认2周）。
• 旧会话过期后，修改User模型的主键：

  class CustomUser(AbstractBaseUser):
      username = models.CharField(max_length=150, unique=True)  # 移除primary_key=True
      internal_id = models.BigAutoField(primary_key=True)  # 设置为主键
      # 其他字段...
  

  生成并执行迁移，同时更新所有关联User模型的外键字段，确保它们指向新主键internal_id。

注意事项 #

• 操作前务必备份数据库，避免数据丢失。
• 大量用户的数据填充和迁移建议在低峰期执行。
• 若使用Django Admin，需确保自定义后端兼容Admin的登录逻辑。

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