将 python 列表插入 Postgres 数据库

Question

我在使用 psycopg 格式化插入列表时遇到问题。 这是我正在尝试执行的代码示例。 基本上我只是从一个表中读取数据并尝试将其插入到另一个表中。

代码：

cur.execute("""select data from \"Table1\" where lat=-20.004189 and lon=-63.848004""")
rows = cur.fetchall()
print rows
cur.execute("""INSERT INTO \"%s\" (data) VALUES (ARRAY%s)""" % (args.tableName,rows)))

第一个选择查询返回的结果如下：

[([6193, 3975, 4960, 5286, 3380, 970, 3328, 3173, 2897, 2457, 2443, 2674, 2172, 2740, 3738, 4907, 3691, 4234, 3651, 3215],)]

当我尝试将其插入到另一个表中时，出现以下格式错误。

   cur.execute(cur.mogrify("""INSERT INTO \"%s\" (data) VALUES (%s)""" % (args.tableName,rows)))
psycopg2.ProgrammingError: syntax error at or near "["
LINE 1: INSERT INTO "DUMMY1km" (data) VALUES ([([6193, 3975, 4960, 5...

我尝试了 cur.mogrify，但它似乎没有帮助。

如果有人可以解决此问题，请告诉我。

谢谢 阿迪

Answer 1

我认为这里不需要 mogrify 。使用 executemany 并传递 rows 作为第二个参数。

cur.executemany(
    """INSERT INTO "%s" (data) VALUES (%%s)""" % (args.tableName),rows)

使用参数化参数有助于防止SQL 注入。

表名无法参数化，因此我们必须使用字符串插值将表名放入 SQL 查询中。 %%s gets 转义百分号，字符串插值后变为 %s。

---

顺便说一句，（正如 a_horse_with_no_name 已经指出的）您可以使用 INSERT INTO ... SELECT form of INSERT 将两个 SQL 查询作为一个执行：

cur.execute(
    """INSERT INTO %s (data)
       SELECT data FROM Table1
       WHERE lat=-20.004189 AND lon=-63.848004""" % (args.tableName))

---

根据注释中的问题，如果有多个字段，则SQL 变为：（

cur.executemany(
    """INSERT INTO {t} (lat,lon,data1,data2) 
       VALUES (%s,%s,%s,%s)""".format(t=args.tableName),rows)

如果使用 format 方法，则不必转义所有其他 %ss。）