如何使用 pymysql 从多个选择语句中获取结果

Question

基本上，我有以下示例（假设 cur 来自有效连接）：

>>> con =  pymysql.connect(<parameters go here>)
>>> cur = con.cursor()
>>> sql = "SELECT @a := 0; SELECT @a := @a+2; SELECT @a;"
>>> res = cur.execute(sql)
>>> res
1

如您所见，res 返回整数 1，这意味着 sql 运行良好。但是，最后一个选择应该返回数字 2，我需要那个数字。

如果我运行这段代码（由@falsetru 建议），我也得不到我需要的东西：

>>> cur.execute(sql)
1
>>> cur.fetchall()
[{u'@a := 0': 0}]

我怎样才能找回它？是否可以不分离 SQL 语句？

Answer 1

使用Cursor.fetchone 获取单行，或使用Cursor.fetchmany/Cursor.fetchall 获取多个或所有结果行。

row = cur.fetchone()
a = row[0]

---

UPDATE Cursor.execute 执行单个 sql 语句，因此单独执行语句（通过将 sql 拆分为 ;）

import pymysql
con =  pymysql.connect(user='root', password='root')
cur = con.cursor()
sql = "SELECT @a := 0; SELECT @a := @a+2; SELECT @a;"
for stmt in sql.split(';'):
    if stmt.strip():
        cur.execute(stmt)
row = cur.fetchone()
a = row[0]
print(a)  # => 2