在SQLAlChemy中以字典形式检索查询结果

2022-03-10 00:00:00 python sqlalchemy flask-sqlalchemy mysql

我使用的是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"。


解决方案

结果是一个元组,我希望它是dict()类型

SQLAlChemy 1.4的更新答案:

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

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将返回SQLAlChemyRow对象,fetchall将返回listRow对象。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

如果需要Truedict对象,则只需将其转换即可:

my_dict = dict(result)
print(my_dict)  # {'FirstName': 'Gord', 'LastName': 'Thompson'}

相关文章