payment 优化

src/payment.py

@@ -79,7 +79,7 @@ def export_list_read_data():
     return merged_df
 
 
-def export_details_read_data(invoice_number):
+def export_details_read_data(origin_invoice_number, invoice_number):
     params = {
         "invoiceNumber": invoice_number,
         "payeeCode": payeeCode,
@@ -91,17 +91,20 @@ def export_details_read_data(invoice_number):
     page_get(full_url)
 
     # # 读取详情内容
-    file_name = f"payment\\{invoice_number}.csv"
-    # if os.path.isfile(file_name):
-    #     return pd.read_csv(file_name)
+    file_name = f"payment\\{origin_invoice_number}.csv"
+    if os.path.isfile(file_name):
+        return pd.read_csv(file_name)
 
+    count = 0
     while True:
         try:
-            page.ele("#line-items-export-to-spreadsheet-announce", timeout=5).click.to_download(rename=file_name)
+            page.ele("#line-items-export-to-spreadsheet-announce", timeout=3).click.to_download(rename=file_name)
             file.wait_for_downloads(file_name)
             excel.remove_last_comma(file_name)
             break
         except ElementNotFoundError:
+            if count == 3: return None
+            count += 1
             log.warning("导出按钮不存在刷新网页")
             page.refresh()
 
@@ -260,10 +263,12 @@ def main():
     i = 0
     for _, data in list_data.iterrows():
         i += 1
-        invoice_number = data.get("Invoice Number")
+        origin_invoice_number = data.get("Invoice Number")
 
+        invoice_number = origin_invoice_number
         # 获取当前订单的Payee和优惠比例
         vendor_payment_terms = get_po_code(i, invoice_number[:8])
+        vendor_payment_terms['invoice_number'] = origin_invoice_number
         log.info(vendor_payment_terms)
 
         vendor = vendor_payment_terms['vendor']
@@ -275,11 +280,14 @@ def main():
             # 检查后两位是否在测试列表中
             if last_two in ["PC", "MA", "SC"]:
                 invoice_number = invoice_number[:-2]  # 去掉后两位
-            if last_three == "PCR":
+            if last_three in ["PCR", "+SC", "SC-"]:
                 invoice_number = invoice_number[:-3]  # 去掉最后三位
 
         # 下载excel文件并读取数据
-        detail_data = export_details_read_data(invoice_number)
+        detail_data = export_details_read_data(origin_invoice_number, invoice_number)
+        if detail_data is None:
+            log.error("数据存在问题,请手动处理")
+            continue
 
         # 获取争议数据
         all_price_data = click_get_price_data()