Python pymysql 将变量值动态绑定到数据库字段

Question

我正在尝试动态绑定要插入到数据库表列中的变量值。

json 中的示例变量值：

document= {'zipCode': '99999', 
           'name': 'tester', 
           'company': 'xxxx'}

而我的数据库表列为：

表名：table1

列：id、邮编、姓名、公司

我在 python 中的代码：

with connection.cursor() as cursor:
     sql = "INSERT INTO table1(zip_code, name, company) VALUES (%s,%s,%s)"
            cursor.execute(sql,(document['zipCode'],
                                document['name'],
                                document['company']))
connection.commit()

但是，如果文档中的某个键值不存在，则 INSERT 查询肯定会遇到错误。即仅文档['name'] 存在于文档变量中

有没有想过要处理这个以获得高效的代码？

Answer 1

通常情况下，像 SQLAlchemy 或 Peewee 这样的 ORM 可以很容易地为您解决这个问题。

但是，如果我要实现，我可能会根据可用键做一些“动态”的事情：

QUERY = "INSERT INTO table1({columns}) VALUES ({values})"


def get_query(document):
   columns = list(document.keys())

   return QUERY.format(columns=", ".join(columns),
                       values=", ".join('%({})s'.format(column) for column in columns))

示例用法：

In [12]: get_query({'zipCode': '99999', 'name': 'tester', 'company': 'xxxx'})
Out[12]: 'INSERT INTO table1(company, zipCode, name) VALUES (%(company)s, %(zipCode)s, %(name)s)'

In [13]: get_query({'name': 'tester'})
Out[13]: 'INSERT INTO table1(name) VALUES (%(name)s)'

然后，您只需使用 document 字典对查询进行参数化，因为我们在查询中创建了 命名占位符：

cursor.execute(get_query(document), document)