只需格式化 FastApi 端点返回的 SQLAlchemy 模型

Question

假设我有一个简单的 SQLAlchemy 类和一个简单的 Flask o FastAPI 实现，如下所示：

from sqlalchemy.ext.declarative import declarative_base
from pydantic import BaseModel
Base = declarative_base()

class A(Base):
    __tablename__ = 'as'
    my_id = Column(String)

class AModel(BaseModel):
    myId:str = None

还有一个像这样的简单端点：

@app_router.get('/a')
def get_all_a(session:Session = Depends(get_session)):
    return session.query(A).all()

我怎样才能确保这个端点的返回列表在 camelCase 中产生如下：

[{'myId': 'id1'},{'myId': 'id2'}, ...]

注意：我的应用程序相当复杂，因为我也实现了分页，并且一些后处理需要更多的操作，而不仅仅是将蛇形大小写转换为驼峰大小写，因此最简单的解决方案是最好的。

我尝试过重写 dict() 方法和类似的东西，但没有成功，根本无法理解 FastAPI 如何处理结果以获取 JSON。

Answer 1

需要更多的只是从蛇形大小写到骆驼大小写的转换

好吧，如果你不使用 response_model 你没有很多选择。

解决方案是通过从蛇形大小写转换为骆驼大小写来返回您的 dict。您有递归执行此操作的函数。

为什么它是最好的解决方案？

使用正则表达式它超级快，比任何将 dict 转换为 obj 的库（如 pydantic）都快

如果您绝对不想这样做，那么您唯一的解决方案是使用 pydantic 模型、attrs 或数据类，并将您的 db 查询输出转换为带有 camelCase 变量名称（脏）的模型类型之一。

由于您使用的是 fastapi，因此您应该使用它的所有功能。

我建议这样做：

from typing import List

from sqlalchemy.ext.declarative import declarative_base
from pydantic import BaseModel, Field
from pydantic import parse_obj_as
Base = declarative_base()

class A(Base):
    __tablename__ = 'as'
    my_id = Column(String)

class AModel(BaseModel):
    myId: str = Field(alias="my_id", default=None)

@app_router.get('/a', response_model=List[AModel])
def get_all_a(session:Session = Depends(get_session)):
    return parse_obj_as(List[AModel], session.query(A).all())

请记住，在 CamelCase 中使用类变量不是一个好习惯。

黄金答案是不返回骆驼而是蛇，如果需要，让您的客户完成转换工作。