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

在本文中，我们将介绍如何使用FastAPI将SQLAlchemy的查询结果转换为嵌套字典。SQLAlchemy是一种流行的Python ORM（对象关系映射）库，用于在Python应用程序中与数据库进行交互。FastAPI是一个快速（高性能）的Web框架，它可以与SQLAlchemy无缝集成。当处理数据库查询结果时，有时候我们需要将结果转换为嵌套字典的形式，以便更好地组织和展示数据。

阅读更多：FastAPI 教程

SQLAlchemy 查询结果

在使用SQLAlchemy进行查询时，我们通常会得到一个查询结果，该结果以行的形式表示。每一行数据都可以表示为一个字典或命名元组。以下是一个示例：

from sqlalchemy import create_engine, select, Table, Column, Integer, String, MetaData

# 创建数据库引擎和元数据
engine = create_engine('sqlite:///example.db')
metadata = MetaData(bind=engine)

# 创建一个表
users = Table('users', metadata,
    Column('id', Integer, primary_key=True),
    Column('name', String),
    Column('age', Integer),
)

# 执行查询
stmt = select([users])
result_proxy = engine.execute(stmt)
results = result_proxy.fetchall()

# 遍历结果集
for row in results:
    print(row)

以上代码创建了一个SQLite数据库，并在其中创建了一个名为”users”的表。然后执行了一个查询，将结果保存在results变量中。接下来，我们将介绍如何将上述查询结果转换为嵌套字典的形式。

将查询结果转换为嵌套字典

首先，我们需要定义一个函数，该函数将递归地将行数据转换为嵌套字典。以下是一个示例函数：

def row_to_dict(row):
    result = {}
    for column, value in row.items():
        if isinstance(value, dict):
            result[column] = row_to_dict(value)
        else:
            result[column] = value
    return result

以上函数使用递归的方式将行数据转换为嵌套字典。对于每一列，函数将列名作为键，列值作为对应的值，并递归处理嵌套的字典。

接下来，我们可以使用上述函数将查询结果转换为嵌套字典的形式。以下是示例代码：

nested_dicts = [row_to_dict(row) for row in results]
print(nested_dicts)

以上代码使用列表推导式遍历查询结果中的每一行，并将每一行数据转换为嵌套字典。最终，将所有转换后的嵌套字典保存在nested_dicts列表中，并打印输出。

完整示例

我们将上述代码片段整合到一个完整的示例中，以便更好地理解如何将SQLAlchemy的查询结果转换为嵌套字典。以下是示例代码：

from sqlalchemy import create_engine, select, Table, Column, Integer, String, MetaData

# 定义将行数据转换为嵌套字典的函数
def row_to_dict(row):
    result = {}
    for column, value in row.items():
        if isinstance(value, dict):
            result[column] = row_to_dict(value)
        else:
            result[column] = value
    return result

# 创建数据库引擎和元数据
engine = create_engine('sqlite:///example.db')
metadata = MetaData(bind=engine)

# 创建一个表
users = Table('users', metadata,
    Column('id', Integer, primary_key=True),
    Column('name', String),
    Column('age', Integer),
)

# 执行查询
stmt = select([users])
result_proxy = engine.execute(stmt)
results = result_proxy.fetchall()

# 将查询结果转换为嵌套字典
nested_dicts = [row_to_dict(row) for row in results]
print(nested_dicts)

将以上代码保存为Python脚本并运行，将会输出嵌套字典形式的查询结果。

总结

在本文中，我们介绍了如何使用FastAPI将SQLAlchemy的查询结果转换为嵌套字典。通过定义一个递归函数，我们可以将每一行数据转换为嵌套字典的形式，并更好地组织和展示数据。这对于处理数据库查询结果非常有用，尤其是在构建Web应用程序时。希望本文能帮助您在FastAPI中处理SQLAlchemy查询结果时更加灵活和高效。

🚀 FastAPI，享受高性能Web开发的乐趣！🚀