在 SQLAlchemy 中以 dict 形式检索查询结果

2024-01-12

我正在使用 Flask SQLAlchemy,并且有以下代码可以通过来自 MySQL 数据库的原始 SQL 查询从数据库获取用户:

connection = engine.raw_connection()
cursor = connection.cursor()
cursor.execute("SELECT * from User where id=0")
results = cursor.fetchall()

results变量是一个元组,我希望它的类型为 dict()。有办法实现这一点吗?

当我使用 pymysql 构建数据库连接时,我能够做到

cursor = connection.cursor(pymysql.cursors.DictCursor)

SQLAlchemy中有类似的东西吗?

注意:我想要进行此更改的原因是为了摆脱在我的代码中使用 pymysql,而只使用 SQLAlchemy 功能,即我不想在我的代码中的任何地方有“import pymysql”。


results 是一个元组,我希望它的类型为 dict()

SQLAlchemy 1.4 的更新答案:

1.4 版本已弃用旧版本engine.execute()模式并改变了方式.execute()内部运作。.execute()现在返回一个游标结果 https://docs.sqlalchemy.org/en/14/core/connections.html#sqlalchemy.engine.CursorResult对象具有.mappings() https://docs.sqlalchemy.org/en/14/core/connections.html#sqlalchemy.engine.CursorResult.mappings method:

import sqlalchemy as sa

# …

with engine.begin() as conn:
    qry = sa.text("SELECT FirstName, LastName FROM clients WHERE ID < 3")
    resultset = conn.execute(qry)
    results_as_dict = resultset.mappings().all()
    pprint(results_as_dict)
    """
    [{'FirstName': 'Gord', 'LastName': 'Thompson'}, 
     {'FirstName': 'Bob', 'LastName': 'Loblaw'}]
    """

(之前对 SQLAlchemy 1.3 的回答)

如果您使用 SQLAlchemy 已经为您完成此操作engine.execute代替raw_connection(). With engine.execute, fetchone将返回一个 SQLAlchemyRow对象和fetchall将返回一个list of Row对象。Row对象可以通过键来访问,就像dict:

sql = "SELECT FirstName, LastName FROM clients WHERE ID = 1"
result = engine.execute(sql).fetchone()
print(type(result))  # <class 'sqlalchemy.engine.result.Row'>
print(result['FirstName'])  # Gord

如果你需要一个真实的dict对象然后你可以转换它:

my_dict = dict(result)
print(my_dict)  # {'FirstName': 'Gord', 'LastName': 'Thompson'}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在 SQLAlchemy 中以 dict 形式检索查询结果 的相关文章

随机推荐