Python中pymysql通过字典插入数据

引子：

今天面试，被大佬要求写一段实操代码，首先他在写字板上开头：

def insert(table,data):
    #这里你知道table和data，其中data是一个字典，写插入数据库的代码

想了一下，想到的一些方法被否定了，后来被允许查资料后找到方法一（参考：http://www.bubuko.com/infodetail-2788655.html和https://www.cnblogs.com/zengsf/p/9734476.html）：

import pymysql
def insert(table,data):
    #这里你知道table和data，其中data是一个字典，写插入数据库的代码
    data = {'age': '123', 'name': 'happy'}
    table = 'users'
    # 获取到一个以键且为逗号分隔的字符串，返回一个字符串
    keys = ','.join(data.keys())
    values = ','.join(['%s'] *len(data))
    sql = 'INSERT INTO {table}({keys}) VALUES({values})'.format(table=table, keys=keys, values=values)
    # 这里的第二个参数传入的要是一个元组
    if cursor.execute(sql, tuple(data.values())):
        print('users')
        db.commit()
db =  pymysql.connect(host='127.0.0.1', port=3306, user='root', password='root', db='test')
cursor = db.cursor()
insert("1","2")

勉强算过，不过按大佬的说法 cursor.execute（）这里可以直接用字典为参数，面试结束后又查了一下资料（https://www.cnblogs.com/zengsf/p/9734476.html）知道可以将res_sql转换后可以把字典当做参数，而且可以用cursor.executemany（）进行批量插入，代码如下：

import pymysql
def insert2(table,data):
    #这里你知道table和data，其中data是一个字典，写插入数据库的代码
    data = {"name": 'happy222222', 'age': '123'}
    cols = ", ".join('`{}`'.format(k) for k in data.keys())
    print(cols)  # '`name`, `age`'

    val_cols = ', '.join('%({})s'.format(k) for k in data.keys())
    print(val_cols)  # '%(name)s, %(age)s'

    sql = "insert into users(%s) values(%s)"
    res_sql = sql % (cols, val_cols)
    print(res_sql)  # 'insert into users(`name`, `age`) values(%(name)s, %(age)s)'

    cursor.execute(res_sql, data)  # 将字典data传入
    db.commit()

    a = [{'age': '123', 'name': 'fake'}, {'age': 456, 'name': 'gigi'}, {'age': 23, 'name': 'qq'}]  # 拼出要插入的内容
    cursor.executemany(res_sql, a)# 将字典列表a传入
    db.commit()
db =  pymysql.connect(host='127.0.0.1', port=3306, user='root', password='root', db='test')
cursor = db.cursor()
insert2("1","2")

这样写法确实非常简洁，而且为接口的编写也带来了轻便、省事，不知道多进程/线程时运行效率的提升如何，以后有机会再测试吧

 

技术研究。侵告删