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)