diff --git a/main.py b/main.py index ea9d7b3..feb5cd4 100644 --- a/main.py +++ b/main.py @@ -174,21 +174,21 @@ class MyPlugin(Star): cur.close() self.sqllitaa.close() # 删除源文件 - file_list = os.listdir(f"{self.data_path}") - try: - for nam in file_list: - if "zhangdan" not in nam: - os.remove(str(self.data_path)+"\\"+nam) - file_listaa = os.listdir(f"{self.data_path}/zhangdan/") - for namaa in file_listaa: - os.remove(str(self.data_path)+"\\zhangdan\\"+namaa) - except Exception as e: - for nam in file_list: - if "zhangdan" not in nam: - os.remove(str(self.data_path)+"/"+nam) - file_listaa = os.listdir(f"{self.data_path}/zhangdan/") - for namaa in file_listaa: - os.remove(str(self.data_path)+"/zhangdan/"+namaa) + # file_list = os.listdir(f"{self.data_path}") + # try: + # for nam in file_list: + # if "zhangdan" not in nam: + # os.remove(str(self.data_path)+"\\"+nam) + # file_listaa = os.listdir(f"{self.data_path}/zhangdan/") + # for namaa in file_listaa: + # os.remove(str(self.data_path)+"\\zhangdan\\"+namaa) + # except Exception as e: + # for nam in file_list: + # if "zhangdan" not in nam: + # os.remove(str(self.data_path)+"/"+nam) + # file_listaa = os.listdir(f"{self.data_path}/zhangdan/") + # for namaa in file_listaa: + # os.remove(str(self.data_path)+"/zhangdan/"+namaa) yield event.plain_result("保存完成__使用/chazhang 查询最后10条记录") except Exception as e: @@ -225,11 +225,13 @@ class MyPlugin(Star): if ".csv" in filename: ata = pd.read_csv(filename, encoding='gb18030', on_bad_lines='skip',header=22) # 读取表格数据 - for biao in range(10): + for biao in range(len(ata)-22): # 暂存当行数据 shuju = [] for aac in ata.loc[biao]: shuju.append(str(aac)) + if biao == 0: + logger.info(f"读取到数据:{shuju}") # id:流水号,payid:账单号,paytype:交易类型,paydate:交易时间,payname:交易名称,payto:收入/支出,paysize:交易金额 # paystu:交易状态,paybox:收支方式,payqd:记录渠道 # 插入数据 @@ -237,11 +239,13 @@ class MyPlugin(Star): elif ".xls" in filename: ata = pd.read_excel(filename,header=17) - for biao in range(10): + for biao in range(len(ata)-17): # 暂存当行数据 shuju = [] for aac in ata.loc[biao]: shuju.append(str(aac)) + if biao == 0: + logger.info(f"读取到数据:{shuju}") # id:流水号,payid:账单号,paytype:交易类型,paydate:交易时间,payname:交易名称,payto:收入/支出,paysize:交易金额 # paystu:交易状态,paybox:收支方式,payqd:记录渠道 # 插入数据