Python SQLite 插入动态列或使用 DEFAULT 值

Question

我正在使用flask-restful，用户可以发布一些必需的和一些可选的数据，我想用这些数据创建一个SQLite 插入语句。问题是我不想通过使用变量作为列名并检查哪个参数是None 以及哪个有一些数据来破坏 sql 语句。我希望我可以只使用所有列，当 arg2 或 arg3 为None 时，数据库将使用列默认值。可悲的是，我收到了NOT NULL constraint failed: mytable.col1 的消息。使用我的方法从表创建中删除 NOT NULL 只会在使用我的语句时插入 NULL。

是否有明确的方式告诉 SQLite 使用其默认值？

如果没有，是否有更优雅的方式来获取默认值而不是执行某些表模式查询？

如果没有并且不存在其他解决方案，那么屠宰 sql 语句的最有效方法是什么？

parser_post = reqparse.RequestParser()
parser_post.add_argument('arg1', type=str, required=True)
parser_post.add_argument('arg2', type=str)
parser_post.add_argument('arg3', type=str)
args = parser_post.parse_args()

# ... db init code

cur.execute("insert into mytable (col1, col2, col3) values (?, ?, ?)", 
(args['arg1'], args['arg2'], args['arg3']))

表是这样创建的：

sql = """CREATE TABLE mytable (
    PK_id       INTEGER PRIMARY KEY AUTOINCREMENT,
    col1        INTEGER NOT NULL,
    col2        TEXT DEFAULT "global" NOT NULL,
    col3        TEXT DEFAULT "global" NOT NULL)"""
cursor.execute(sql)

Answer 1

使用数据库端默认值的一种方法是不为插入中的特定列提供数据。

在这种情况下，服务器将为该列插入默认值。如果您没有在 CREATE TABLE（或更高版本的 ALTER TABLEs）中指定默认值，则默认值为 NULL（对应于 Python None）。

然而，由于 col2 确实有一个数据库指定的默认值，通过col2 TEXT DEFAULT "global" NOT NULL，那么在插入中不指定col2 将导致sqlite 使用"global"。同上col3。

您可以动态构建查询，以排除带有None 的列，只要您小心处理查询字符串中的内容即可。

def prep_qry(args, colnames):
    """this query is secure as long as `colnames` contains trusted data
    standard parametrized query mechanism secures `args`"""

    binds,use = [],[]

    for colname, value in zip(colnames,args):
        if value is not None:
            use.extend([colname,","])
            binds.extend(["?",","])


    parts = ["insert into mytable ("]
    use = use[:-1]
    binds = binds[:-1]
    
    parts.extend(use)
    parts.append(") values(")
    parts.extend(binds)
    parts.append(")")

    qry = " ".join(parts)

    return qry, tuple([v for v in args if not v is None])

print(prep_qry([1,None,3], ["col1", "col2", "col3"]))
print(prep_qry([1,2,3], ["col1", "col2", "col3"]))

输出：

('insert into mytable ( col1 , col3 ) values( ? , ? )', (1, 3))
('insert into mytable ( col1 , col2 , col3 ) values( ? , ? , ? )', (1, 2, 3))