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

Question

我正在使用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´´´。

Answer 1

results 是一个元组，我希望它是 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 将返回 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'}

Answer 2

您可以使用 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)

Answer 3

如果raw_connection() 正在返回一个PyMySQL Connection 对象，那么您可以继续使用DictCursor，如下所示：

engine = create_engine("mysql+pymysql://root:whatever@localhost:3307/mydb")
connection = engine.raw_connection()
cursor = connection.cursor(pymysql.cursors.DictCursor)
cursor.execute("SELECT 1 AS foo, 'two' AS bar")
result = cursor.fetchall()
print(result)  # [{'foo': 1, 'bar': 'two'}]

Answer 4

我个人会使用 pandas：

import pandas as pd
connection = engine.raw_connection()
df = pd.read_sql_query('SELECT * from User where id=0' , connection)
mydict = df.to_dict()