唯一约束失败 - 如果不是唯一键，则为好消息

Question

假设我有一个可以添加如下值的数据库：

def information(cursor):
    a = input("Please define a 'a':")
    b = input("Please define a b:")
    c = input("Please define a c:")
    d = input("Please define a d:")
   
    statement=f'INSERT INTO data VALUES ({a}, "{b}", "{c}", "{d}");'
    cursor.execute(statement)
    cursor.close()

如果 Value 已经在表中，我如何才能收到一条好消息。

Answer 1

我必须说，我对 Python 并不完全熟悉，尤其是它的 SQLite 驱动程序是如何工作的。

但解决方案的要点是有条件地插入，然后获取插入了多少行。在 SQL 的许多实现中，我们可以执行 SELECT @@ROWCOUNT 之类的操作，但我认为这在 SQLite 中行不通。

但我认为这可行，这取决于插入失败的唯一键：

INSERT OR IGNORE INTO data VALUES ({a}, "{b}", "{c}", "{d}");

然后获取行数：

cursor.rowcount

所以我们可以使用该结果来检查发生了什么。如果为零，则插入失败。

顺便说一句，我相信您当前的实现对 SQL 注入攻击是开放的。您应该改为在准备好的查询中使用参数。