Django自定义AbstractBaseUser模型如何设置主键起始自增值？

方法一：数据库迁移层面设置（推荐） #

这种方法直接在数据库层面配置自增序列的起始值，后续新增用户会自动按规则递增，无需修改视图逻辑。

1. 确保自定义User模型的主键为AutoField（默认即为此类型，也可显式声明）：

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

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

class CustomUser(AbstractBaseUser):
    id = models.AutoField(primary_key=True)
    # 其他字段（如username、email等）
    objects = CustomUserManager()

2. 生成并编辑迁移文件：
   执行python manage.py makemigrations生成迁移文件后，打开对应的迁移文件（如yourapp/migrations/0001_initial.py），在operations列表中添加RunSQL操作来重置序列起始值：

from django.db import migrations, models

class Migration(migrations.Migration):
    initial = True

    dependencies = [
        # 原有依赖项
    ]

    operations = [
        # 原有创建表的操作...
        migrations.RunSQL(
            # 替换为你的应用名和模型名，格式为appname_modelname_id_seq
            "ALTER SEQUENCE yourapp_customuser_id_seq RESTART WITH 2022001;",
            reverse_sql="ALTER SEQUENCE yourapp_customuser_id_seq RESTART WITH 1;"
        )
    ]

执行python manage.py migrate完成设置，之后新用户的主键会从2022001开始自动递增。

方法二：在注册视图中手动设置主键 #

如果不想修改迁移文件，可在注册逻辑中手动计算并设置主键值：

修改你的register_user视图：

from yourapp.models import CustomUser  # 导入自定义User模型

def register_user(request):
    if request.method == 'POST':
        form = RegistrationForm(request.POST)
        if form.is_valid():
            # 生成用户实例但暂不保存到数据库
            user = form.save(commit=False)
            # 获取当前最大主键，若无用户则用起始值前一位，加1得到初始值
            max_id = CustomUser.objects.aggregate(models.Max('id'))['id__max']
            user.id = max_id + 1 if max_id else 2022001
            user.save()
            # 后续认证登录逻辑不变
            username = form.cleaned_data['username']
            password = form.cleaned_data['password1']
            user = authenticate(username=username, password=password)
            login(request, user)
            messages.success(request, "You have successfully registered.")
            return redirect('home')
    else:
        form = RegistrationForm()

    return render(request, 'register/register_user.html', {'form': form})

注意：这种方法在高并发场景下可能出现主键冲突（多个请求同时拿到相同的max_id），适合低流量的站点使用。

内容的提问来源于stack exchange，提问作者Pizza is the best