FastAPI中如何将字符串转为模型类型？SQLAlchemy结果转Pydantic对象

1. 解决字符串转模型类型及报错问题 #

报错str object has no attribute all()的核心原因是你直接对字符串"student"调用了SQLAlchemy查询集的all()方法，但字符串本身不具备该方法。要解决这个问题，需要先将字符串映射到对应的SQLAlchemy模型类，再基于模型执行查询操作。

实现步骤： #

• 创建模型映射字典，将字符串名称与实际的SQLAlchemy模型类绑定
• 在convert_model函数中，先通过字典获取模型类，再执行查询

示例代码：

from sqlalchemy.orm import Session
from fastapi import FastAPI
# 从你的模型文件导入Student模型
from models import Student

# 模型名称与类的映射字典
MODEL_MAPPING = {
    "student": Student,
    # 可扩展添加其他模型，比如"teacher": Teacher
}

app = FastAPI()

def convert_model(model_name: str, db: Session):
    # 通过映射字典获取对应的模型类
    model_cls = MODEL_MAPPING.get(model_name)
    if not model_cls:
        raise ValueError(f"未知模型名称: {model_name}")
    # 基于模型类执行查询，而非直接操作字符串
    query_result = db.query(model_cls).all()
    return query_result

调用convert_model("student", db)时，会先获取到Student模型类，再执行查询，不会再触发字符串调用all()的报错。

2. 将SQLAlchemy查询结果转换为Pydantic对象 #

需要先定义与SQLAlchemy模型对应的Pydantic模型，再利用Pydantic的from_orm方法完成转换（支持单条或批量数据）。

实现步骤： #

• 定义Pydantic响应模型，与SQLAlchemy模型字段对应
• 将查询结果（单个实例或列表）转换为Pydantic对象

示例代码：

from pydantic import BaseModel
from typing import List

# 定义Pydantic响应模型
class StudentResponse(BaseModel):
    id: int
    name: str
    age: int
    # 按需添加Student模型的其他字段

    class Config:
        orm_mode = True  # 开启ORM模式，支持从SQLAlchemy实例转换

# 修改convert_model函数，返回Pydantic对象列表
def convert_model(model_name: str, db: Session) -> List[StudentResponse]:
    model_cls = MODEL_MAPPING.get(model_name)
    if not model_cls:
        raise ValueError(f"未知模型名称: {model_name}")
    query_result = db.query(model_cls).all()
    # 批量转换为Pydantic对象
    return [StudentResponse.from_orm(item) for item in query_result]

单条数据转换示例： #

如果查询的是单个实例（比如db.query(Student).get(1)），直接执行StudentResponse.from_orm(student_instance)即可完成转换。

完整整合示例 #

from fastapi import FastAPI, Depends, HTTPException
from sqlalchemy.orm import Session
from pydantic import BaseModel
from typing import List
# 假设数据库会话依赖和模型定义在以下文件
from database import get_db
from models import Student

app = FastAPI()

MODEL_MAPPING = {
    "student": Student
}

class StudentResponse(BaseModel):
    id: int
    name: str
    age: int

    class Config:
        orm_mode = True

def convert_model(model_name: str, db: Session) -> List[StudentResponse]:
    model_cls = MODEL_MAPPING.get(model_name)
    if not model_cls:
        raise HTTPException(status_code=400, detail=f"未知模型: {model_name}")
    db_objects = db.query(model_cls).all()
    return [StudentResponse.from_orm(obj) for obj in db_objects]

# FastAPI接口示例
@app.get("/models/{model_name}")
def get_model_data(model_name: str, db: Session = Depends(get_db)):
    return convert_model(model_name, db)

调用GET /models/student接口时，会返回转换后的Pydantic格式学生数据列表，同时彻底解决了之前的报错问题。

内容的提问来源于stack exchange，提问作者M.K.S