讀取excel數據存儲到mysql並進行重命名

import pandas as pd
import pymysql
import os
conn = pymysql.connect(host='......',
                       user='......',
                       password='.......',
                       db='......',
                       charset='utf8')
cursor = conn.cursor()


def save_excel():
    print('開始查找Excel文件夾下面文件')
    folder_num = os.listdir(r'Excel\\')
    if folder_num is []:
        print('沒有存在xlsx文件')
        return
    for folder in folder_num:
        if '已存儲' in folder:
            print(folder, '文件不需要存儲')
            continue
        print('開始存儲', folder, '文件')
        path = os.getcwd() + '\\' + 'Excel\\' + folder
        da = pd.read_excel(path, encoding='ANSI', sheet_name='Sheet1')
        data = da.astype(object).where(pd.notnull(da), None)
        for dm,category_id,ch_name,en_name in zip(data['dm'],data['category_id'],data['ch_name'],data['en_name']):
            print(dm,category_id,ch_name,en_name)
            sql = """insert into `pd_period` (dm,category_id,ch_name,en_name) values (%s,%s,%s,%s)"""
            try:
                cursor.execute(sql,(dm,category_id,ch_name,en_name))
                conn.commit()
            except Exception as e:
                print(e)
        print(folder, '文件存儲完成')
        os.rename(path, path.split('.')[0] + '已存儲.xls')
    print('全部文件存儲完成')


save_excel()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章