229e474b58
- main_router: add run_automation_test that runs each site's download flows in a cross sequence (expected/actual) plus a pass/fail report with timings; expose it as menu [8] - Site download functions now return False on their exception paths so the test harness can record failures (baishi also flags config-read failure) - Replace overblown log/comment phrasing across all modules with plain statements; trim docstrings Co-Authored-By: Claude <noreply@anthropic.com>
413 lines
18 KiB
Python
413 lines
18 KiB
Python
# main_router.py
|
||
|
||
import os
|
||
import time
|
||
import yaml
|
||
import pandas as pd
|
||
from playwright.sync_api import sync_playwright
|
||
|
||
from paths import DOWNLOAD_DIR, CONFIG_PATH
|
||
|
||
# 导入抽离出去的各个网点模块
|
||
import site_shunxin
|
||
import site_baishi
|
||
import site_zto
|
||
import site_yunda
|
||
|
||
# 定义网点及对应的初始登录 URL
|
||
SITES_CONFIG = {
|
||
"顺心": "https://sxne.sxjdfreight.com",
|
||
"百世": "https://v5.800best.com",
|
||
"中通": "https://ws.zto56.com/",
|
||
"韵达": "https://ky-sso.yunda56.com",
|
||
}
|
||
|
||
# ====================================================================
|
||
# 站点就绪特征：每个站点登录成功进入工作台后的标志性控件
|
||
# ====================================================================
|
||
READY_SELECTORS = {
|
||
"顺心": 'h1:has-text("盟商门户网")',
|
||
"百世": 'h1[title="百世快运"]',
|
||
"中通": '.logo:has-text("网点版")',
|
||
"韵达": '.el-menu-item:has-text("首页")',
|
||
}
|
||
|
||
|
||
def task_process_undelivered_data(site_name="顺心"):
|
||
"""应到未到比对：找出应到但未实到的运单 (按站点前缀)"""
|
||
print(f"\n▶ 开始执行【{site_name} - 应到未到数据处理】任务...")
|
||
|
||
download_dir = DOWNLOAD_DIR
|
||
expected_path = os.path.join(download_dir, f"{site_name}-应到货物数据.xlsx")
|
||
actual_path = os.path.join(download_dir, f"{site_name}-实到货物数据.xlsx")
|
||
output_path = os.path.join(download_dir, f"{site_name}-应到未到货物数据.xlsx")
|
||
|
||
if not os.path.exists(expected_path):
|
||
print(f"❌ 错误：找不到【{site_name}-应到货物数据】主文档：{expected_path}")
|
||
return
|
||
|
||
if not os.path.exists(actual_path):
|
||
print(f"❌ 错误：找不到【{site_name}-实到货物数据】主文档：{actual_path}")
|
||
return
|
||
|
||
try:
|
||
print(">> 正在载入本地 Excel 文档...")
|
||
# 全部按字符串读取并保留空串：避免 18 位运单号被当成浮点数丢精度，也避免空单元格变成 NaN。
|
||
df_expected = pd.read_excel(expected_path, dtype=str, keep_default_na=False)
|
||
df_actual = pd.read_excel(actual_path, dtype=str, keep_default_na=False)
|
||
|
||
if "运单号" not in df_expected.columns or "运单号" not in df_actual.columns:
|
||
print("❌ 校验失败：数据源缺少【运单号】字段，请检查导出配置。")
|
||
return
|
||
|
||
print(">> 正在比对应到与实到数据...")
|
||
|
||
# 统一运单号为去空白字符串，消除 int/float 与 str 混读导致 isin 永不命中的隐患
|
||
df_expected["运单号"] = df_expected["运单号"].astype(str).str.strip()
|
||
df_actual["运单号"] = df_actual["运单号"].astype(str).str.strip()
|
||
|
||
# 剔除空白运单号，避免空值被误判为“应到未到”
|
||
df_expected = df_expected[df_expected["运单号"] != ""]
|
||
actual_set = set(df_actual["运单号"]) - {""}
|
||
|
||
# Left Anti-Join：在应到中找出不存在于实到里的运单号
|
||
df_undelivered = df_expected[~df_expected["运单号"].isin(actual_set)]
|
||
|
||
target_columns = ["班次号", "交接单号", "运单号"]
|
||
available_columns = [
|
||
col for col in target_columns if col in df_undelivered.columns
|
||
]
|
||
df_output = df_undelivered[available_columns]
|
||
|
||
print(f">> 比对完成，共筛选出【应到未到】运单: {len(df_output)} 条。")
|
||
|
||
df_output.to_excel(output_path, index=False)
|
||
print(f"====================================================")
|
||
print(f" 比对完成，结果已输出。")
|
||
print(f" 📁 输出路径: {output_path}")
|
||
print(f"====================================================")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 比对过程中发生异常: {e}")
|
||
|
||
|
||
# ====================================================================
|
||
# 自动化测试入口
|
||
# ====================================================================
|
||
# 每个(非百世)站点的交叉测试序列：覆盖两种下载流程之间的全部 4 种相邻转换，
|
||
# 用于验证无论上一个流程把页面留在什么状态，下一个流程都能正常运行：
|
||
# 应到->实到、实到->应到、应到->应到、实到->实到
|
||
CROSS_TEST_SEQUENCE = ["expected", "actual", "expected", "expected", "actual", "actual"]
|
||
|
||
|
||
def run_automation_test(pages_map):
|
||
"""自动化测试入口：按交叉序列逐个跑通各站点的下载流程，结束后打印统计报告。
|
||
|
||
判定规则：流程函数返回 False 或抛出异常记为 FAIL，其余记为 PASS。
|
||
"""
|
||
# 站点 -> {流程键: (中文名, 流程函数)}；百世为单流程，单独处理
|
||
flow_table = {
|
||
"顺心": {
|
||
"expected": ("应到", site_shunxin.shunxin_expected_download),
|
||
"actual": ("实到", site_shunxin.shunxin_actual_download),
|
||
},
|
||
"中通": {
|
||
"expected": ("应到", site_zto.zto_expected_download),
|
||
"actual": ("实到", site_zto.zto_actual_download),
|
||
},
|
||
"韵达": {
|
||
"expected": ("应到", site_yunda.yunda_expected_download),
|
||
"actual": ("实到", site_yunda.yunda_actual_download),
|
||
},
|
||
}
|
||
|
||
# 构建测试计划：[(站点, 流程中文名, 流程函数), ...]
|
||
plan = []
|
||
for site_name, flows in flow_table.items():
|
||
if site_name not in pages_map:
|
||
continue
|
||
for flow_key in CROSS_TEST_SEQUENCE:
|
||
label, func = flows[flow_key]
|
||
plan.append((site_name, label, func))
|
||
# 百世：单流程，跑一次即可
|
||
if "百世" in pages_map:
|
||
plan.append(("百世", "应到未到", site_baishi.baishi_download_undelivered_data))
|
||
|
||
if not plan:
|
||
print("\n⚠️ 当前没有已就绪的站点，无法执行自动化测试。")
|
||
return
|
||
|
||
total = len(plan)
|
||
print("\n====================================================")
|
||
print(f"自动化测试开始，共 {total} 个步骤。")
|
||
print("（双流程站点按应到/实到交叉序列执行，覆盖全部相邻转换）")
|
||
print("====================================================")
|
||
|
||
results = [] # [(站点, 流程, 状态, 耗时秒, 错误信息)]
|
||
for idx, (site_name, label, func) in enumerate(plan, start=1):
|
||
print("\n----------------------------------------------------")
|
||
print(f"[步骤 {idx}/{total}] 站点【{site_name}】流程【{label}】")
|
||
print("----------------------------------------------------")
|
||
page = pages_map[site_name]
|
||
start = time.time()
|
||
status = "PASS"
|
||
err = ""
|
||
try:
|
||
page.bring_to_front()
|
||
ret = func(page)
|
||
if ret is False:
|
||
status = "FAIL"
|
||
err = "流程返回失败状态"
|
||
except Exception as e:
|
||
status = "FAIL"
|
||
err = str(e)
|
||
elapsed = time.time() - start
|
||
results.append((site_name, label, status, elapsed, err))
|
||
print(f">> 步骤结果: {status} (耗时 {elapsed:.1f}s)")
|
||
|
||
_print_test_report(results)
|
||
|
||
|
||
def _print_test_report(results):
|
||
"""打印自动化测试统计报告。"""
|
||
passed = sum(1 for r in results if r[2] == "PASS")
|
||
failed = len(results) - passed
|
||
|
||
print("\n====================================================")
|
||
print("自动化测试统计报告")
|
||
print("====================================================")
|
||
for i, (site_name, label, status, elapsed, err) in enumerate(results, start=1):
|
||
mark = "✅" if status == "PASS" else "❌"
|
||
print(f" {i:>2}. {mark} {status} {site_name} - {label} (耗时 {elapsed:.1f}s)")
|
||
if err:
|
||
note = err if len(err) <= 60 else err[:57] + "..."
|
||
print(f" 说明: {note}")
|
||
print("----------------------------------------------------")
|
||
print(f" 合计 {len(results)} 步：通过 {passed}，失败 {failed}")
|
||
if failed == 0:
|
||
print(" ✅ 全部流程跑通。")
|
||
else:
|
||
print(" ❌ 存在失败流程，请结合上方说明与运行日志排查。")
|
||
print("====================================================")
|
||
|
||
|
||
def run_multi_site_daemon():
|
||
"""多站点自动化主控流程"""
|
||
|
||
# 1. 读取配置文件
|
||
debug_mode = False
|
||
debug_target = ""
|
||
try:
|
||
if os.path.exists(CONFIG_PATH):
|
||
with open(CONFIG_PATH, "r", encoding="utf-8") as f:
|
||
config = yaml.safe_load(f)
|
||
debug_mode = config.get("debug", {}).get("enabled", False)
|
||
debug_target = config.get("debug", {}).get("target_site", "")
|
||
except Exception as e:
|
||
print(f"⚠️ 读取 config.yaml 异常，将使用全量模式启动: {e}")
|
||
|
||
# 动态确定需要挂载启动的网页
|
||
active_sites = {}
|
||
if debug_mode and debug_target in SITES_CONFIG:
|
||
print(f"\n🛠️ 【调试模式】仅加载目标站点: [{debug_target}]")
|
||
active_sites = {debug_target: SITES_CONFIG[debug_target]}
|
||
else:
|
||
active_sites = SITES_CONFIG
|
||
|
||
with sync_playwright() as p:
|
||
browser = p.chromium.launch(headless=False)
|
||
context = browser.new_context(viewport={"width": 1920, "height": 1080})
|
||
|
||
pages_map = {}
|
||
|
||
print("\n====================================================")
|
||
print("【启动】正在打开各站点页面...")
|
||
print("====================================================")
|
||
|
||
for site_name, url in active_sites.items():
|
||
print(f">> 正在启动【{site_name}】页面: {url}")
|
||
page = context.new_page()
|
||
page.goto(url)
|
||
pages_map[site_name] = page
|
||
|
||
print("\n====================================================")
|
||
print("【登录检测】正在准备各站点登录...")
|
||
print("====================================================")
|
||
|
||
# 针对支持纯代码自动登录的站点，在此处前置注入登录事件
|
||
if "韵达" in pages_map:
|
||
try:
|
||
pages_map["韵达"].bring_to_front()
|
||
site_yunda.yunda_login(pages_map["韵达"])
|
||
except Exception as e:
|
||
print(f" ⚠️ 韵达前置自动登录模块发生波动: {e}")
|
||
|
||
# ====================================================================
|
||
# 就绪轮询 (Ready Guard Polling)
|
||
# 自动识别各站点登录完成状态，无需手动回车
|
||
# ====================================================================
|
||
ready_status = {site: False for site in active_sites.keys()}
|
||
|
||
print("\n>> 正在轮询各站点就绪状态 (自动登录或手动登录均可)...")
|
||
while not all(ready_status.values()):
|
||
for site_name, page in pages_map.items():
|
||
if not ready_status[site_name]:
|
||
try:
|
||
# 0.5 秒轻量探测，避免阻塞主循环
|
||
if page.locator(READY_SELECTORS[site_name]).is_visible(
|
||
timeout=500
|
||
):
|
||
ready_status[site_name] = True
|
||
print(f" ✅ 【{site_name}】已检测到主页，登录就绪。")
|
||
except Exception:
|
||
pass
|
||
|
||
pending_sites = [s for s, ready in ready_status.items() if not ready]
|
||
if pending_sites:
|
||
print(
|
||
f" ⏳ 等待以下站点完成登录: [{', '.join(pending_sites)}] ... (请在浏览器中操作)"
|
||
)
|
||
page.wait_for_timeout(3000) # 等待 3 秒后进行下一轮检查
|
||
|
||
print("\n====================================================")
|
||
print("【准备】所有站点已就绪，正在清理初始弹窗...")
|
||
print("====================================================")
|
||
|
||
# 顺心：处理初始弹窗
|
||
if "顺心" in pages_map:
|
||
try:
|
||
sx_page = pages_map["顺心"]
|
||
sx_page.bring_to_front()
|
||
print(">> 正在处理【顺心】弹窗与遮罩...")
|
||
sx_page.locator("a").nth(4).click(timeout=2000)
|
||
sx_page.wait_for_timeout(500)
|
||
sx_page.get_by_role("button", name="Close").click(timeout=2000)
|
||
sx_page.wait_for_timeout(500)
|
||
sx_page.get_by_role("button", name="不再询问").click(timeout=2000)
|
||
print(" ✅ 【顺心】初始弹窗处理完成。")
|
||
except Exception:
|
||
pass # 环境可能很干净没有弹窗，无视报错
|
||
|
||
# 百世：循环关闭初始弹窗
|
||
if "百世" in pages_map:
|
||
try:
|
||
bs_page = pages_map["百世"]
|
||
bs_page.bring_to_front()
|
||
print(">> 正在处理【百世】阅读完毕与关闭按钮...")
|
||
for round_idx in range(4):
|
||
handled_any = False
|
||
try:
|
||
read_btns = bs_page.locator("button:has-text('阅读完毕')")
|
||
if read_btns.count() > 0:
|
||
for i in range(read_btns.count()):
|
||
if read_btns.nth(i).is_visible(timeout=500):
|
||
read_btns.nth(i).click()
|
||
handled_any = True
|
||
except Exception:
|
||
pass
|
||
try:
|
||
if bs_page.locator("button:has-text('关 闭')").is_visible(
|
||
timeout=500
|
||
):
|
||
bs_page.locator("button:has-text('关 闭')").click()
|
||
handled_any = True
|
||
except Exception:
|
||
pass
|
||
if not handled_any:
|
||
break
|
||
bs_page.wait_for_timeout(800)
|
||
print(" ✅ 【百世】初始弹窗处理完成。")
|
||
except Exception:
|
||
pass
|
||
|
||
if "中通" in pages_map:
|
||
print(" ✅ 【中通】已就绪。")
|
||
|
||
if "韵达" in pages_map:
|
||
print(" ✅ 【韵达】已就绪。")
|
||
|
||
def is_site_ready(site_name):
|
||
if site_name not in pages_map:
|
||
print(f"\n🚫 站点 [{site_name}] 未加载（当前为调试模式），已跳过。")
|
||
return False
|
||
return True
|
||
|
||
while True:
|
||
print("\n====================================================")
|
||
print(" 物流数据下载主菜单 ")
|
||
if debug_mode:
|
||
print(f" [ 调试模式，仅加载: {debug_target} ]")
|
||
print("====================================================")
|
||
print(" 模块一：【顺心】数据处理流")
|
||
print(" [1] 执行 - 应到货物数据下载")
|
||
print(" [2] 执行 - 实到货物数据下载")
|
||
print("-" * 52)
|
||
print(" 模块二：【百世】数据处理流")
|
||
print(" [3] 执行 - 一键提取应到未到异常数据")
|
||
print("-" * 52)
|
||
print(" 模块三：【中通】数据处理流")
|
||
print(" [4] 执行 - 应到货物数据下载")
|
||
print(" [5] 执行 - 实到货物数据下载")
|
||
print("-" * 52)
|
||
print(" 模块四：【韵达】数据处理流")
|
||
print(" [6] 执行 - 应到货物数据下载")
|
||
print(" [7] 执行 - 实到货物数据下载")
|
||
print("-" * 52)
|
||
print(" 自动化测试")
|
||
print(" [8] 执行 - 全站点下载流程自动化测试 (交叉跑通校验)")
|
||
print("-" * 52)
|
||
print(" 全局离线数据处理")
|
||
print(" [9] 执行 - 异常数据清洗比对 (Left Anti-Join)")
|
||
print("-" * 52)
|
||
print(" [0] 退出系统")
|
||
print("====================================================")
|
||
|
||
choice = input("请输入任务编号并回车: ")
|
||
|
||
try:
|
||
if choice == "1" and is_site_ready("顺心"):
|
||
pages_map["顺心"].bring_to_front()
|
||
site_shunxin.shunxin_expected_download(pages_map["顺心"])
|
||
elif choice == "2" and is_site_ready("顺心"):
|
||
pages_map["顺心"].bring_to_front()
|
||
site_shunxin.shunxin_actual_download(pages_map["顺心"])
|
||
elif choice == "3" and is_site_ready("百世"):
|
||
pages_map["百世"].bring_to_front()
|
||
site_baishi.baishi_download_undelivered_data(pages_map["百世"])
|
||
elif choice == "4" and is_site_ready("中通"):
|
||
pages_map["中通"].bring_to_front()
|
||
site_zto.zto_expected_download(pages_map["中通"])
|
||
elif choice == "5" and is_site_ready("中通"):
|
||
pages_map["中通"].bring_to_front()
|
||
site_zto.zto_actual_download(pages_map["中通"])
|
||
elif choice == "6" and is_site_ready("韵达"):
|
||
pages_map["韵达"].bring_to_front()
|
||
site_yunda.yunda_expected_download(pages_map["韵达"])
|
||
elif choice == "7" and is_site_ready("韵达"):
|
||
pages_map["韵达"].bring_to_front()
|
||
site_yunda.yunda_actual_download(pages_map["韵达"])
|
||
elif choice == "8":
|
||
run_automation_test(pages_map)
|
||
elif choice == "9":
|
||
site_name = input(
|
||
"请输入要比对的网点名称 (如 顺心/中通/韵达): "
|
||
).strip()
|
||
if not site_name:
|
||
site_name = "顺心"
|
||
task_process_undelivered_data(site_name)
|
||
elif choice == "0":
|
||
print("\n正在关闭浏览器并退出...")
|
||
break
|
||
else:
|
||
if choice not in ["1", "2", "3", "4", "5", "6", "7", "8", "9", "0"]:
|
||
print("\n⚠️ 无效输入，请查证后回车。")
|
||
except Exception as e:
|
||
print(f"❌ 任务调度异常: {e}")
|
||
|
||
browser.close()
|
||
print("程序已退出。")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
run_multi_site_daemon()
|