FastAPI如何管理WTF Forms？Flask迁移类式表单实现疑问

FastAPI本身没有官方集成WTForms的类式用法，但完全可以通过自定义依赖项实现和Flask中类似的类式表单管理，下面是具体方案：

1. 保留WTForms表单类定义 #

先按照你的习惯定义WTForms表单类，补充完整字段：

from wtforms import Form, StringField, PasswordField
from wtforms.validators import DataRequired, Length, Email, EqualTo

class SignUpForm(Form):
    email = StringField('Email', validators=[DataRequired(), Length(1, 100), Email()])
    password = PasswordField('Password', validators=[DataRequired(), Length(min=6)])
    confirm_password = PasswordField('Confirm Password', validators=[DataRequired(), EqualTo('password', message='Passwords must match')])

2. 编写FastAPI依赖项适配WTForms #

创建一个通用依赖函数，用来解析请求表单数据、实例化WTForms类并完成验证：

from fastapi import Request, HTTPException, Depends
from typing import Type

def get_wtform(form_class: Type[Form]):
    async def form_dependency(request: Request):
        form_data = await request.form()
        form = form_class(form_data)
        if not form.validate():
            # 收集验证错误并返回422状态码
            errors = {field: errors for field, errors in form.errors.items()}
            raise HTTPException(status_code=422, detail=errors)
        return form
    return form_dependency

3. 在路由中使用类式表单 #

在接口路由里通过Depends调用上面的依赖，直接获取验证后的WTForms实例：

from fastapi import FastAPI

app = FastAPI()

@app.post("/signup")
async def signup(form: SignUpForm = Depends(get_wtform(SignUpForm))):
    # 直接使用表单数据，比如 form.email.data、form.password.data
    return {
        "email": form.email.data,
        "message": "Signup data validated successfully"
    }

关于FastAPI表单管理机制 #

FastAPI底层基于Starlette处理表单请求，默认通过单个Form参数接收表单字段，更倾向于用Pydantic模型结合Form字段类实现类式表单（官网教程的方式），但如果习惯WTForms的验证逻辑和字段定义，上面的依赖适配方案是最便捷的替代方式。

如果想完全贴合FastAPI原生生态，也可以用Pydantic模型定义表单，通过Annotated和Form标记每个字段，示例：

from pydantic import BaseModel, EmailStr, Field, model_validator
from fastapi import Form, Depends
from typing import Annotated

class SignUpModel(BaseModel):
    email: EmailStr
    password: str = Field(min_length=6)
    confirm_password: str = Field(min_length=6)

    # 自定义验证逻辑，对应WTForms的EqualTo
    @model_validator(mode="after")
    def check_passwords_match(self):
        if self.password != self.confirm_password:
            raise ValueError("Passwords must match")
        return self

# 定义依赖将表单数据转成Pydantic模型
def form_to_model(
    email: Annotated[str, Form()],
    password: Annotated[str, Form()],
    confirm_password: Annotated[str, Form()]
) -> SignUpModel:
    return SignUpModel(email=email, password=password, confirm_password=confirm_password)

@app.post("/signup-pydantic")
async def signup_pydantic(model: SignUpModel = Depends(form_to_model)):
    return {"email": model.email, "message": "Data validated successfully"}

内容的提问来源于stack exchange，提问作者Nicolas-Fractal