日常报销和差旅报销都可以走通

src/bot/__init__.py

@@ -0,0 +1,93 @@
+"""
+浏览器自动化填报
+
+使用 Playwright 操作财务报销系统，自动完成登录、填单、上传附件等操作。
+
+对外接口:
+    run_bot(config, travel_info, normal_info)     启动浏览器并执行填报流程
+    run_bot_web(config, work_dir)                 Web 模式填报（从缓存加载信息）
+"""
+
+from pathlib import Path
+from typing import Any
+
+from .. import get_logger
+from .base import BaseBot
+
+log = get_logger("bot")
+
+
+def run_bot(
+    config: dict[str, Any],
+    headless: bool = False,
+    work_dir: Path | None = None,
+    travel_info: dict[str, Any] | None = None,
+    normal_info: dict[str, Any] | None = None,
+) -> None:
+    """执行完整的浏览器填报流程，根据发票类型自动路由
+
+    Args:
+        config: 配置字典。
+        headless: 是否无头模式。
+        work_dir: 工作目录。
+        travel_info: 差旅信息（由 pipeline 层提前提取并传入，非差旅时传 None）。
+        normal_info: 普通发票信息（由 pipeline 层提前提取并传入，非普通时传 None）。
+    """
+    if not config["username"] or not config["password"]:
+        raise ValueError("缺少用户名或密码")
+
+    if not work_dir:
+        raise ValueError("缺少工作目录")
+
+    bot = BaseBot(config, headless=headless)
+    bot.work_dir = work_dir
+
+    try:
+        bot.launch()
+        bot.login_portal()
+
+        if travel_info is not None:
+            log.info("处理差旅发票...")
+            bot.navigate_to_reimburse(page_key="travel_page")
+            bot.create_new_form()
+            from . import travel
+
+            travel.run(bot, travel_info)
+        elif normal_info is not None:
+            log.info("处理普通发票...")
+            bot.navigate_to_reimburse(page_key="reimburse_page")
+            bot.create_new_form()
+            from . import normal
+
+            normal.run(bot, normal_info)
+        else:
+            raise ValueError("缺少差旅信息（travel_info）和普通发票信息（normal_info），无法继续填报")
+
+    except Exception as e:
+        log.error(f"操作失败: {e}")
+        try:
+            bot._screenshot("error")
+        except Exception:
+            pass
+        raise
+    finally:
+        bot.close()
+
+
+def run_bot_web(config: dict[str, Any], work_dir: Path) -> None:
+    """Web 模式填报 — headless，附件从指定目录读取
+
+    Web 端的信息提取由 app.py 的管道负责，此处从缓存加载。
+    """
+    from ..doc.llm_extractor import load_cache
+
+    cache_map = load_cache(work_dir)
+    travel_info = cache_map.get("travel_info")
+    normal_info = cache_map.get("normal_info")
+    run_bot(
+        config,
+        headless=True,
+        work_dir=work_dir,
+        travel_info=travel_info,
+        normal_info=normal_info,
+    )