在 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

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

更新了 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.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'}