351 lines
15 KiB
Python
351 lines
15 KiB
Python
"""CLI：argparse 与分发（票据合规预审助手）。
|
||
|
||
只负责参数解析并转交 service；业务内核在 prereview_service / standards_service，
|
||
勿在此按入口复制逻辑。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
import sys
|
||
from typing import List, Optional
|
||
|
||
from service.task_service import (
|
||
cmd_add_whitelist,
|
||
cmd_config_path,
|
||
cmd_extract_invoices,
|
||
cmd_health,
|
||
cmd_init_db,
|
||
cmd_list_holidays,
|
||
cmd_list_rpa_invoices,
|
||
cmd_list_reports,
|
||
cmd_list_whitelist,
|
||
cmd_mark_verified,
|
||
cmd_prereview,
|
||
cmd_prereview_batch_input,
|
||
cmd_remove_whitelist,
|
||
cmd_render_html,
|
||
cmd_rpa_invoice_stats,
|
||
cmd_run,
|
||
cmd_set_standard,
|
||
cmd_show_rpa_invoice,
|
||
cmd_show_report,
|
||
cmd_stats,
|
||
cmd_test_rpa_connection,
|
||
cmd_version,
|
||
cmd_view_standards,
|
||
)
|
||
from util.config_bootstrap import bootstrap_skill_config
|
||
from util.argparse_zh import ZhArgumentParser
|
||
from util.constants import LOG_LOGGER_NAME, SKILL_SLUG
|
||
from util.logging_config import get_skill_logger, setup_skill_logging
|
||
|
||
|
||
def _cli_str_or_none(raw: Optional[str]) -> Optional[str]:
|
||
if raw is None:
|
||
return None
|
||
v = str(raw).strip()
|
||
return v or None
|
||
|
||
|
||
def _handle_run(args: argparse.Namespace) -> int:
|
||
tail = [str(x).strip() for x in (args.run_tail or []) if str(x).strip()]
|
||
if len(tail) > 2:
|
||
print("参数过多。")
|
||
print("用法：python main.py run [target [input_id]] | run [-t target] [-i input_id]")
|
||
return 1
|
||
|
||
t_target: Optional[str] = None
|
||
t_input: Optional[str] = None
|
||
if len(tail) == 2:
|
||
t_target, t_input = tail[0], tail[1]
|
||
elif len(tail) == 1:
|
||
if tail[0].isdigit():
|
||
t_input = tail[0]
|
||
else:
|
||
t_target = tail[0]
|
||
|
||
pick_t = _cli_str_or_none(getattr(args, "target", None))
|
||
pick_i = _cli_str_or_none(getattr(args, "input_id", None))
|
||
target = pick_t or t_target
|
||
input_id = pick_i or t_input
|
||
return cmd_run(target=target, input_id=input_id)
|
||
|
||
|
||
def _handle_prereview(args: argparse.Namespace) -> int:
|
||
return cmd_prereview(
|
||
invoice_json=str(args.invoice_json or "").strip(),
|
||
purpose=_cli_str_or_none(args.purpose) or "",
|
||
project_no=_cli_str_or_none(args.project_no) or "",
|
||
trip_start=_cli_str_or_none(args.trip_start) or "",
|
||
trip_end=_cli_str_or_none(args.trip_end) or "",
|
||
trip_city=_cli_str_or_none(args.trip_city) or "",
|
||
staff_title=_cli_str_or_none(args.staff_title) or "",
|
||
)
|
||
|
||
|
||
def _handle_render_html(args: argparse.Namespace) -> int:
|
||
report_id = getattr(args, "report_id", None)
|
||
return cmd_render_html(report_id=int(report_id) if report_id else None)
|
||
|
||
|
||
def _handle_list_reports(args: argparse.Namespace) -> int:
|
||
limit = int(args.limit or 20)
|
||
return cmd_list_reports(limit=limit, verdict=_cli_str_or_none(args.verdict))
|
||
|
||
|
||
def _handle_show_report(args: argparse.Namespace) -> int:
|
||
return cmd_show_report(report_id=int(args.report_id))
|
||
|
||
|
||
def _handle_mark_verified(args: argparse.Namespace) -> int:
|
||
return cmd_mark_verified(
|
||
report_id=int(args.report_id),
|
||
result=str(args.result or "").strip(),
|
||
note=_cli_str_or_none(args.note) or "",
|
||
)
|
||
|
||
|
||
def _handle_view_standards(args: argparse.Namespace) -> int:
|
||
return cmd_view_standards(category=_cli_str_or_none(args.category))
|
||
|
||
|
||
def _handle_set_standard(args: argparse.Namespace) -> int:
|
||
return cmd_set_standard(
|
||
category=str(args.category or "").strip(),
|
||
city_tier=str(args.city_tier or "").strip(),
|
||
staff_title=str(args.staff_title or "").strip(),
|
||
max_amount=float(args.max_amount),
|
||
unit=_cli_str_or_none(args.unit) or "",
|
||
basis_doc=_cli_str_or_none(args.basis_doc) or "",
|
||
clause=_cli_str_or_none(args.clause) or "",
|
||
city_example=_cli_str_or_none(args.city_example) or "",
|
||
)
|
||
|
||
|
||
def _handle_add_whitelist(args: argparse.Namespace) -> int:
|
||
return cmd_add_whitelist(
|
||
tax_no=str(args.tax_no or "").strip(),
|
||
org_name=str(args.org_name or "").strip(),
|
||
note=_cli_str_or_none(args.note) or "",
|
||
)
|
||
|
||
|
||
def _handle_remove_whitelist(args: argparse.Namespace) -> int:
|
||
return cmd_remove_whitelist(tax_no=str(args.tax_no or "").strip())
|
||
|
||
|
||
def _handle_list_holidays(args: argparse.Namespace) -> int:
|
||
return cmd_list_holidays(year=_cli_str_or_none(args.year))
|
||
|
||
|
||
def _handle_extract_invoices(args: argparse.Namespace) -> int:
|
||
return cmd_extract_invoices(
|
||
username=str(args.username or "").strip(),
|
||
password=str(args.password or "").strip(),
|
||
profile_dir=str(args.profile_dir or "").strip(),
|
||
headless=bool(args.headless),
|
||
detail_mode=bool(args.detail_mode),
|
||
)
|
||
|
||
|
||
def _handle_test_rpa_connection(args: argparse.Namespace) -> int:
|
||
return cmd_test_rpa_connection(
|
||
username=str(args.username or "").strip(),
|
||
password=str(args.password or "").strip(),
|
||
profile_dir=str(args.profile_dir or "").strip(),
|
||
)
|
||
|
||
|
||
def _handle_list_rpa_invoices(args: argparse.Namespace) -> int:
|
||
return cmd_list_rpa_invoices(
|
||
limit=int(args.limit or 20),
|
||
batch_id=_cli_str_or_none(args.batch_id),
|
||
username=_cli_str_or_none(args.username),
|
||
status=_cli_str_or_none(args.status),
|
||
)
|
||
|
||
|
||
def _handle_show_rpa_invoice(args: argparse.Namespace) -> int:
|
||
return cmd_show_rpa_invoice(record_id=int(args.record_id))
|
||
|
||
|
||
def _handle_rpa_invoice_stats(args: argparse.Namespace) -> int:
|
||
return cmd_rpa_invoice_stats()
|
||
|
||
|
||
def _print_full_usage() -> None:
|
||
print("票据合规预审助手（main.py）可用命令：")
|
||
print(" python main.py prereview --invoice-json <file|-> [--purpose 文本] [--project-no 编号]")
|
||
print(" [--trip-start YYYY-MM-DD] [--trip-end YYYY-MM-DD]")
|
||
print(" [--trip-city 城市] [--staff-title 职级]")
|
||
print(" python main.py prereview-batch-input # 批量预审 input/ 目录 JSON")
|
||
print(" python main.py render-html [--report-id N] # 重建初审意见单（缺省最近 10 份）")
|
||
print(" python main.py list-reports [--limit N] [--verdict 通过|待确认|驳回]")
|
||
print(" python main.py show-report --id N")
|
||
print(" python main.py mark-verified --report-id N --result pass|fail [--note 备注]")
|
||
print(" python main.py view-standards [--category 住宿费|交通费]")
|
||
print(" python main.py set-standard --category C --city-tier T --staff-title J --max-amount N")
|
||
print(" python main.py list-whitelist")
|
||
print(" python main.py add-whitelist --tax-no X --name Y [--note 备注]")
|
||
print(" python main.py remove-whitelist --tax-no X")
|
||
print(" python main.py list-holidays [--year YYYY]")
|
||
print(" python main.py stats")
|
||
print(" python main.py run [target [input_id]] [-t target] [-i input_id]")
|
||
print(" python main.py extract-invoices --username U --password P --profile-dir DIR")
|
||
print(" [--headless] [--detail-mode]")
|
||
print(" python main.py test-rpa-connection --username U --password P --profile-dir DIR")
|
||
print(" python main.py health")
|
||
print(" python main.py config-path")
|
||
print(" python main.py version")
|
||
print(" python main.py init-db")
|
||
|
||
|
||
def build_parser() -> ZhArgumentParser:
|
||
p = ZhArgumentParser(
|
||
prog="main.py",
|
||
description="票据合规预审助手：发票字段完整性、税号白名单、日期逻辑、经费标准四步审核，红黄绿灯预警并生成《初审意见单》。",
|
||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||
)
|
||
sub = p.add_subparsers(dest="cmd", required=True, parser_class=ZhArgumentParser)
|
||
|
||
# prereview
|
||
sp = sub.add_parser("prereview", help="票据合规预审（主命令）")
|
||
sp.add_argument("--invoice-json", required=True, metavar="发票JSON(文件|内联|-)", dest="invoice_json")
|
||
sp.add_argument("--purpose", default="", metavar="报销事由")
|
||
sp.add_argument("--project-no", default="", metavar="预算项目编号", dest="project_no")
|
||
sp.add_argument("--trip-start", default="", metavar="出差开始 YYYY-MM-DD", dest="trip_start")
|
||
sp.add_argument("--trip-end", default="", metavar="出差结束 YYYY-MM-DD", dest="trip_end")
|
||
sp.add_argument("--trip-city", default="", metavar="目的地城市", dest="trip_city")
|
||
sp.add_argument("--staff-title", default="", metavar="职级(副高及以上/中级及以下)", dest="staff_title")
|
||
sp.set_defaults(handler=_handle_prereview)
|
||
|
||
# prereview-batch-input
|
||
sp = sub.add_parser("prereview-batch-input", help="批量预审 input/ 目录下的发票 JSON 文件")
|
||
sp.set_defaults(handler=lambda _a: cmd_prereview_batch_input())
|
||
|
||
# render-html
|
||
sp = sub.add_parser("render-html", help="重建《初审意见单》HTML（缺省最近 10 份）")
|
||
sp.add_argument("--report-id", default=None, metavar="报告编号", dest="report_id")
|
||
sp.set_defaults(handler=_handle_render_html)
|
||
|
||
# list-reports
|
||
sp = sub.add_parser("list-reports", help="查询初审历史")
|
||
sp.add_argument("--limit", default=20, metavar="条数")
|
||
sp.add_argument("--verdict", default=None, metavar="结论筛选")
|
||
sp.set_defaults(handler=_handle_list_reports)
|
||
|
||
# show-report
|
||
sp = sub.add_parser("show-report", help="查看一份初审报告详情")
|
||
sp.add_argument("--report-id", required=True, metavar="报告编号", dest="report_id")
|
||
sp.set_defaults(handler=_handle_show_report)
|
||
|
||
# mark-verified
|
||
sp = sub.add_parser("mark-verified", help="回填人工查验结果（pass/fail）")
|
||
sp.add_argument("--report-id", required=True, metavar="报告编号", dest="report_id")
|
||
sp.add_argument("--result", required=True, metavar="pass|fail")
|
||
sp.add_argument("--note", default=None, metavar="查验备注")
|
||
sp.set_defaults(handler=_handle_mark_verified)
|
||
|
||
# view-standards
|
||
sp = sub.add_parser("view-standards", help="查看经费标准库")
|
||
sp.add_argument("--category", default=None, metavar="费用类别")
|
||
sp.set_defaults(handler=_handle_view_standards)
|
||
|
||
# set-standard
|
||
sp = sub.add_parser("set-standard", help="新增/更新经费标准")
|
||
sp.add_argument("--category", required=True, metavar="费用类别")
|
||
sp.add_argument("--city-tier", required=True, metavar="城市等级", dest="city_tier")
|
||
sp.add_argument("--staff-title", required=True, metavar="职级", dest="staff_title")
|
||
sp.add_argument("--max-amount", required=True, type=float, metavar="标准值", dest="max_amount")
|
||
sp.add_argument("--unit", default=None, metavar="单位")
|
||
sp.add_argument("--basis-doc", default=None, metavar="依据文件", dest="basis_doc")
|
||
sp.add_argument("--clause", default=None, metavar="条款")
|
||
sp.add_argument("--city-example", default=None, metavar="城市示例(逗号分隔)", dest="city_example")
|
||
sp.set_defaults(handler=_handle_set_standard)
|
||
|
||
# whitelist
|
||
sp = sub.add_parser("list-whitelist", help="查看税号白名单")
|
||
sp.set_defaults(handler=lambda _a: cmd_list_whitelist())
|
||
|
||
sp = sub.add_parser("add-whitelist", help="新增税号白名单")
|
||
sp.add_argument("--tax-no", required=True, metavar="税号", dest="tax_no")
|
||
sp.add_argument("--name", required=True, metavar="单位名称", dest="org_name")
|
||
sp.add_argument("--note", default=None, metavar="备注")
|
||
sp.set_defaults(handler=_handle_add_whitelist)
|
||
|
||
sp = sub.add_parser("remove-whitelist", help="删除税号白名单")
|
||
sp.add_argument("--tax-no", required=True, metavar="税号", dest="tax_no")
|
||
sp.set_defaults(handler=_handle_remove_whitelist)
|
||
|
||
# holidays
|
||
sp = sub.add_parser("list-holidays", help="查看节假日表")
|
||
sp.add_argument("--year", default=None, metavar="年份")
|
||
sp.set_defaults(handler=_handle_list_holidays)
|
||
|
||
# stats
|
||
sp = sub.add_parser("stats", help="预审统计")
|
||
sp.set_defaults(handler=lambda _a: cmd_stats())
|
||
|
||
# RPA commands
|
||
sp = sub.add_parser("extract-invoices", help="从网站提取发票信息")
|
||
sp.add_argument("--username", required=True, metavar="用户名")
|
||
sp.add_argument("--password", required=True, metavar="密码")
|
||
sp.add_argument("--profile-dir", required=True, metavar="浏览器配置目录", dest="profile_dir")
|
||
sp.add_argument("--headless", action="store_true", help="无头模式（生产不建议）")
|
||
sp.add_argument("--detail-mode", action="store_true", help="提取详细信息", dest="detail_mode")
|
||
sp.set_defaults(handler=_handle_extract_invoices)
|
||
|
||
sp = sub.add_parser("test-rpa-connection", help="测试RPA连接和登录")
|
||
sp.add_argument("--username", required=True, metavar="用户名")
|
||
sp.add_argument("--password", required=True, metavar="密码")
|
||
sp.add_argument("--profile-dir", required=True, metavar="浏览器配置目录", dest="profile_dir")
|
||
sp.set_defaults(handler=_handle_test_rpa_connection)
|
||
|
||
# RPA数据查询命令
|
||
sp = sub.add_parser("list-rpa-invoices", help="查询RPA提取的发票数据")
|
||
sp.add_argument("--limit", default=20, type=int, metavar="条数")
|
||
sp.add_argument("--batch-id", default=None, metavar="批次ID", dest="batch_id")
|
||
sp.add_argument("--username", default=None, metavar="用户名")
|
||
sp.add_argument("--status", default=None, metavar="状态")
|
||
sp.set_defaults(handler=_handle_list_rpa_invoices)
|
||
|
||
sp = sub.add_parser("show-rpa-invoice", help="查看RPA提取的发票详情")
|
||
sp.add_argument("--record-id", required=True, type=int, metavar="记录ID", dest="record_id")
|
||
sp.set_defaults(handler=_handle_show_rpa_invoice)
|
||
|
||
sp = sub.add_parser("rpa-invoice-stats", help="获取RPA提取统计信息")
|
||
sp.set_defaults(handler=_handle_rpa_invoice_stats)
|
||
|
||
sp = sub.add_parser("run", help="任务执行命令骨架")
|
||
sp.add_argument("--target", "-t", default=None, metavar="目标")
|
||
sp.add_argument("--input-id", "-i", default=None, metavar="输入id", dest="input_id")
|
||
sp.add_argument("run_tail", nargs="*", metavar="位置参数")
|
||
sp.set_defaults(handler=_handle_run)
|
||
|
||
sp = sub.add_parser("health", help="健康检查")
|
||
sp.set_defaults(handler=lambda _a: cmd_health())
|
||
|
||
sp = sub.add_parser("config-path", help="输出用户 .env 与模板路径")
|
||
sp.set_defaults(handler=lambda _a: cmd_config_path())
|
||
|
||
sp = sub.add_parser("version", help="版本信息(JSON)")
|
||
sp.set_defaults(handler=lambda _a: cmd_version())
|
||
|
||
sp = sub.add_parser("init-db", help="幂等初始化本地数据库与数据管理元数据")
|
||
sp.set_defaults(handler=lambda _a: cmd_init_db())
|
||
return p
|
||
|
||
|
||
def main(argv: Optional[List[str]] = None) -> int:
|
||
argv = argv if argv is not None else sys.argv[1:]
|
||
bootstrap_skill_config()
|
||
setup_skill_logging(SKILL_SLUG, LOG_LOGGER_NAME)
|
||
get_skill_logger().info("cli_start argv=%s", sys.argv)
|
||
if not argv:
|
||
_print_full_usage()
|
||
return 1
|
||
parser = build_parser()
|
||
args = parser.parse_args(argv)
|
||
return int(args.handler(args))
|