FastAPI新手疑问：用Pydantic模型校验非对象关联信息是否合理？

这种在Pydantic模型内部校验数据库中是否存在同名用户的做法并不是最佳实践，主要原因如下：

• 职责混淆：Pydantic的核心定位是做数据的结构定义、类型校验和格式校验（比如你的用户名正则规则），而数据库查询、业务规则校验属于业务逻辑层的范畴。把两者混在一起会让模型职责模糊，后续维护和测试都会变麻烦。
• 依赖注入不便：在Pydantic的field_validator里很难优雅地注入数据库连接（比如SQLAlchemy Session），硬编码连接会导致测试时无法替换为测试数据库，违背依赖注入的设计原则。
• 复用性差：这个模型会和数据库强绑定，没法在不需要数据库的场景（比如单元测试、离线数据处理）复用。

推荐的做法 #

把校验拆分成两层：

1. 格式校验：留在Pydantic模型中，负责确保输入数据的格式、类型、长度符合要求；
2. 业务规则校验：放到FastAPI的路由处理函数，或者专门的业务逻辑服务中，处理和外部资源（比如数据库）相关的校验。

示例代码

1. 纯格式校验的Pydantic模型

from pydantic import BaseModel, Field, ConfigDict

class CreateUser(BaseModel):
    model_config = ConfigDict(strict=True)
    username: str = Field(pattern=r"[0-9a-zA-Z!@#$%&*_.-]{3,}")
    password: str
    secret: str

2. 路由中处理业务校验

from fastapi import FastAPI, HTTPException, Depends
from sqlalchemy.orm import Session
# 假设你的数据库模块定义了get_db和User模型
from your_db_module import get_db, User

app = FastAPI()

@app.post("/users/")
def create_user(user: CreateUser, db: Session = Depends(get_db)):
    # 校验用户名是否已存在
    if db.query(User).filter(User.username == user.username).first():
        raise HTTPException(status_code=400, detail="用户名已存在")
    
    # 执行创建用户的逻辑
    new_user = User(**user.model_dump())
    db.add(new_user)
    db.commit()
    db.refresh(new_user)
    
    return {"id": new_user.id, "username": new_user.username}

如果有大量重复的业务校验逻辑，可以把这些逻辑封装成FastAPI依赖项或者专门的业务服务类，进一步提升代码的可复用性和可维护性。

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