来自Python的MySQL使用MySQLdb

时间:2017-02-23 11:49:03

标签: python mysql python-2.7 mysql-python

我正在尝试使用MySQLdb python包向MySQL数据库添加值。我的一些列不能为空,我在努力找到一种方法来检查在尝试向表中添加数据之前必须存在哪些列值。 目前我不得不解析一个文件,提取一些键值对,找出应该添加到哪些表中,然后使用以下方法添加它们:

cursor.execute("""INSERT INTO %s (%s) VALUES (\"%s\")""" %(tbl, cols, vals)

我可以确保以正确的顺序添加列,并且我可以获取自动增量ID以链接不同的表。我想要做的是为每个表,检查它是否包含一个不可为空的列,然后,如果有必要,相应地修改输入值,然后再尝试将值添加到表中。关于我如何做到这一点的任何建议/更好的想法?

1 个答案:

答案 0 :(得分:1)

您可以在MySQL中测试INFORMATION_SCHEMA.COLUMNS表,以确定列是否接受NULL值。

SELECT `IS_NULLABLE`
FROM INFORMATION_SCHEMA.`COLUMNS`
WHERE `TABLE_SCHEMA` = 'SchemaName'
AND `TABLE_NAME` = tbl
AND `COLUMN_NAME` = col
; # returns YES/NO

您应该改进cursor.execute(stmt, [params])语法的语法来传递参数,而不是在文本sql中对其进行硬编码。这将更加可靠:

sql = "INSERT INTO %s (%s)" % (tbl, col)
sql += " VALUES(%s)"
cursor.execute(sql, [val])
connection.commit()

MySQL Docs for cursor.execute