FastAPI 如何将SQLAlchemy查询结果转换为嵌套字典

在本文中，我们将介绍如何将SQLAlchemy查询结果转换为嵌套字典。FastAPI是一个基于Python的现代Web框架，它使用SQLAlchemy作为其默认的数据库访问工具。SQLAlchemy是一个功能强大的关系型数据库工具包，它提供了多种用于数据库操作的API。

阅读更多：FastAPI 教程

SQLAlchemy查询结果简介

在使用SQLAlchemy进行数据库查询后，我们通常会得到一个包含多个行的结果集。每一行都是一个Python对象，这些对象的属性对应数据库表的列。要将查询结果转换为嵌套字典，我们需要遍历每一行并将其转换为字典。

将SQLAlchemy查询结果转换为嵌套字典的方法

下面是将SQLAlchemy查询结果转换为嵌套字典的方法：

定义一个用于转换行的函数。

def row_to_dict(row):
    result = {}
    for key in row.__table__.columns.keys():
        value = getattr(row, key)
        result[key] = value
    return result

这个函数用于将给定的行对象转换为字典。它会遍历行对象的每个属性，并将属性名作为键，属性值作为值存储在字典中。

遍历查询结果并使用转换函数将每一行转换为字典。

result_dicts = []
for row in query_result:
    result_dict = row_to_dict(row)
    result_dicts.append(result_dict)

在这个示例中，我们遍历了查询结果并使用row_to_dict函数将每一行转换为字典。我们将每个字典存储在result_dicts列表中。

对转换后的字典进行处理或返回给前端。

# 处理转换后的字典
for result_dict in result_dicts:
    # 对字典进行处理操作
    pass

# 返回给前端
return result_dicts

在这个示例中，我们可以对转换后的字典进行任何处理操作，比如修改、删除或者其他操作。另外，我们还可以将转换后的字典直接返回给前端。

示例

假设我们有一个名为User的数据库表，它包含id、name和age等列。我们可以使用SQLAlchemy进行查询并将结果转换为嵌套字典：

from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

# 创建数据库引擎和会话
engine = create_engine("sqlite:///database.db")
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
session = SessionLocal()

# 执行查询
query_result = session.query(User).all()

# 将查询结果转换为嵌套字典
result_dicts = []
for row in query_result:
    result_dict = row_to_dict(row)
    result_dicts.append(result_dict)

# 对转换后的字典进行处理或返回给前端
for result_dict in result_dicts:
    # 对字典进行处理操作
    pass

# 返回给前端
return result_dicts

在上面的示例中，我们使用了SQLAlchemy的查询接口来获取User表的所有行。然后，我们使用之前定义的row_to_dict函数将每一行转换为字典。最后，我们对转换后的字典进行处理或返回给前端。