如何在FastAPI中实现用户注册密码验证功能？

第一步：改造密码验证函数 #

你的原函数仅打印结果，需要修改为返回验证状态和具体错误信息，这样FastAPI才能返回精准的错误提示：

def validate_password(password: str) -> tuple[bool, str]:
    l, u, p, d = 0, 0, 0, 0
    capitalalphabets = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
    smallalphabets = "abcdefghijklmnopqrstuvwxyz"
    specialchar = """ ~`!@#$%^&*()_-+={[}]|\:;"'<,>.?/ """
    digits = "0123456789"

    # 检查密码长度
    if len(password) < 8:
        return False, "密码长度不能少于8位"
    
    for char in password:
        if char in smallalphabets:
            l += 1
        elif char in capitalalphabets:
            u += 1
        elif char in digits:
            d += 1
        elif char in specialchar:
            p += 1
        else:
            # 存在未允许的字符
            return False, "密码包含非法字符，仅支持大小写字母、数字和指定特殊字符"
    
    # 逐一检查各类型字符要求
    if l < 1:
        return False, "密码必须包含至少一个小写字母"
    if u < 1:
        return False, "密码必须包含至少一个大写字母"
    if d < 1:
        return False, "密码必须包含至少一个数字"
    if p < 1:
        return False, "密码必须包含至少一个指定的特殊字符"
    
    return True, "密码验证通过"

第二步：集成到FastAPI中 #

推荐两种常用的集成方式，根据场景选择：

方式一：使用Pydantic字段验证器（最规范）

利用FastAPI依赖的Pydantic模型做请求参数校验，自动返回标准化错误：

from fastapi import FastAPI
from pydantic import BaseModel, field_validator

app = FastAPI()

# 上面的validate_password函数放在这里

# 定义注册请求模型
class RegisterRequest(BaseModel):
    username: str
    password: str

    # 给password字段添加验证器
    @field_validator("password")
    def check_password_rules(cls, value):
        is_valid, error_msg = validate_password(value)
        if not is_valid:
            raise ValueError(error_msg)
        return value

# 注册接口
@app.post("/register")
async def register(user: RegisterRequest):
    # 这里编写实际的注册逻辑（比如写入数据库）
    return {"status": "success", "message": f"用户 {user.username} 注册成功"}

当密码不符合规则时，FastAPI会自动返回422 Unprocessable Entity状态码，响应体包含详细错误：

{
  "detail": [
    {
      "type": "value_error",
      "loc": ["body", "password"],
      "msg": "密码必须包含至少一个大写字母",
      "input": "test123!",
      "ctx": {"error": "密码必须包含至少一个大写字母"}
    }
  ]
}

方式二：路由内手动校验（灵活简单）

如果不需要Pydantic的自动校验，也可以在路由函数里直接调用验证函数，手动抛出错误：

from fastapi import FastAPI, HTTPException

app = FastAPI()

# 上面的validate_password函数放在这里

@app.post("/register")
async def register(username: str, password: str):
    is_valid, error_msg = validate_password(password)
    if not is_valid:
        # 抛出400错误，返回具体原因
        raise HTTPException(status_code=400, detail=error_msg)
    
    # 注册逻辑
    return {"status": "success", "message": f"用户 {username} 注册成功"}

这种方式会返回400 Bad Request状态码，响应体格式如下：

{
  "detail": "密码长度不能少于8位"
}

测试验证 #

启动FastAPI服务后，可通过自动生成的/docs界面（如http://localhost:8000/docs）测试不同密码场景，验证错误提示是否符合预期。

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