如何在SQLAlchemy查询结果中移除主键并适配Pydantic模型?
更优解决方案推荐
方法一:直接用Pydantic的from_orm方法
Pydantic的BaseModel自带from_orm方法,能直接从SQLAlchemy的ORM对象生成响应模型,自动忽略响应模型中未定义的字段(比如id),代码极简:
query_result = db.query(models.User).first() return UserResponseBody.from_orm(query_result)
这是最推荐的方式,完全不需要手动处理字典或删除字段,Pydantic会自动匹配字段、过滤冗余数据。
方法二:查询时只选择需要的字段
从根源上解决问题,查询数据库时只获取name和age字段,返回的结果本身就不含id:
query_result = db.query(models.User.name, models.User.age).first() # 两种方式生成响应模型 return UserResponseBody(name=query_result.name, age=query_result.age) # 或者更简洁的写法: return UserResponseBody(**query_result._asdict())
这种方式减少了数据库传输的数据量,性能更优,同时避免了后续的字段过滤操作。
方法三:使用SQLAlchemy的load_only加载指定字段
如果需要保留完整的ORM对象用于后续操作,但返回响应时只需要部分字段,可以用load_only选项控制加载的字段:
from sqlalchemy.orm import load_only query_result = db.query(models.User).options(load_only("name", "age")).first() return UserResponseBody.from_orm(query_result)
内容的提问来源于stack exchange,提问作者Joe
相关产品推荐
相关产品推荐

