在循环中执行 SQL 的 Python 脚本

Question

我正在尝试编写调用 mysql SQL 的 python 脚本。

我依靠第一个查询结果来运行第二个查询。问题是它抛出错误。

错误:cursor2.execute(field_sql,id)

import boto3
import importlib
import psutil
import pymysql
import pymysql.cursors
import subprocess
import sys

rdsConn = pymysql.connect(host = 'XXXX'),
                          db = 'XXXX',
                          user = 'XXXX',
                          password = 'XXXX',
                          charset = 'utf8mb4',
                          cursorclass=pymysql.cursors.DictCursor)

cursor1 = rdsConn.cursor()
cursor2 = rdsConn.cursor()


name = 'Test'
sql = "select id from Table1 where name = %s"
cursor1.execute(sql,name)
result = cursor1.fetchall()
for id in result:
    field_sql= "select columnname from  Table2 where id = %s"
    cursor2.execute(field_sql,id)
    fieldresult = cursor2.fetchall()
    for fieldrow in fieldresult:
        print(fieldrow)

cursor1.close()
cursor2.close()
rdsConn.close()

Answer 1

您的查询使用字典游标，因此它将返回字典列表，例如:

[{'id': 1}, {'id': '2'}, ...]

这意味着您的 id* 将是一个字典，而不是一个元组。这意味着您将参数作为字典传递给第二个查询。如果这样做，则需要使用 pyformat 样式来使用命名参数:

for rowdict in result:
    field_sql = "select columnname from  Table2 where id = %(id)s"
    cursor2.execute(field_sql, rowdict)
    fieldresult = cursor2.fetchall()
    for fieldrow in fieldresult:
        print(fieldrow)

您会看到打印的 fieldrow 也是 dict。

此外，查询参数应作为字典(命名参数)或元组(位置参数)传递。 pymysql 接受 cursor.execute(sql, "name") 形式，其他 dbapi2 连接器不接受。规范形式为 cursor.execute(sql, ("name",))。

*btw，您不应该使用 id 作为名称，它隐藏了内置的 id 函数