在SQLAlChemy中以字典形式检索查询结果
我使用的是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.execute
,fetchone
将返回SQLAlChemyRow
对象,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
如果需要Truedict
对象,则只需将其转换即可:
my_dict = dict(result)
print(my_dict) # {'FirstName': 'Gord', 'LastName': 'Thompson'}
相关文章