文件夹下csv数据写入mysql数据库

import os
import pandas as pd
import csv
import pymysql


def save_db(row):
    db = pymysql.connect(host='127.0.0.1', user='shijuan', password='123456', database='shijuan',
                         charset="utf8")
    cursor = db.cursor()
    insert_sql = 'insert into tp_keyword(keyword) values(%s)'
    try:
        # cursor.execute(insert_sql)
        cursor.executemany(insert_sql, row)
        db.commit()
    except:
        # 如果发生错误则回滚
        db.rollback()
        cursor.close()
        # 关闭数据库连接
        db.close()


path = 'E:\网站\关键词\\12'
files = os.listdir(path)
for file in files:
    url=path + '\\' + file
    df = pd.read_csv(url,low_memory=False,quoting=csv.QUOTE_NONE)
    result = df.values.tolist()
    # print(result)
    save_db(result)

 

类似文章