8

我有我自己的个人数据库，我是为了好玩而创建的（所以我不关心 sql 注入作为我自己创建的私有数据库），并且正在尝试更改我创建的使用字符串格式 (.format()) 和占位符 (? , %s 等）并使用 f 字符串代替。我遇到了一个问题，在我将 sqlite3 查询更改为 f 字符串后，我的一个将指定列更新为指定行的函数将无法运行。

这是我当前使用 f 字符串的函数：

import sqlite3
from tabulate import tabulate
conn = sqlite3.connect("Table.db")
c = conn.cursor()

def updatedb(Column, Info, IdNum):  
    with conn:
        data = c.execute(f"UPDATE Table_name SET {Column} = {Info} WHERE IdNum={IdNum}")
        c.execute(f"SELECT * FROM Table_name WHERE IdNum = {IdNum}")
    print(tabulate(data, headers="keys", tablefmt="grid", stralign='center', numalign='center'))

该函数通过使用您想要在该列中的新信息更新指定行的指定列来更新表。例如，在 3 x 3 表中，如果该列是年龄或其他内容，我可以使用该函数将第 1 行第 2 列更新为 18，而不是第 1 行第 2 列。之后的 select 查询只是选择已更新的特定行，之后的 print 语句使用 tabulate 包打印出整洁有序的表格。

每当我尝试使用此功能时，我得到的错误是：

sqlite3.OperationalError: no such column: Info

无论我在函数中为 Info 变量输入什么，错误都会变成什么，但我不知道如何解决这个问题。

这是我在尝试更改为 f 字符串之前的更新语句，对我来说效果很好：

data = c.execute("UPDATE Table_name SET {} = ? WHERE IdNum=?".format(Column), (Info, IdNum))

将上述查询更改为 af 字符串似乎并没有那么大的变化，但它不起作用，所以任何反馈都将不胜感激。

4

4 回答 4

11

这里的问题是理解参数化 - 它仅适用于参数，不适用于列名和其他内容。

在这个例子中：

 query = 'SELECT * FROM foo WHERE bar = ? AND baz = ?'
 params = (a, b)
 cursor.execute(query, params)

请注意，查询和数据是分别传递给的.execute——进行插值是数据库的工作——这使您摆脱了引用地狱，并通过禁用任何类型的 sql 注入使您的程序更安全。它还可以表现得更好——它允许数据库缓存已编译的查询并在您更改参数时使用它。

现在这只适用于data。如果您想在变量中包含实际的列名，您必须自己在查询中插入它：

 col1 = 'bar'
 col2 = 'baz'
 query = f'SELECT * FROM foo WHERE {col1} = ? AND {col2} = ?'
 cursor.execute(query, params)
于 2018-07-12T19:24:54.750 回答
2

参数化和字符串替换是两个非常不同的东西。

在最基本的层面上，您的问题是字符串替换正在生成以下形式的命令：

 . . . SET some_column = some_info . . .

SQL 需要的地方

 . . . SET some_column = 'some_info' . . .

（即字符串值的引号）。SQLite 将不带引号的字符串解释为由列名组成的表达式。

正如您可能认为的那样，您无法通过简单地将单引号添加到格式字符串中来解决此问题。如果替换值本身包含单引号会发生什么？这种方式既存在字符串转义的疯狂，也存在 SQL 注入的危险。

相反，只需使用参数化。

但是，请注意，您不能参数化列名 ( {Column})，您必须为此使用字符串替换，同时注意不要让注入发生。

于 2018-07-12T19:34:32.017 回答
0

为了改进上面关于使用f-strings的nosklo答案，您只需在变量周围使用转义引号，以便将它们作为字符串发送到 SQLite：

import sqlite3

conn = sqlite3.connect("somedatabase.db")
c = conn.cursor()

def updatedb(colname, colvalue, rowid):
    with conn:
        c.execute(f"UPDATE tablename SET \"{colname}\" = \"{colvalue}\" WHERE rowidentifier = \"{rowid}\"")

笔记

出于安全原因，请勿对依赖外部用户输入的数据库使用字符串插值。但是，如果你清理输入字符串，我想你可以让它工作。

于 2020-09-17T22:37:47.823 回答
0

我正在使用mysql.connector

query = "INSERT INTO PICS (name,image) VALUES (%s,%s)"
param = (fname, DATA)
cursor.execute(query, param)
print("DONEEEEEE" * 100)
conn.commit()
cursor.close()
conn.close()

这是2021 年最新版本的“8.0.27”

这对我有用！！！！

于 2021-11-05T07:54:21.723 回答