13

在 pysqlite 中,违反 aNOT NULLUNIQUE约束同样会引发 IntegrityError。不幸的是,这种异常类型不提供错误代码,而只提供一条消息。

所以,假设我想忽略唯一约束违规,因为我知道这在给定数据上是安全的,但应该报告键列中的 Null 值。

我想出了以下解决方案:

con = sqlite3.connect(':MEMORY:')
con.execute('''CREATE TABLE ABCD (A TEXT NOT NULL,
                                  B TEXT NOT NULL,
                                  C TEXT NOT NULL,
                                  D TEXT NOT NULL,
                                  PRIMARY KEY (A, B))''')
with con:
    for a, b, c, d in inputs:
        try:
            con.execute('INSERT INTO ABCD VALUES (?, ?, ?, ?)',
                        (a, b, c, d))
        except sqlite3.IntegrityError as e:
            # Skip 'not unique' errors, but raise others.
            if not e.message.endswith('not unique'):
                raise
con.close()

但是,解析错误消息似乎是错误的,并且可能不可靠。有没有更好的方法来做到这一点,甚至可以使用con.executemany()

4

3 回答 3

3

这就是我最终做的事情:

con = sqlite3.connect(':MEMORY:')
con.execute('''CREATE TABLE ABCD (A TEXT NOT NULL,
                                  B TEXT NOT NULL,
                                  C TEXT NOT NULL,
                                  D TEXT NOT NULL,
                                  PRIMARY KEY (A, B))''')
with con:
    for a, b, c, d in inputs:
        if any(elem is None for elem in (a, b, c, d)):
            raise ValueError('Fields must not be empty.')
        con.execute('INSERT OR IGNORE INTO ABCD VALUES (?, ?, ?, ?)',
                    (a, b, c, d))
con.close()

像这样,在执行 DB 操作之前“手动”捕获空值。如果期间发生任何错误execute(例如违反UNIQUE约束),则跳过该条目。请注意,INSERT OR IGNORE这并不意味着忽略唯一性约束,而是忽略(即跳过)输入行。

此解决方案的缺点是检查空值会进行两次。我想这并不算太糟糕,因为它可能是相当便宜的手术。我认为,它仍然比解析错误消息更干净,并且可能对更改更健壮(例如 pysqlite 更新,这可能会更改错误消息中的某些细节)。

致谢:这个想法来自与 Lutz 的讨论。它也是由 Martijn 独立提出的。

于 2014-06-26T08:04:08.340 回答
1

更优雅的解决方案是完全依赖 SQL(ite) 功能。通过为主键 ( ) 指定一个冲突子句,ON CONFLICT IGNORE已经实现了所需的行为:

con = sqlite3.connect(':memory:')
con.execute('''CREATE TABLE ABCD (A TEXT NOT NULL,
                                  B TEXT NOT NULL,
                                  C TEXT NOT NULL,
                                  D TEXT NOT NULL,
                                  PRIMARY KEY (A, B) ON CONFLICT IGNORE)''')

因此,重复的行(违反主键的唯一性约束)会被静默地跳过,而 Null 值会导致中止(导致sqlite3异常)。这一切都是在没有预先过滤 Null/None 值的数据或摆弄sqlite3API 的错误消息的情况下实现的。我们现在可以简单地调用con.executemany(),不用多说:

with con:
    con.executemany('INSERT INTO ABCD VALUES (?, ?, ?, ?)', inputs)
于 2015-08-08T20:17:53.337 回答
-1

下面是一个工作代码:

import sqlite3

con = sqlite3.connect(':memory:')
con.execute('''CREATE TABLE ABCD (A TEXT NOT NULL,
                                  B TEXT NOT NULL,
                                  C TEXT NOT NULL,
                                  D TEXT NOT NULL,
                                  PRIMARY KEY (A, B));''')

inputs = [('cow', 'pig', 'cat', 'dog'), ('cow', 'pig', 'quail', 'turkey')]
with con:
    for a, b, c, d in inputs:
        try:
            con.execute('INSERT INTO ABCD VALUES (?, ?, ?, ?);',
                        (a, b, c, d))
        except sqlite3.IntegrityError as e:
            if 'not null' in e.args[0].lower():
                print('There is a NULL value')
            elif 'unique constraint' in e.args[0].lower():
                print('There is unique violation')
            else:
                raise

测试:

>>> 
There is a NULL value
>>> 

第二次测试结果:

>>> 
There is unique violation
>>> 

希望,能帮到你。

于 2014-06-22T08:07:49.693 回答