如何为FastAPI中的Pydantic模型实现自增主键生成？

一、基于SQLAlchemy ORM的标准方案（推荐） #

核心是将数据库自增逻辑与Pydantic验证逻辑分离，这是生产环境最常用的方式：

1. 数据库模型配置自增主键
   在SQLAlchemy的数据库模型中，给id字段添加autoincrement=True，让数据库自动生成自增ID：

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class PinDB(Base):
    __tablename__ = "pins"
    id = Column(Integer, primary_key=True, autoincrement=True)  # 开启数据库自增
    title = Column(String, index=True)
    # 其他业务字段

2. 拆分Pydantic模型
   分别定义「创建请求模型」和「响应返回模型」，避免验证冲突：

from pydantic import BaseModel

# 创建接口专用：不含id字段，用户无需传入
class CreatePin(BaseModel):
    title: str
    # 仅保留需要用户输入的字段

# 响应返回专用：包含id，返回完整数据给前端
class PinResponse(BaseModel):
    id: int
    title: str

    class Config:
        orm_mode = True  # 支持从SQLAlchemy模型自动转换为Pydantic模型

3. 接口层处理逻辑
   接收请求后将数据存入数据库，数据库自动生成ID，刷新后返回给前端：

from fastapi import FastAPI, Depends
from sqlalchemy.orm import Session
from .database import get_db, PinDB
from .schemas import CreatePin, PinResponse

app = FastAPI()

@app.post("/pins/", response_model=PinResponse)
def create_pin(pin: CreatePin, db: Session = Depends(get_db)):
    db_pin = PinDB(**pin.dict())
    db.add(db_pin)
    db.commit()
    db.refresh(db_pin)  # 刷新获取数据库生成的自增ID
    return db_pin

二、无ORM场景下的手动自增方案 #

如果不用ORM（比如内存存储、非关系型数据库），可以通过以下方式实现：

1. Pydantic模型默认生成ID
   利用default_factory自动生成自增ID：

from pydantic import BaseModel, Field
import itertools

# 全局自增计数器
id_counter = itertools.count(1)

class PinSchema(BaseModel):
    id: int = Field(default_factory=lambda: next(id_counter))  # 自动生成ID
    title: str
    # 其他字段

2. 接口层手动生成ID
   也可以在接口中判断ID是否为空，为空则生成新ID：

# 模拟存储，实际项目替换为数据库操作
pins = []
current_max_id = 0

class PinSchema(BaseModel):
    id: int | None = None
    title: str

@app.post("/pins/")
def create_pin(pin: PinSchema):
    global current_max_id
    if pin.id is None:
        current_max_id += 1
        pin.id = current_max_id
    pins.append(pin.dict())
    return pin

三、让ID字段跳过验证的临时方案 #

如果必须在同一个Pydantic模型中保留ID字段，可将其设为可选并忽略传入值，强制使用自动生成的ID：

class PinSchema(BaseModel):
    id: int | None = None
    title: str

@app.post("/pins/")
def create_pin(pin: PinSchema):
    # 忽略用户传入的id，强制生成新的自增ID
    new_id = get_next_id()  # 实现自定义的ID生成逻辑（比如从数据库取最大ID+1）
    pin.id = new_id
    # 保存数据逻辑
    return pin

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