Commit 8ae7c60c authored by 邱阿朋's avatar 邱阿朋

refactor(src): 修改文件路径和命名规则

- 移除 country 参数,统一文件保存路径
- 更改 spa_search 模块中的文件名和路径
- 优化代码结构,注释掉不必要的逻辑
parent 11c1d255
No preview for this file type
No preview for this file type
No preview for this file type
......@@ -43,7 +43,7 @@ def export_list_read_data():
def export_item_read_data(return_id):
file_name = f"{country}_return_goods\\{return_id}.xls"
file_name = f"return_goods\\{return_id}.xls"
if not os.path.isfile(file_name):
while True:
try:
......
......@@ -39,7 +39,7 @@ def page_get(url):
def export_list_read_data():
file_name = "spa-de.xls"
file_name = "ContraCogsInvoices.xls"
if not os.path.isfile(file_name):
raise FileNotFoundError(f"{file_name},文件不存在")
return pd.read_excel(file_name)
......@@ -119,7 +119,7 @@ def get_report_agreement_text(invoice_id):
def export_item_read_data(invoice_id):
file_name = f"{country}_spa\\{invoice_id}.xlsx"
file_name = f"spa\\{invoice_id}.xlsx"
if os.path.isfile(file_name):
df = pd.read_excel(file_name, sheet_name=None)
return df
......@@ -130,8 +130,9 @@ def export_item_read_data(invoice_id):
# 提取所有链接
links = tree.xpath('//table[@id="backup-report-table"]//a/@href')
if len(links) == 0:
data_list = get_report_agreement_text(invoice_id)
return {"Accrual For Current Period": pd.DataFrame(data_list)}
# data_list = get_report_agreement_text(invoice_id)
# return {"Accrual For Current Period": pd.DataFrame(data_list)}
return None
for link in links:
# 解析链接中的查询参数
......@@ -198,22 +199,7 @@ def main():
# 获取当前发票的 item 列表
item_dict = export_item_read_data(invoice_id)
if item_dict is None:
push_data = {
'ad_date': coop.get('Invoice date', ""), # spa费用数据日期
'erp_sku': "", # ERP SKU
'ad_amount': coop.get('Original balance', ""), # spa费用金额
'ad_amount_currency': "", # spa费用币制
'funding_type': coop.get('Funding Type', ""), # 资金类型
'transaction_type': "", # 交易类型
'group_name': "", # 组别 运营一组 运营二组
'group_code': "", # 组别 T1 T2
'asin': "", # ASIN
'shop_code': shop_code, # 店铺code
'type': 2, # 1 sheet1 2 其他sheet
'parent_id': "Sheet1",
'order_no': "", # 订单号
}
sheet_data.append(push_data)
sheet_data.append(coop)
log.warning(f"{invoice_id} 暂无报告信息")
continue
......@@ -415,6 +401,8 @@ if __name__ == '__main__':
try:
country = helper.get_input_with_default("国家(目前支持[DE,FR,JP,CA,UK,US])", "US")
shop_code = helper.get_input_with_default("店铺编码(DE-VC,FR-VC,JP-VC,CA-VC,UK-VC,VECELO])", "VECELO")
# test("ContraCogsInvoices.xls","2025-02-27-18-08_US_spa.xlsx")
# push_data_queue("2025-02-28-09-13_US_spa.xlsx")
domain.domain_page(page, country)
main()
page.close()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment