需求:将Python字典类型的数据存储在mysql数据库中
思路分析:
- 数据库中表的字段设计成“键”,目的是方便SQL插入语句的构造
- 定义函数构建插入的SQL语句
- pymysql连接数据库执行SQL语句插入
# 根据字典数据和插入表信息创建插入SQl语句
def insert_sql(res, table, uid):
# 列的字段
keys = ', '.join(res.keys())
# 行字段
values = list(res.values())
values = str(values).strip('[]')
sql = 'INSERT INTO {table}(主键, {keys}) VALUES ({uid},{values});'.format(uid=uid,table=table, keys=keys, values=values)
return sql
插入SQL语句的构成:
insert into table_name(uid, name, sex) values (1, '张三', '男');
问题1:如何优雅的构建插入SQL语句字段和值
前面这一段:uid, name, sex 这个可以处理成一个字符串格式,就是将字典中的所有键进行组合形成类似结构的字符串
# 列的字段
keys = ', '.join(res.keys())
# 执行完字符串格式: uid, name, sex
后面values那个字段涉及SQL语法问题:中文需要添加引号处理。如果写成如下
values =','.join(res.values) 结果是:1,张三,男 ;没有引号后续会报错
这里的处理思路是将所有的values值转化成列表,再转化成字符串再去除二边的 [ ]
# 行字段
values = list(res.values())
values = str(values).strip('[]')
[ 1, '张三', '男'] ---> 1, '张三', '男'
还有一种方法参考如下博客:Python优雅的将字典数据存入数据库中_亥虫的博客-CSDN博客_python 字典插入数据库
# 列的字段
keys = ', '.join(data.keys())
# 行字段
values = ', '.join(['%s']*len(data))
sql = 'INSERT INTO {table}({keys}) VALUES ({values})'.format(table=table, keys=keys, values=values)
# 将字段的value转化为元祖存入
cursor.execute(sql, tuple(data.values()))
db.commit()
问题2:连接mysql数据库执行SQL命令
插入的SQL语句构建完成,下面就是使用pymysql连接数据库,获取数据库对象,游标对象执行SQL语句。
参考之前写的博客:PyMySQL使用_小杜℃的博客-CSDN博客_pymysql 使用