詳解PyMySQL插入字典類型的數(shù)據(jù)

更新時(shí)間：2022年07月18日 14:55:59 作者：藍(lán)莓薄荷

在我們爬蟲或者調(diào)API獲取數(shù)據(jù)后，需要將數(shù)據(jù)存入到數(shù)據(jù)庫中，如果數(shù)據(jù)是列表嵌套字典格式的話，一般做法是遍歷列表，然后用字典生成對(duì)應(yīng)的SQL語句再執(zhí)行插入到表中，本文將介紹一種更加簡便的方法來插入字典類型的數(shù)據(jù),需要的朋友可以參考下

在我們爬蟲或者調(diào)API獲取數(shù)據(jù)后，需要將數(shù)據(jù)存入到數(shù)據(jù)庫中，如果數(shù)據(jù)是列表嵌套字典格式的話，一般做法是遍歷列表，然后用字典生成對(duì)應(yīng)的SQL語句再執(zhí)行插入到表中。本文將介紹一種更加簡便的方法來插入字典類型的數(shù)據(jù)。

我們有如下數(shù)據(jù)列表，需要將該數(shù)據(jù)插入到對(duì)應(yīng)的數(shù)據(jù)表student中（表的字段名和數(shù)據(jù)列表中字典的鍵名相同）

student_list = [
    {"name": "Tony", "age": 19, "sex": "male"},
    {"name": "Lisa", "age": 18, "sex": "female"},
    {"name": "Jack", "age": 20, "sex": "male"}
]

使用pymysql批量插入數(shù)據(jù)可以用以下方法

cursor.executemany(sql, data_list)

上面的sql是一條包含占位符的插入語句，data_list就是需要插入的數(shù)據(jù)

可以通過以下方法來生成對(duì)應(yīng)的sql

def generate_sql(data_list):
    data = data_list[0]  # 取出第一條
    cols = ", ".join('`{}`'.format(k) for k in data.keys())  # 字段名拼接
    val_cols = ', '.join('%({})s'.format(k) for k in data.keys())  # 對(duì)應(yīng)值的占位符拼接
    sql = """
    INSERT INTO student(%s) VALUES(%s)
    """ % (cols, val_cols)
    return sql

生成的sql如下：

INSERT INTO student(`name`, `age`, `sex`) VALUES(%(name)s, %(age)s, %(sex)s)

生成sql后就可以使用executemany()方法批量插入數(shù)據(jù)到數(shù)據(jù)庫中，全部代碼如下

import pymysql
MYSQL_CONFIG = {
    'host': 'localhost',  # IP地址
    'port': 3306,  # 端口
    'user': 'myuser',  # 用戶名
    'passwd': '111111',  # 密碼
    'db': 'myclass',  # 數(shù)據(jù)庫
    'charset': 'utf8',  # 編碼
}
def generate_sql(data_list):
    data = data_list[0]
    cols = ", ".join('`{}`'.format(k) for k in data.keys())
    val_cols = ', '.join('%({})s'.format(k) for k in data.keys())
    sql = """
    INSERT INTO student(%s) VALUES(%s)
    """ % (cols, val_cols)
    return sql
def main():
    student_list = [
        {'name': 'Tony', 'age': 19, 'sex': 'male'},
        {'name': 'Lisa', 'age': 18, 'sex': 'female'},
        {'name': 'Jack', 'age': 20, 'sex': 'male'}
    ]
    conn = pymysql.connect(**MYSQL_CONFIG)  # 數(shù)據(jù)庫連接
    cur = conn.cursor()  # 游標(biāo)對(duì)象
    sql = generate_sql(student_list)
    cur.executemany(sql, student_list)
    conn.commit()
    cur.close()
    conn.close()
if __name__ == '__main__':
    main()

執(zhí)行以上代碼，就可以看到數(shù)據(jù)成功插入到表中