在 SQLAlchemy 中以 dict 形式检索查询结果
P粉187160883
P粉187160883 2023-10-20 13:59:52
0
2
751

我正在使用 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,而只使用 SQLAlcehmy 功能,即我不想在我的代码中的任何地方都有“import pymysql”。

P粉187160883
P粉187160883

全部回复(2)
P粉486743671

您可以使用sqlalchemy游标和游标的描述。

def rows_as_dicts(cursor):
    """convert tuple result to dict with cursor"""
    col_names = [i[0] for i in cursor.description]
    return [dict(zip(col_names, row)) for row in cursor]


db = SQLAlchemy(app)
# get cursor
cursor = db.session.execute(sql).cursor
# tuple result to dict
result = rows_as_dicts(cursor)
P粉421119778

更新了 SQLAlchemy 1.4 的答案:

版本 1.4 已弃用旧的 engine.execute() 模式,并更改了 .execute() 内部运行的方式。 .execute() 现在返回 CursorResult 对象,带有 .mappings ()方法:

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 的回答)

如果您使用 engine.execute 而不是 raw_connection(),SQLAlchemy 已经为您完成了此操作。使用engine.executefetchone将返回一个SQLAlchemy Row对象,fetchall将返回一个list 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'}
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责声明 Sitemap
PHP中文网:公益在线PHP培训,帮助PHP学习者快速成长!