feat: 票据合规预审助手首版 v1.0.0（verify-invoice-default）
This commit is contained in:
chenfuqiang
277
scripts/cli/app.py
Normal file
277
scripts/cli/app.py
Normal file
@@ -0,0 +1,277 @@
|
||||
"""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_health,
|
||||
cmd_init_db,
|
||||
cmd_list_holidays,
|
||||
cmd_list_reports,
|
||||
cmd_list_whitelist,
|
||||
cmd_mark_verified,
|
||||
cmd_prereview,
|
||||
cmd_prereview_batch_input,
|
||||
cmd_remove_whitelist,
|
||||
cmd_render_html,
|
||||
cmd_run,
|
||||
cmd_set_standard,
|
||||
cmd_show_report,
|
||||
cmd_stats,
|
||||
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 _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 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())
|
||||
|
||||
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))
|
||||
350
scripts/db/connection.py
Normal file
350
scripts/db/connection.py
Normal file
@@ -0,0 +1,350 @@
|
||||
"""SQLite 连接与幂等初始化（票据合规预审助手）。
|
||||
|
||||
业务表 5 张：
|
||||
- invoice_records（发票登记，只读，预审自动写入）
|
||||
- prereview_reports（初审意见，结论只读；查验回填走 mark-verified）
|
||||
- expense_standards（经费标准库，可编辑）
|
||||
- taxid_whitelist（税号白名单，可编辑）
|
||||
- holiday_calendar（节假日，可编辑）
|
||||
|
||||
表间禁止 SQLite 外键，仅存业务 id（见 references/SCHEMA.md）。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import sqlite3
|
||||
|
||||
from db.display_metadata import (
|
||||
init_display_metadata,
|
||||
seed_column_metadata_batch,
|
||||
upsert_table_metadata,
|
||||
)
|
||||
from db.seed_data import seed_all
|
||||
from db.timestamp_columns import init_updated_at_trigger
|
||||
from util.runtime_paths import get_db_path
|
||||
|
||||
|
||||
def get_conn() -> sqlite3.Connection:
|
||||
return sqlite3.connect(get_db_path())
|
||||
|
||||
|
||||
_INVOICE_RECORDS_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS invoice_records (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
batch_no TEXT NOT NULL,
|
||||
invoice_type TEXT,
|
||||
invoice_code TEXT,
|
||||
invoice_number TEXT NOT NULL,
|
||||
issue_date TEXT,
|
||||
amount REAL,
|
||||
amount_in_words TEXT,
|
||||
tax_rate REAL,
|
||||
buyer_name TEXT,
|
||||
buyer_tax_no TEXT,
|
||||
seller_name TEXT,
|
||||
seller_tax_no TEXT,
|
||||
expense_category TEXT,
|
||||
city TEXT,
|
||||
days INTEGER,
|
||||
purpose TEXT,
|
||||
project_no TEXT,
|
||||
trip_context_json TEXT,
|
||||
created_at INTEGER NOT NULL DEFAULT (unixepoch())
|
||||
)
|
||||
"""
|
||||
|
||||
_INVOICE_RECORDS_INDEX = """
|
||||
CREATE INDEX IF NOT EXISTS idx_invoice_records_batch
|
||||
ON invoice_records(batch_no)
|
||||
"""
|
||||
|
||||
_PREREVIEW_REPORTS_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS prereview_reports (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
batch_no TEXT NOT NULL,
|
||||
invoice_id INTEGER,
|
||||
invoice_number TEXT,
|
||||
seller_name TEXT,
|
||||
amount REAL,
|
||||
verdict TEXT NOT NULL,
|
||||
pass_json TEXT,
|
||||
warn_json TEXT,
|
||||
reject_json TEXT,
|
||||
summary TEXT,
|
||||
verify_status TEXT NOT NULL DEFAULT '待查验',
|
||||
verify_note TEXT,
|
||||
verify_updated_at INTEGER,
|
||||
report_file TEXT,
|
||||
created_at INTEGER NOT NULL DEFAULT (unixepoch())
|
||||
)
|
||||
"""
|
||||
|
||||
_PREREVIEW_REPORTS_INDEX = """
|
||||
CREATE INDEX IF NOT EXISTS idx_prereview_reports_batch
|
||||
ON prereview_reports(batch_no)
|
||||
"""
|
||||
|
||||
_EXPENSE_STANDARDS_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS expense_standards (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
category TEXT NOT NULL,
|
||||
city_tier TEXT NOT NULL,
|
||||
city_example TEXT,
|
||||
staff_title TEXT NOT NULL,
|
||||
max_amount REAL NOT NULL,
|
||||
unit TEXT,
|
||||
basis_doc TEXT,
|
||||
clause TEXT,
|
||||
effective_date INTEGER,
|
||||
created_at INTEGER NOT NULL DEFAULT (unixepoch()),
|
||||
updated_at INTEGER NOT NULL DEFAULT (unixepoch()),
|
||||
UNIQUE (category, city_tier, staff_title)
|
||||
)
|
||||
"""
|
||||
|
||||
_TAXID_WHITELIST_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS taxid_whitelist (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
tax_no TEXT NOT NULL UNIQUE,
|
||||
org_name TEXT,
|
||||
note TEXT,
|
||||
created_at INTEGER NOT NULL DEFAULT (unixepoch()),
|
||||
updated_at INTEGER NOT NULL DEFAULT (unixepoch())
|
||||
)
|
||||
"""
|
||||
|
||||
_HOLIDAY_CALENDAR_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS holiday_calendar (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
holiday_date TEXT NOT NULL UNIQUE,
|
||||
holiday_name TEXT,
|
||||
created_at INTEGER NOT NULL DEFAULT (unixepoch()),
|
||||
updated_at INTEGER NOT NULL DEFAULT (unixepoch())
|
||||
)
|
||||
"""
|
||||
|
||||
_TS_COLUMN_NAMES = {"created_at", "updated_at", "effective_date", "verify_updated_at"}
|
||||
|
||||
|
||||
def _col(column_name: str, display_name: str, description: str, **extra: object) -> dict[str, object]:
|
||||
base: dict[str, object] = {
|
||||
"column_name": column_name,
|
||||
"display_name": display_name,
|
||||
"description": description,
|
||||
"visible": 1,
|
||||
"searchable": 0,
|
||||
"editable": 0,
|
||||
}
|
||||
base.update(extra)
|
||||
if column_name in _TS_COLUMN_NAMES:
|
||||
base["display_type"] = "datetime_unix_seconds"
|
||||
return base
|
||||
|
||||
|
||||
_INVOICE_RECORDS_COLUMNS: tuple[dict[str, object], ...] = (
|
||||
_col("id", "编号", "主键", searchable=1),
|
||||
_col("batch_no", "预审批次号", "同一次预审请求的批次号，同批多张发票共享。", searchable=1),
|
||||
_col("invoice_type", "发票类型", "如增值税电子普通发票、增值税专用发票、出租车票等。", searchable=1),
|
||||
_col("invoice_code", "发票代码", "票面发票代码（数电票可能为空）。", searchable=1),
|
||||
_col("invoice_number", "发票号码", "票面发票号码。", searchable=1),
|
||||
_col("issue_date", "开票日期", "票面开票日期（YYYY-MM-DD）。", searchable=1),
|
||||
_col("amount", "金额", "票面价税合计或金额（元）。", searchable=1, display_type="number"),
|
||||
_col("amount_in_words", "大写金额", "票面大写金额（默认隐藏）。", visible=0),
|
||||
_col("tax_rate", "税率", "票面税率（小数，如 0.06 表示 6%）。", display_type="number"),
|
||||
_col("buyer_name", "购买方名称", "票面购买方全称。", searchable=1),
|
||||
_col("buyer_tax_no", "购买方税号", "购买方统一社会信用代码，用于白名单比对。", searchable=1),
|
||||
_col("seller_name", "销售方名称", "票面销售方全称。", searchable=1),
|
||||
_col("seller_tax_no", "销售方税号", "销售方统一社会信用代码，用于集中开票风控。", searchable=1),
|
||||
_col("expense_category", "费用类别", "报销费用类别，如住宿费、交通费、会议费。", searchable=1),
|
||||
_col("city", "消费城市", "发票对应消费城市（住宿/交通判断用）。", searchable=1),
|
||||
_col("days", "天数", "该发票覆盖的天数（住宿费核算用，缺省按 1 天）。", searchable=1, display_type="number"),
|
||||
_col("purpose", "报销事由", "用户填报的报销事由文字。", searchable=1),
|
||||
_col("project_no", "预算项目编号", "系统带入的预算项目编号。", searchable=1),
|
||||
_col("trip_context_json", "差旅上下文", "出差起止、目的地、职级等上下文（JSON）。", visible=0, display_type="textarea"),
|
||||
_col("created_at", "创建时间", "记录入库时间（Unix 秒级时间戳，自动生成）。"),
|
||||
)
|
||||
|
||||
_PREREVIEW_REPORTS_COLUMNS: tuple[dict[str, object], ...] = (
|
||||
_col("id", "报告编号", "主键，初审意见单编号。", searchable=1),
|
||||
_col("batch_no", "预审批次号", "同一次预审请求的批次号。", searchable=1),
|
||||
_col("invoice_id", "发票编号", "关联发票登记表主键。", searchable=1),
|
||||
_col("invoice_number", "发票号码", "本报告对应的发票号码。", searchable=1),
|
||||
_col("seller_name", "销售方名称", "本报告对应的销售方。", searchable=1),
|
||||
_col("amount", "金额", "本报告对应的票面金额（元）。", searchable=1, display_type="number"),
|
||||
_col(
|
||||
"verdict",
|
||||
"预审结论",
|
||||
"通过 / 待确认 / 驳回。",
|
||||
searchable=1,
|
||||
options_json='[{"value":"通过","label":"✅ 通过"},{"value":"待确认","label":"⚠️ 待确认"},{"value":"驳回","label":"❌ 驳回"}]',
|
||||
),
|
||||
_col("pass_json", "通过项", "通过项清单（JSON 数组）。", visible=0, display_type="textarea"),
|
||||
_col("warn_json", "待确认项", "待确认项清单（JSON 数组，附规则引用）。", visible=0, display_type="textarea"),
|
||||
_col("reject_json", "驳回项", "驳回项清单（JSON 数组，附违反条款）。", visible=0, display_type="textarea"),
|
||||
_col("summary", "结论摘要", "结论一句话摘要。", searchable=1),
|
||||
_col(
|
||||
"verify_status",
|
||||
"真伪查验",
|
||||
"待查验 / 已查验通过 / 已查验不符。",
|
||||
searchable=1,
|
||||
options_json='[{"value":"待查验","label":"待查验"},{"value":"已查验通过","label":"已查验通过"},{"value":"已查验不符","label":"已查验不符"}]',
|
||||
),
|
||||
_col("verify_note", "查验备注", "人工查验回填的备注。", searchable=1),
|
||||
_col("verify_updated_at", "查验回填时间", "最近一次人工查验回填时间（Unix 秒级）。"),
|
||||
_col("report_file", "意见单文件", "《初审意见单》HTML 文件路径。", visible=0),
|
||||
_col("created_at", "创建时间", "报告创建时间（Unix 秒级时间戳，自动生成）。"),
|
||||
)
|
||||
|
||||
_EXPENSE_STANDARDS_COLUMNS: tuple[dict[str, object], ...] = (
|
||||
_col("id", "编号", "主键", searchable=1),
|
||||
_col("category", "费用类别", "住宿费 / 交通费等。", searchable=1),
|
||||
_col("city_tier", "城市等级", "一线城市 / 省会及计划单列市 / 其他地区 / 通用。", searchable=1),
|
||||
_col("city_example", "城市示例", "该等级覆盖的城市（逗号分隔），用于城市归档。", searchable=1, editable=1),
|
||||
_col("staff_title", "职级", "如 副高及以上 / 中级及以下。", searchable=1),
|
||||
_col("max_amount", "标准值", "住宿费为上限金额（元）；交通费为席位等级代码。", searchable=1, editable=1, display_type="number"),
|
||||
_col("unit", "单位", "如 元/人/天。", searchable=1, editable=1),
|
||||
_col("basis_doc", "依据文件", "标准出处文件名，如《差旅费管理办法》。", searchable=1, editable=1),
|
||||
_col("clause", "条款", "标准出处条款号，如 第十条。", searchable=1, editable=1),
|
||||
_col("effective_date", "生效日期", "标准生效日期（Unix 秒级时间戳）。", searchable=1, editable=1),
|
||||
_col("created_at", "创建时间", "记录创建时间（Unix 秒级，自动生成）。"),
|
||||
_col("updated_at", "更新时间", "记录最后更新时间（Unix 秒级，自动维护）。"),
|
||||
)
|
||||
|
||||
_TAXID_WHITELIST_COLUMNS: tuple[dict[str, object], ...] = (
|
||||
_col("id", "编号", "主键", searchable=1),
|
||||
_col("tax_no", "税号", "购买方统一社会信用代码（唯一）。", searchable=1),
|
||||
_col("org_name", "单位名称", "税号对应单位全称。", searchable=1, editable=1),
|
||||
_col("note", "备注", "维护说明。", searchable=1, editable=1),
|
||||
_col("created_at", "创建时间", "记录创建时间（Unix 秒级，自动生成）。"),
|
||||
_col("updated_at", "更新时间", "记录最后更新时间（Unix 秒级，自动维护）。"),
|
||||
)
|
||||
|
||||
_HOLIDAY_CALENDAR_COLUMNS: tuple[dict[str, object], ...] = (
|
||||
_col("id", "编号", "主键", searchable=1),
|
||||
_col("holiday_date", "日期", "节假日日期（YYYY-MM-DD，唯一）。", searchable=1),
|
||||
_col("holiday_name", "节假日名称", "如 春节、国庆节。", searchable=1, editable=1),
|
||||
_col("created_at", "创建时间", "记录创建时间（Unix 秒级，自动生成）。"),
|
||||
_col("updated_at", "更新时间", "记录最后更新时间（Unix 秒级，自动维护）。"),
|
||||
)
|
||||
|
||||
|
||||
def _init_invoice_records_metadata(cursor) -> None:
|
||||
cursor.execute(_INVOICE_RECORDS_DDL)
|
||||
cursor.execute(_INVOICE_RECORDS_INDEX)
|
||||
upsert_table_metadata(
|
||||
cursor,
|
||||
table_name="invoice_records",
|
||||
display_name="发票登记",
|
||||
description="每次预审自动写入的发票主档，含票面要素与报销上下文。",
|
||||
grid_help="按批次号或开票日期查看；本表由预审自动写入，请勿手工修改。",
|
||||
visible=1,
|
||||
readonly=1,
|
||||
default_sort_json='[{"column":"id","direction":"desc"}]',
|
||||
default_visible_columns_json='["id","batch_no","invoice_type","invoice_number","issue_date","amount","seller_name","expense_category","project_no"]',
|
||||
)
|
||||
seed_column_metadata_batch(
|
||||
cursor,
|
||||
table_name="invoice_records",
|
||||
columns=_INVOICE_RECORDS_COLUMNS,
|
||||
valid_column_names=[str(c["column_name"]) for c in _INVOICE_RECORDS_COLUMNS],
|
||||
)
|
||||
|
||||
|
||||
def _init_prereview_reports_metadata(cursor) -> None:
|
||||
cursor.execute(_PREREVIEW_REPORTS_DDL)
|
||||
cursor.execute(_PREREVIEW_REPORTS_INDEX)
|
||||
upsert_table_metadata(
|
||||
cursor,
|
||||
table_name="prereview_reports",
|
||||
display_name="初审意见",
|
||||
description="每张发票的预审结论与红黄绿灯清单；人工查验结果在行级回填。",
|
||||
grid_help="“待确认/驳回”的行请在行级按钮回填人工查验结果；HTML 意见单可用顶栏按钮重建。",
|
||||
visible=1,
|
||||
readonly=1,
|
||||
default_sort_json='[{"column":"id","direction":"desc"}]',
|
||||
default_visible_columns_json='["id","invoice_number","seller_name","amount","verdict","verify_status","summary","created_at"]',
|
||||
)
|
||||
seed_column_metadata_batch(
|
||||
cursor,
|
||||
table_name="prereview_reports",
|
||||
columns=_PREREVIEW_REPORTS_COLUMNS,
|
||||
valid_column_names=[str(c["column_name"]) for c in _PREREVIEW_REPORTS_COLUMNS],
|
||||
)
|
||||
|
||||
|
||||
def _init_expense_standards_metadata(cursor) -> None:
|
||||
cursor.execute(_EXPENSE_STANDARDS_DDL)
|
||||
init_updated_at_trigger(cursor, "expense_standards")
|
||||
upsert_table_metadata(
|
||||
cursor,
|
||||
table_name="expense_standards",
|
||||
display_name="经费标准库",
|
||||
description="住宿费上限与交通席位等级标准（城市等级 × 职级），预审硬约束依据。",
|
||||
grid_help="数值为示例种子，请以学校现行管理办法为准；可直接编辑标准值/单位/条款。",
|
||||
visible=1,
|
||||
readonly=0,
|
||||
)
|
||||
seed_column_metadata_batch(
|
||||
cursor,
|
||||
table_name="expense_standards",
|
||||
columns=_EXPENSE_STANDARDS_COLUMNS,
|
||||
valid_column_names=[str(c["column_name"]) for c in _EXPENSE_STANDARDS_COLUMNS],
|
||||
)
|
||||
|
||||
|
||||
def _init_taxid_whitelist_metadata(cursor) -> None:
|
||||
cursor.execute(_TAXID_WHITELIST_DDL)
|
||||
init_updated_at_trigger(cursor, "taxid_whitelist")
|
||||
upsert_table_metadata(
|
||||
cursor,
|
||||
table_name="taxid_whitelist",
|
||||
display_name="税号白名单",
|
||||
description="准予报销的购买方税号清单；预审时逐张比对，不在名单内即驳回。",
|
||||
grid_help="顶栏按钮新增税号；行级按钮可删除错误条目。",
|
||||
visible=1,
|
||||
readonly=0,
|
||||
)
|
||||
seed_column_metadata_batch(
|
||||
cursor,
|
||||
table_name="taxid_whitelist",
|
||||
columns=_TAXID_WHITELIST_COLUMNS,
|
||||
valid_column_names=[str(c["column_name"]) for c in _TAXID_WHITELIST_COLUMNS],
|
||||
)
|
||||
|
||||
|
||||
def _init_holiday_calendar_metadata(cursor) -> None:
|
||||
cursor.execute(_HOLIDAY_CALENDAR_DDL)
|
||||
init_updated_at_trigger(cursor, "holiday_calendar")
|
||||
upsert_table_metadata(
|
||||
cursor,
|
||||
table_name="holiday_calendar",
|
||||
display_name="节假日",
|
||||
description="节假日开票风控所用的日历数据（周末之外的法定节假日）。",
|
||||
grid_help="以国务院办公厅当年通知为准，请管理员按通知维护连休范围。",
|
||||
visible=1,
|
||||
readonly=0,
|
||||
)
|
||||
seed_column_metadata_batch(
|
||||
cursor,
|
||||
table_name="holiday_calendar",
|
||||
columns=_HOLIDAY_CALENDAR_COLUMNS,
|
||||
valid_column_names=[str(c["column_name"]) for c in _HOLIDAY_CALENDAR_COLUMNS],
|
||||
)
|
||||
|
||||
|
||||
def init_db() -> dict[str, int]:
|
||||
"""幂等创建/迁移本地 SQLite、数据管理元数据与业务种子，返回种子统计。"""
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
init_display_metadata(cur)
|
||||
_init_invoice_records_metadata(cur)
|
||||
_init_prereview_reports_metadata(cur)
|
||||
_init_expense_standards_metadata(cur)
|
||||
_init_taxid_whitelist_metadata(cur)
|
||||
_init_holiday_calendar_metadata(cur)
|
||||
seeded = seed_all(cur)
|
||||
conn.commit()
|
||||
return seeded
|
||||
finally:
|
||||
conn.close()
|
||||
324
scripts/db/display_metadata.py
Normal file
324
scripts/db/display_metadata.py
Normal file
@@ -0,0 +1,324 @@
|
||||
"""匠厂数据管理展示元数据（_jiangchang_tables / _jiangchang_columns）。
|
||||
|
||||
SQLite 无原生表/字段 COMMENT；中文展示名称通过这两张元数据表提供给宿主「数据管理」页面。
|
||||
字段在界面上的顺序由 CREATE TABLE / PRAGMA table_info(cid) 决定，**不**使用 display_order 排序。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Iterable
|
||||
|
||||
# 宿主会隐藏这些表，不作为业务表展示（见 jiangchang electron/data-management/types.ts）
|
||||
INTERNAL_METADATA_TABLES = frozenset({"_jiangchang_tables", "_jiangchang_columns"})
|
||||
LEGACY_METADATA_TABLES = frozenset({"_schema_meta"})
|
||||
METADATA_TABLES = INTERNAL_METADATA_TABLES | LEGACY_METADATA_TABLES
|
||||
|
||||
DATETIME_UNIX_SECONDS = "datetime_unix_seconds"
|
||||
STANDARD_TIMESTAMP_COLUMNS = frozenset({"created_at", "updated_at"})
|
||||
|
||||
_JIANGCHANG_TABLES_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS _jiangchang_tables (
|
||||
table_name TEXT PRIMARY KEY,
|
||||
display_name TEXT NOT NULL,
|
||||
description TEXT,
|
||||
grid_help TEXT,
|
||||
sort_order INTEGER,
|
||||
visible INTEGER NOT NULL DEFAULT 1,
|
||||
readonly INTEGER NOT NULL DEFAULT 0,
|
||||
default_sort_json TEXT,
|
||||
default_visible_columns_json TEXT
|
||||
)
|
||||
"""
|
||||
|
||||
_JIANGCHANG_COLUMNS_DDL = """
|
||||
CREATE TABLE IF NOT EXISTS _jiangchang_columns (
|
||||
table_name TEXT NOT NULL,
|
||||
column_name TEXT NOT NULL,
|
||||
display_name TEXT NOT NULL,
|
||||
description TEXT,
|
||||
grid_help TEXT,
|
||||
display_order INTEGER,
|
||||
visible INTEGER NOT NULL DEFAULT 1,
|
||||
searchable INTEGER NOT NULL DEFAULT 1,
|
||||
editable INTEGER NOT NULL DEFAULT 1,
|
||||
display_type TEXT,
|
||||
options_json TEXT,
|
||||
PRIMARY KEY (table_name, column_name)
|
||||
)
|
||||
"""
|
||||
|
||||
def _pragma_column_names(cursor, table_name: str) -> set[str]:
|
||||
cursor.execute(f"PRAGMA table_info({table_name!r})")
|
||||
return {str(row[1]) for row in cursor.fetchall()}
|
||||
|
||||
|
||||
def _ensure_jiangchang_tables_columns(cursor) -> None:
|
||||
"""旧库补列：CREATE IF NOT EXISTS 不会给已有表加新字段。"""
|
||||
cols = _pragma_column_names(cursor, "_jiangchang_tables")
|
||||
if "default_sort_json" not in cols:
|
||||
cursor.execute("ALTER TABLE _jiangchang_tables ADD COLUMN default_sort_json TEXT")
|
||||
if "default_visible_columns_json" not in cols:
|
||||
cursor.execute(
|
||||
"ALTER TABLE _jiangchang_tables ADD COLUMN default_visible_columns_json TEXT"
|
||||
)
|
||||
if "grid_help" not in cols:
|
||||
cursor.execute("ALTER TABLE _jiangchang_tables ADD COLUMN grid_help TEXT")
|
||||
|
||||
|
||||
def _ensure_jiangchang_columns_columns(cursor) -> None:
|
||||
cols = _pragma_column_names(cursor, "_jiangchang_columns")
|
||||
if "grid_help" not in cols:
|
||||
cursor.execute("ALTER TABLE _jiangchang_columns ADD COLUMN grid_help TEXT")
|
||||
|
||||
|
||||
def init_display_metadata_tables(cursor) -> None:
|
||||
cursor.execute(_JIANGCHANG_TABLES_DDL)
|
||||
cursor.execute(_JIANGCHANG_COLUMNS_DDL)
|
||||
_ensure_jiangchang_tables_columns(cursor)
|
||||
_ensure_jiangchang_columns_columns(cursor)
|
||||
|
||||
|
||||
def upsert_table_metadata(
|
||||
cursor,
|
||||
*,
|
||||
table_name: str,
|
||||
display_name: str,
|
||||
description: str | None = None,
|
||||
grid_help: str | None = None,
|
||||
visible: int = 1,
|
||||
readonly: int = 0,
|
||||
default_sort_json: str | None = None,
|
||||
default_visible_columns_json: str | None = None,
|
||||
) -> None:
|
||||
"""写入表级元数据。
|
||||
|
||||
default_sort_json: 宿主数据管理默认排序，JSON 数组，例如
|
||||
'[{"column":"sent_at","direction":"desc"},{"column":"id","direction":"desc"}]'
|
||||
为空则宿主回退到 updated_at → created_at → 主键。
|
||||
|
||||
default_visible_columns_json: 宿主数据管理默认展示列，JSON 字符串数组，例如
|
||||
'["stat_type","period","stage"]'
|
||||
为空则全显；用户改过「列」后以本机偏好为准（软默认，区别于 visible=0 硬隐藏）。
|
||||
|
||||
grid_help: 数据管理表格上方说明；空则不出横幅。与表单用的 description 独立，禁止互拷。
|
||||
"""
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO _jiangchang_tables (
|
||||
table_name, display_name, description, grid_help, visible, readonly,
|
||||
default_sort_json, default_visible_columns_json
|
||||
)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(table_name) DO UPDATE SET
|
||||
display_name = excluded.display_name,
|
||||
description = excluded.description,
|
||||
grid_help = excluded.grid_help,
|
||||
visible = excluded.visible,
|
||||
readonly = excluded.readonly,
|
||||
default_sort_json = excluded.default_sort_json,
|
||||
default_visible_columns_json = excluded.default_visible_columns_json
|
||||
""",
|
||||
(
|
||||
table_name,
|
||||
display_name,
|
||||
description,
|
||||
grid_help,
|
||||
visible,
|
||||
readonly,
|
||||
default_sort_json,
|
||||
default_visible_columns_json,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
def upsert_column_metadata(
|
||||
cursor,
|
||||
*,
|
||||
table_name: str,
|
||||
column_name: str,
|
||||
display_name: str,
|
||||
description: str | None = None,
|
||||
grid_help: str | None = None,
|
||||
visible: int = 1,
|
||||
searchable: int = 1,
|
||||
editable: int = 1,
|
||||
display_type: str | None = None,
|
||||
options_json: str | None = None,
|
||||
) -> None:
|
||||
# display_order 保留为 NULL：宿主按 PRAGMA table_info cid 展示字段，不读 display_order。
|
||||
# grid_help: 表头「?」；空则不出。与表单用的 description 独立，禁止互拷。
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO _jiangchang_columns (
|
||||
table_name, column_name, display_name, description, grid_help,
|
||||
display_order, visible, searchable, editable, display_type, options_json
|
||||
)
|
||||
VALUES (?, ?, ?, ?, ?, NULL, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(table_name, column_name) DO UPDATE SET
|
||||
display_name = excluded.display_name,
|
||||
description = excluded.description,
|
||||
grid_help = excluded.grid_help,
|
||||
display_order = NULL,
|
||||
visible = excluded.visible,
|
||||
searchable = excluded.searchable,
|
||||
editable = excluded.editable,
|
||||
display_type = excluded.display_type,
|
||||
options_json = excluded.options_json
|
||||
""",
|
||||
(
|
||||
table_name,
|
||||
column_name,
|
||||
display_name,
|
||||
description,
|
||||
grid_help,
|
||||
visible,
|
||||
searchable,
|
||||
editable,
|
||||
display_type,
|
||||
options_json,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
def seed_column_metadata_batch(
|
||||
cursor,
|
||||
*,
|
||||
table_name: str,
|
||||
columns: Iterable[dict[str, object]],
|
||||
valid_column_names: Iterable[str] | None = None,
|
||||
) -> None:
|
||||
"""批量 upsert 字段元数据；传入 valid_column_names 时会在 upsert 后清理陈旧字段元数据。"""
|
||||
for column in columns:
|
||||
upsert_column_metadata(
|
||||
cursor,
|
||||
table_name=table_name,
|
||||
column_name=str(column["column_name"]),
|
||||
display_name=str(column["display_name"]),
|
||||
description=str(column["description"]) if column.get("description") else None,
|
||||
grid_help=str(column["grid_help"]) if column.get("grid_help") else None,
|
||||
visible=int(column.get("visible", 1)),
|
||||
searchable=int(column.get("searchable", 1)),
|
||||
editable=int(column.get("editable", 1)),
|
||||
display_type=str(column["display_type"]) if column.get("display_type") else None,
|
||||
options_json=str(column["options_json"]) if column.get("options_json") else None,
|
||||
)
|
||||
if valid_column_names is not None:
|
||||
prune_stale_column_metadata(cursor, table_name, valid_column_names)
|
||||
|
||||
|
||||
def prune_stale_column_metadata(cursor, table_name: str, valid_column_names: Iterable[str]) -> None:
|
||||
"""删除表中已不存在物理字段的元数据行（幂等初始化后不得残留旧字段）。"""
|
||||
valid = set(valid_column_names)
|
||||
cursor.execute(
|
||||
"SELECT column_name FROM _jiangchang_columns WHERE table_name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
for (column_name,) in cursor.fetchall():
|
||||
if str(column_name) not in valid:
|
||||
cursor.execute(
|
||||
"DELETE FROM _jiangchang_columns WHERE table_name = ? AND column_name = ?",
|
||||
(table_name, column_name),
|
||||
)
|
||||
|
||||
|
||||
def get_table_readonly(cursor, table_name: str) -> int | None:
|
||||
cursor.execute(
|
||||
"SELECT readonly FROM _jiangchang_tables WHERE table_name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
row = cursor.fetchone()
|
||||
if not row:
|
||||
return None
|
||||
return int(row[0] or 0)
|
||||
|
||||
|
||||
def get_column_editable(cursor, table_name: str, column_name: str) -> int | None:
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT editable FROM _jiangchang_columns
|
||||
WHERE table_name = ? AND column_name = ?
|
||||
""",
|
||||
(table_name, column_name),
|
||||
)
|
||||
row = cursor.fetchone()
|
||||
if not row:
|
||||
return None
|
||||
return int(row[0] or 0)
|
||||
|
||||
|
||||
def list_editable_business_columns(cursor, table_name: str) -> list[str]:
|
||||
"""返回 editable=1 的业务字段名（不含宿主侧主键/generated 等二次过滤）。"""
|
||||
physical = list_physical_column_names(cursor, table_name)
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT column_name, editable
|
||||
FROM _jiangchang_columns
|
||||
WHERE table_name = ?
|
||||
""",
|
||||
(table_name,),
|
||||
)
|
||||
editable_set = {
|
||||
str(column_name)
|
||||
for column_name, editable in cursor.fetchall()
|
||||
if editable is not None and int(editable) == 1
|
||||
}
|
||||
return [name for name in physical if name in editable_set]
|
||||
|
||||
|
||||
def table_allows_manual_create(cursor, table_name: str) -> bool:
|
||||
"""表级 readonly=0 且至少有一个 editable=1 字段时，宿主才应提供空白新增表单。"""
|
||||
readonly = get_table_readonly(cursor, table_name)
|
||||
if readonly is None or int(readonly) != 0:
|
||||
return False
|
||||
return bool(list_editable_business_columns(cursor, table_name))
|
||||
|
||||
|
||||
def init_display_metadata(cursor) -> None:
|
||||
init_display_metadata_tables(cursor)
|
||||
|
||||
|
||||
def drop_deprecated_user_table(
|
||||
cursor,
|
||||
table_name: str,
|
||||
*,
|
||||
triggers: Iterable[str] | None = None,
|
||||
) -> None:
|
||||
"""永久退役用户可见表：删 trigger、DROP 物理表、清理展示元数据（幂等）。
|
||||
|
||||
用于废弃 task_logs 等旧表。禁止只删元数据而保留物理表。
|
||||
见 references/SCHEMA.md「表增删改与退役」。
|
||||
"""
|
||||
for trigger_name in triggers or ():
|
||||
cursor.execute(f"DROP TRIGGER IF EXISTS {trigger_name}")
|
||||
default_updated_at = f"{table_name}_set_updated_at"
|
||||
cursor.execute(f"DROP TRIGGER IF EXISTS {default_updated_at}")
|
||||
cursor.execute(f"DROP TABLE IF EXISTS {_quote_identifier(table_name)}")
|
||||
cursor.execute(
|
||||
"DELETE FROM _jiangchang_columns WHERE table_name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
cursor.execute(
|
||||
"DELETE FROM _jiangchang_tables WHERE table_name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
|
||||
|
||||
def iter_user_tables(cursor) -> Iterable[str]:
|
||||
cursor.execute(
|
||||
"SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%' ORDER BY name"
|
||||
)
|
||||
for (name,) in cursor.fetchall():
|
||||
if name in METADATA_TABLES:
|
||||
continue
|
||||
yield str(name)
|
||||
|
||||
|
||||
def _quote_identifier(name: str) -> str:
|
||||
return '"' + name.replace('"', '""') + '"'
|
||||
|
||||
|
||||
def list_physical_column_names(cursor, table_name: str) -> list[str]:
|
||||
cursor.execute(f"PRAGMA table_info({_quote_identifier(table_name)})")
|
||||
rows = cursor.fetchall()
|
||||
rows.sort(key=lambda row: int(row[0]))
|
||||
return [str(row[1]) for row in rows]
|
||||
592
scripts/db/display_metadata_validator.py
Normal file
592
scripts/db/display_metadata_validator.py
Normal file
@@ -0,0 +1,592 @@
|
||||
"""数据管理展示元数据校验（模板开发自检，不依赖宿主）。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
import sqlite3
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
from db.display_metadata import (
|
||||
DATETIME_UNIX_SECONDS,
|
||||
METADATA_TABLES,
|
||||
STANDARD_TIMESTAMP_COLUMNS,
|
||||
iter_user_tables,
|
||||
list_editable_business_columns,
|
||||
list_physical_column_names,
|
||||
table_allows_manual_create,
|
||||
)
|
||||
from db.timestamp_columns import has_updated_at_trigger
|
||||
from util.constants import SKILL_SLUG
|
||||
from util.slug_naming import LEGACY_EXEMPT_SLUGS, validate_slug_semantics
|
||||
|
||||
_TEMPLATE_PLACEHOLDER_SLUGS = frozenset({"your-skill-slug", "your_skill_slug"})
|
||||
_ENGLISH_PLACEHOLDER_NAME_PATTERNS = (
|
||||
re.compile(r"^my\s+skill$", re.I),
|
||||
re.compile(r"^your[-_\s]?skill", re.I),
|
||||
re.compile(r"^skill[-_\s]?template$", re.I),
|
||||
re.compile(r"^[a-z0-9_-]+$"),
|
||||
)
|
||||
_KEBAB_SLUG = re.compile(r"^[a-z0-9]+(?:-[a-z0-9]+)*$")
|
||||
_CJK_RE = re.compile(r"[\u4e00-\u9fff]")
|
||||
_FK_DDL_PATTERN = re.compile(r"\bFOREIGN\s+KEY\b|\bREFERENCES\b", re.I)
|
||||
|
||||
|
||||
@dataclass
|
||||
class ValidationReport:
|
||||
errors: list[str] = field(default_factory=list)
|
||||
warnings: list[str] = field(default_factory=list)
|
||||
|
||||
@property
|
||||
def ok(self) -> bool:
|
||||
return not self.errors
|
||||
|
||||
|
||||
def _has_cjk(text: str) -> bool:
|
||||
return bool(_CJK_RE.search(text))
|
||||
|
||||
|
||||
def _looks_like_raw_physical_name(display_name: str, physical_name: str) -> bool:
|
||||
if not display_name.strip():
|
||||
return True
|
||||
if display_name.strip().lower() == physical_name.strip().lower():
|
||||
return True
|
||||
compact = display_name.replace(" ", "").replace("_", "").lower()
|
||||
if compact == physical_name.replace("_", "").lower():
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def _parse_frontmatter_field(text: str, field_name: str) -> str:
|
||||
parts = text.split("---", 2)
|
||||
if len(parts) < 2:
|
||||
return ""
|
||||
for line in parts[1].splitlines():
|
||||
match = re.match(rf"^{re.escape(field_name)}:\s*(.+)\s*$", line.strip())
|
||||
if match:
|
||||
return match.group(1).strip().strip("\"'")
|
||||
return ""
|
||||
|
||||
|
||||
def _parse_openclaw_slug(text: str) -> str:
|
||||
parts = text.split("---", 2)
|
||||
if len(parts) < 2:
|
||||
return ""
|
||||
lines = parts[1].splitlines()
|
||||
for index, line in enumerate(lines):
|
||||
if re.match(r"^\s+openclaw:\s*$", line):
|
||||
for inner in lines[index + 1 :]:
|
||||
if inner.strip() and inner[0] not in (" ", "\t"):
|
||||
break
|
||||
match = re.match(r"^(\s+)slug:\s*(.+)\s*$", inner)
|
||||
if match:
|
||||
return match.group(2).strip().strip("\"'")
|
||||
return ""
|
||||
|
||||
|
||||
def parse_skill_frontmatter(skill_root: str) -> dict[str, str]:
|
||||
path = os.path.join(skill_root, "SKILL.md")
|
||||
with open(path, encoding="utf-8") as f:
|
||||
text = f.read()
|
||||
return {
|
||||
"name": _parse_frontmatter_field(text, "name"),
|
||||
"slug": _parse_openclaw_slug(text),
|
||||
}
|
||||
|
||||
|
||||
def validate_skill_frontmatter(skill_root: str, *, allow_template_placeholders: bool = True) -> ValidationReport:
|
||||
report = ValidationReport()
|
||||
data = parse_skill_frontmatter(skill_root)
|
||||
name = str(data.get("name") or "").strip()
|
||||
slug = str(data.get("slug") or "").strip()
|
||||
|
||||
if not name:
|
||||
report.errors.append("SKILL.md frontmatter.name 缺失")
|
||||
elif not _has_cjk(name):
|
||||
if allow_template_placeholders and slug in _TEMPLATE_PLACEHOLDER_SLUGS:
|
||||
report.warnings.append("SKILL.md frontmatter.name 建议使用清晰的中文业务名称（模板占位可暂保留）")
|
||||
else:
|
||||
report.errors.append("SKILL.md frontmatter.name 应使用用户可见的中文业务名称")
|
||||
else:
|
||||
for pattern in _ENGLISH_PLACEHOLDER_NAME_PATTERNS:
|
||||
if pattern.fullmatch(name):
|
||||
if allow_template_placeholders and slug in _TEMPLATE_PLACEHOLDER_SLUGS:
|
||||
report.warnings.append(f"SKILL.md frontmatter.name 仍为占位风格：{name!r}")
|
||||
else:
|
||||
report.errors.append(f"SKILL.md frontmatter.name 不应为英文占位名：{name!r}")
|
||||
break
|
||||
|
||||
if not slug:
|
||||
report.errors.append("SKILL.md metadata.openclaw.slug 缺失")
|
||||
elif not _KEBAB_SLUG.fullmatch(slug):
|
||||
report.errors.append(f"metadata.openclaw.slug 必须为 kebab-case：{slug!r}")
|
||||
elif slug in _TEMPLATE_PLACEHOLDER_SLUGS and not allow_template_placeholders:
|
||||
report.errors.append(f"metadata.openclaw.slug 仍为模板占位：{slug!r}")
|
||||
|
||||
constants_slug = SKILL_SLUG.strip()
|
||||
if slug and constants_slug and slug != constants_slug:
|
||||
report.warnings.append(
|
||||
f"SKILL.md slug ({slug!r}) 与 util.constants.SKILL_SLUG ({constants_slug!r}) 不一致"
|
||||
)
|
||||
|
||||
if slug and slug not in LEGACY_EXEMPT_SLUGS and _KEBAB_SLUG.fullmatch(slug):
|
||||
naming_errors, naming_warnings = validate_slug_semantics(
|
||||
slug,
|
||||
strict=not allow_template_placeholders,
|
||||
)
|
||||
report.errors.extend(naming_errors)
|
||||
report.warnings.extend(naming_warnings)
|
||||
|
||||
return report
|
||||
|
||||
|
||||
def _quote_identifier(name: str) -> str:
|
||||
return '"' + name.replace('"', '""') + '"'
|
||||
|
||||
|
||||
def _table_column_info(cursor, table_name: str) -> dict[str, dict[str, object]]:
|
||||
cursor.execute(f"PRAGMA table_info({_quote_identifier(table_name)})")
|
||||
info: dict[str, dict[str, object]] = {}
|
||||
for row in cursor.fetchall():
|
||||
info[str(row[1])] = {
|
||||
"cid": int(row[0]),
|
||||
"name": str(row[1]),
|
||||
"type": str(row[2] or ""),
|
||||
"notnull": int(row[3] or 0),
|
||||
"dflt_value": row[4],
|
||||
"pk": int(row[5] or 0),
|
||||
}
|
||||
return info
|
||||
|
||||
|
||||
def _table_column_xinfo(cursor, table_name: str) -> dict[str, dict[str, object]]:
|
||||
"""PRAGMA table_xinfo 可识别 generated/hidden 字段（SQLite 3.31+）。"""
|
||||
cursor.execute(f"PRAGMA table_xinfo({_quote_identifier(table_name)})")
|
||||
info: dict[str, dict[str, object]] = {}
|
||||
for row in cursor.fetchall():
|
||||
hidden = int(row[6] or 0) if len(row) > 6 else 0
|
||||
info[str(row[1])] = {
|
||||
"cid": int(row[0]),
|
||||
"name": str(row[1]),
|
||||
"type": str(row[2] or ""),
|
||||
"notnull": int(row[3] or 0),
|
||||
"dflt_value": row[4],
|
||||
"pk": int(row[5] or 0),
|
||||
"hidden": hidden,
|
||||
}
|
||||
return info
|
||||
|
||||
|
||||
def _is_generated_column(xinfo: dict[str, object]) -> bool:
|
||||
hidden = int(xinfo.get("hidden") or 0)
|
||||
return hidden in (2, 3)
|
||||
|
||||
|
||||
def _is_blob_column(column_info: dict[str, object]) -> bool:
|
||||
return "BLOB" in str(column_info.get("type") or "").upper()
|
||||
|
||||
|
||||
def _default_uses_unixepoch(default_value: object) -> bool:
|
||||
if default_value is None:
|
||||
return False
|
||||
text = str(default_value).strip().lower()
|
||||
return "unixepoch" in text
|
||||
|
||||
|
||||
def validate_options_json(options_json: str | None) -> tuple[list[str], list[str]]:
|
||||
"""校验 options_json：静态枚举数组，或 lookup 对象。"""
|
||||
errors: list[str] = []
|
||||
warnings: list[str] = []
|
||||
if options_json is None or str(options_json).strip() == "":
|
||||
return errors, warnings
|
||||
try:
|
||||
parsed = json.loads(str(options_json))
|
||||
except json.JSONDecodeError as exc:
|
||||
errors.append(f"options_json 不是合法 JSON：{exc}")
|
||||
return errors, warnings
|
||||
|
||||
if isinstance(parsed, dict):
|
||||
for key in ("ref_table", "value_column", "label_column"):
|
||||
if not str(parsed.get(key) or "").strip():
|
||||
errors.append(f"options_json lookup 必须包含非空 {key}")
|
||||
ident = re.compile(r"^[a-zA-Z_][a-zA-Z0-9_]*$")
|
||||
for key in ("ref_table", "value_column", "label_column", "filter_column"):
|
||||
raw = parsed.get(key)
|
||||
if raw is None or raw == "":
|
||||
continue
|
||||
if not ident.match(str(raw)):
|
||||
errors.append(f"options_json.{key} 必须是安全标识符，当前为 {raw!r}")
|
||||
if parsed.get("filter_column") not in (None, "") and "filter_value" not in parsed:
|
||||
warnings.append("options_json lookup 声明了 filter_column 但未提供 filter_value")
|
||||
if "multiple" in parsed and not isinstance(parsed.get("multiple"), bool):
|
||||
errors.append("options_json lookup.multiple 必须是布尔值")
|
||||
return errors, warnings
|
||||
|
||||
if not isinstance(parsed, list):
|
||||
errors.append("options_json 必须是 JSON 数组（静态枚举）或对象（lookup）")
|
||||
return errors, warnings
|
||||
for index, item in enumerate(parsed):
|
||||
if not isinstance(item, dict):
|
||||
errors.append(f"options_json[{index}] 必须是对象")
|
||||
continue
|
||||
if "value" not in item or "label" not in item:
|
||||
errors.append(f"options_json[{index}] 必须包含 value 与 label")
|
||||
return errors, warnings
|
||||
|
||||
|
||||
def validate_filter_search_metadata(
|
||||
cursor,
|
||||
table_name: str,
|
||||
) -> tuple[list[str], list[str]]:
|
||||
"""对可见但未开启 searchable 的枚举 / boolean 列给出软建议。"""
|
||||
warnings: list[str] = []
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT column_name, visible, searchable, display_type, options_json
|
||||
FROM _jiangchang_columns
|
||||
WHERE table_name = ?
|
||||
""",
|
||||
(table_name,),
|
||||
)
|
||||
for column_name, visible, searchable, display_type, options_json in cursor.fetchall():
|
||||
if visible is not None and int(visible) == 0:
|
||||
continue
|
||||
if searchable is not None and int(searchable) != 0:
|
||||
continue
|
||||
|
||||
display_type_norm = str(display_type or "").strip()
|
||||
is_boolean = display_type_norm == "boolean"
|
||||
|
||||
has_enum = False
|
||||
if options_json is not None and str(options_json).strip():
|
||||
opt_errors, _ = validate_options_json(options_json)
|
||||
if not opt_errors:
|
||||
try:
|
||||
parsed = json.loads(str(options_json))
|
||||
except json.JSONDecodeError:
|
||||
parsed = None
|
||||
has_enum = isinstance(parsed, list) and len(parsed) > 0
|
||||
|
||||
if not has_enum and not is_boolean:
|
||||
continue
|
||||
|
||||
warnings.append(
|
||||
f"{table_name}.{column_name}: 枚举/状态类列建议 searchable=1，"
|
||||
"以便宿主列筛选与按中文 label 全局搜索（见 references/SCHEMA.md）"
|
||||
)
|
||||
|
||||
return [], warnings
|
||||
|
||||
|
||||
def validate_column_permissions(
|
||||
cursor,
|
||||
table_name: str,
|
||||
*,
|
||||
physical_columns: list[str] | None = None,
|
||||
) -> tuple[list[str], list[str]]:
|
||||
"""校验常见系统字段 editable=0，并识别技术上可写但不适合手工新增的表。"""
|
||||
errors: list[str] = []
|
||||
warnings: list[str] = []
|
||||
ordered = physical_columns or list_physical_column_names(cursor, table_name)
|
||||
column_info = _table_column_info(cursor, table_name)
|
||||
column_xinfo = _table_column_xinfo(cursor, table_name)
|
||||
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT column_name, editable, options_json
|
||||
FROM _jiangchang_columns
|
||||
WHERE table_name = ?
|
||||
""",
|
||||
(table_name,),
|
||||
)
|
||||
meta_by_column = {
|
||||
str(row[0]): {"editable": row[1], "options_json": row[2]}
|
||||
for row in cursor.fetchall()
|
||||
}
|
||||
|
||||
for column_name, meta in meta_by_column.items():
|
||||
physical = column_info.get(column_name) or {}
|
||||
xinfo = column_xinfo.get(column_name) or physical
|
||||
if not physical and column_name not in column_xinfo:
|
||||
continue
|
||||
|
||||
pk = int(physical.get("pk") or xinfo.get("pk") or 0)
|
||||
editable = meta["editable"]
|
||||
if pk and editable is not None and int(editable) != 0:
|
||||
errors.append(f"{table_name}.{column_name} 是主键，editable 应为 0")
|
||||
|
||||
if column_name == "id" and editable is not None and int(editable) != 0:
|
||||
errors.append(f"{table_name}.id 的 editable 应为 0（自增主键）")
|
||||
|
||||
if _is_generated_column(xinfo) and editable is not None and int(editable) != 0:
|
||||
errors.append(f"{table_name}.{column_name} 是 generated 字段，editable 应为 0")
|
||||
|
||||
if _is_blob_column(physical or xinfo) and editable is not None and int(editable) != 0:
|
||||
errors.append(f"{table_name}.{column_name} 是 BLOB 字段，editable 应为 0")
|
||||
|
||||
opt_errors, _ = validate_options_json(meta.get("options_json"))
|
||||
for item in opt_errors:
|
||||
errors.append(f"{table_name}.{column_name}: {item}")
|
||||
|
||||
cursor.execute(
|
||||
"SELECT readonly FROM _jiangchang_tables WHERE table_name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
readonly_row = cursor.fetchone()
|
||||
readonly = int(readonly_row[0]) if readonly_row else None
|
||||
editable_columns = list_editable_business_columns(cursor, table_name)
|
||||
if readonly == 0 and not editable_columns:
|
||||
warnings.append(
|
||||
f"{table_name}: 表级 readonly=0 但所有字段 editable=0，"
|
||||
"宿主不应打开空白新增表单；数据应由技能业务流程产生"
|
||||
)
|
||||
if readonly == 1 and editable_columns:
|
||||
warnings.append(
|
||||
f"{table_name}: 表级 readonly=1 但存在 editable=1 字段，"
|
||||
"宿主应以表级只读为准"
|
||||
)
|
||||
|
||||
return errors, warnings
|
||||
|
||||
|
||||
def _expected_timestamp_display_name(column_name: str) -> str:
|
||||
if column_name == "created_at":
|
||||
return "创建时间"
|
||||
if column_name == "updated_at":
|
||||
return "更新时间"
|
||||
return ""
|
||||
|
||||
|
||||
def validate_standard_timestamp_columns(
|
||||
cursor,
|
||||
table_name: str,
|
||||
*,
|
||||
physical_columns: list[str] | None = None,
|
||||
) -> tuple[list[str], list[str]]:
|
||||
errors: list[str] = []
|
||||
warnings: list[str] = []
|
||||
ordered = physical_columns or list_physical_column_names(cursor, table_name)
|
||||
column_info = _table_column_info(cursor, table_name)
|
||||
|
||||
for column_name in STANDARD_TIMESTAMP_COLUMNS:
|
||||
if column_name not in ordered:
|
||||
continue
|
||||
|
||||
physical = column_info.get(column_name)
|
||||
if not physical:
|
||||
errors.append(f"{table_name}.{column_name} 缺少物理字段定义")
|
||||
continue
|
||||
|
||||
if "INT" not in str(physical["type"]).upper():
|
||||
errors.append(
|
||||
f"{table_name}.{column_name} 字段类型应为 INTEGER（Unix 秒级时间戳），当前为 {physical['type']!r}"
|
||||
)
|
||||
|
||||
if column_name == "created_at" and not _default_uses_unixepoch(physical["dflt_value"]):
|
||||
errors.append(
|
||||
f"{table_name}.created_at 应设置数据库默认值 DEFAULT (unixepoch())，当前默认值为 {physical['dflt_value']!r}"
|
||||
)
|
||||
if column_name == "updated_at" and not _default_uses_unixepoch(physical["dflt_value"]):
|
||||
errors.append(
|
||||
f"{table_name}.updated_at 应设置数据库初始默认值 DEFAULT (unixepoch())，当前默认值为 {physical['dflt_value']!r}"
|
||||
)
|
||||
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT display_name, display_type, editable, searchable, display_order
|
||||
FROM _jiangchang_columns
|
||||
WHERE table_name = ? AND column_name = ?
|
||||
""",
|
||||
(table_name, column_name),
|
||||
)
|
||||
meta_row = cursor.fetchone()
|
||||
if not meta_row:
|
||||
errors.append(f"{table_name}.{column_name} 缺少 _jiangchang_columns 元数据")
|
||||
continue
|
||||
|
||||
display_name, display_type, editable, searchable, display_order = meta_row
|
||||
expected_name = _expected_timestamp_display_name(column_name)
|
||||
if str(display_name or "").strip() != expected_name:
|
||||
errors.append(
|
||||
f"{table_name}.{column_name} 的 display_name 应为 {expected_name!r}，当前为 {display_name!r}"
|
||||
)
|
||||
if str(display_type or "").strip() != DATETIME_UNIX_SECONDS:
|
||||
errors.append(
|
||||
f"{table_name}.{column_name} 的 display_type 应为 {DATETIME_UNIX_SECONDS!r}，当前为 {display_type!r}"
|
||||
)
|
||||
if editable is None or int(editable) != 0:
|
||||
errors.append(f"{table_name}.{column_name} 的 editable 应为 0（系统时间字段不可编辑）")
|
||||
if display_order is not None:
|
||||
errors.append(f"{table_name}.{column_name} 的 display_order 应为 NULL")
|
||||
if searchable is not None and int(searchable) != 0:
|
||||
warnings.append(f"{table_name}.{column_name} 的时间字段通常不必开启 searchable")
|
||||
|
||||
if "created_at" in ordered and "updated_at" in ordered:
|
||||
if ordered.index("updated_at") < ordered.index("created_at"):
|
||||
warnings.append(f"{table_name}: updated_at 不应出现在 created_at 之前")
|
||||
|
||||
if "updated_at" in ordered and not has_updated_at_trigger(cursor, table_name):
|
||||
errors.append(
|
||||
f"{table_name}.updated_at 缺少自动维护 trigger（期望 {table_name}_set_updated_at）"
|
||||
)
|
||||
|
||||
return errors, warnings
|
||||
|
||||
|
||||
def column_order_warnings(table_name: str, ordered_names: list[str]) -> list[str]:
|
||||
warnings: list[str] = []
|
||||
if "id" in ordered_names and ordered_names.index("id") != 0:
|
||||
warnings.append(f"{table_name}: 字段 id 通常应定义在 CREATE TABLE 的第一列")
|
||||
if "created_at" in ordered_names:
|
||||
idx = ordered_names.index("created_at")
|
||||
trailing = ordered_names[idx + 1 :]
|
||||
if trailing and trailing != ["updated_at"]:
|
||||
warnings.append(
|
||||
f"{table_name}: created_at 之后通常只应跟 updated_at，不应再放置其他业务字段"
|
||||
)
|
||||
if "updated_at" in ordered_names and "created_at" in ordered_names:
|
||||
if ordered_names.index("updated_at") < ordered_names.index("created_at"):
|
||||
warnings.append(f"{table_name}: updated_at 不应出现在 created_at 之前")
|
||||
return warnings
|
||||
|
||||
|
||||
def _table_exists(cursor, table_name: str) -> bool:
|
||||
cursor.execute(
|
||||
"SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = ? LIMIT 1",
|
||||
(table_name,),
|
||||
)
|
||||
return cursor.fetchone() is not None
|
||||
|
||||
|
||||
def validate_no_sqlite_foreign_keys(cursor) -> tuple[list[str], list[str]]:
|
||||
"""业务表 DDL 不得含 FOREIGN KEY / REFERENCES（关联走 repository/service）。"""
|
||||
errors: list[str] = []
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT name, sql FROM sqlite_master
|
||||
WHERE type = 'table' AND name NOT LIKE 'sqlite_%'
|
||||
"""
|
||||
)
|
||||
for table_name, sql in cursor.fetchall():
|
||||
if table_name in METADATA_TABLES:
|
||||
continue
|
||||
ddl = str(sql or "")
|
||||
if ddl and _FK_DDL_PATTERN.search(ddl):
|
||||
errors.append(
|
||||
f"业务表 {table_name!r} 的 DDL 含 SQLite FOREIGN KEY/REFERENCES；"
|
||||
"请改为业务 id 列 + repository/service 逻辑关联（见 references/SCHEMA.md）"
|
||||
)
|
||||
return errors, []
|
||||
|
||||
|
||||
def validate_display_metadata(conn: sqlite3.Connection, *, check_column_order: bool = True) -> ValidationReport:
|
||||
report = ValidationReport()
|
||||
cur = conn.cursor()
|
||||
|
||||
has_tables_meta = _table_exists(cur, "_jiangchang_tables")
|
||||
has_columns_meta = _table_exists(cur, "_jiangchang_columns")
|
||||
if not has_tables_meta:
|
||||
report.errors.append("缺少数据管理元数据表 _jiangchang_tables")
|
||||
if not has_columns_meta:
|
||||
report.errors.append("缺少数据管理元数据表 _jiangchang_columns")
|
||||
if not has_tables_meta or not has_columns_meta:
|
||||
return report
|
||||
|
||||
fk_errors, _ = validate_no_sqlite_foreign_keys(cur)
|
||||
report.errors.extend(fk_errors)
|
||||
|
||||
for table_name in iter_user_tables(cur):
|
||||
cur.execute(
|
||||
"SELECT display_name FROM _jiangchang_tables WHERE table_name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
if not row or not str(row[0] or "").strip():
|
||||
report.errors.append(f"业务表 {table_name!r} 缺少 _jiangchang_tables.display_name")
|
||||
elif not _has_cjk(str(row[0])):
|
||||
report.errors.append(f"业务表 {table_name!r} 的 display_name 应使用中文业务名称")
|
||||
|
||||
physical_columns = list_physical_column_names(cur, table_name)
|
||||
if check_column_order:
|
||||
report.warnings.extend(column_order_warnings(table_name, physical_columns))
|
||||
|
||||
for column_name in physical_columns:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT display_name, visible
|
||||
FROM _jiangchang_columns
|
||||
WHERE table_name = ? AND column_name = ?
|
||||
""",
|
||||
(table_name, column_name),
|
||||
)
|
||||
col_row = cur.fetchone()
|
||||
if not col_row:
|
||||
report.errors.append(
|
||||
f"字段 {table_name}.{column_name} 缺少 _jiangchang_columns 元数据"
|
||||
)
|
||||
continue
|
||||
display_name = str(col_row[0] or "").strip()
|
||||
visible = col_row[1] is None or int(col_row[1]) != 0
|
||||
if visible and not display_name:
|
||||
report.errors.append(f"字段 {table_name}.{column_name} 的 display_name 为空")
|
||||
elif visible and _looks_like_raw_physical_name(display_name, column_name):
|
||||
report.errors.append(
|
||||
f"字段 {table_name}.{column_name} 的 display_name 不应直接等于英文物理名：{display_name!r}"
|
||||
)
|
||||
elif visible and not _has_cjk(display_name):
|
||||
report.errors.append(
|
||||
f"字段 {table_name}.{column_name} 的 display_name 应使用中文：{display_name!r}"
|
||||
)
|
||||
|
||||
ts_errors, ts_warnings = validate_standard_timestamp_columns(
|
||||
cur,
|
||||
table_name,
|
||||
physical_columns=physical_columns,
|
||||
)
|
||||
report.errors.extend(ts_errors)
|
||||
report.warnings.extend(ts_warnings)
|
||||
|
||||
perm_errors, perm_warnings = validate_column_permissions(
|
||||
cur,
|
||||
table_name,
|
||||
physical_columns=physical_columns,
|
||||
)
|
||||
report.errors.extend(perm_errors)
|
||||
report.warnings.extend(perm_warnings)
|
||||
|
||||
_, filter_warnings = validate_filter_search_metadata(cur, table_name)
|
||||
report.warnings.extend(filter_warnings)
|
||||
|
||||
cur.execute("SELECT table_name, column_name FROM _jiangchang_columns")
|
||||
for table_name, column_name in cur.fetchall():
|
||||
if table_name in METADATA_TABLES:
|
||||
continue
|
||||
cur.execute(
|
||||
"SELECT 1 FROM sqlite_master WHERE type='table' AND name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
if not cur.fetchone():
|
||||
report.errors.append(f"元数据引用了不存在的表：{table_name!r}")
|
||||
continue
|
||||
physical = list_physical_column_names(cur, str(table_name))
|
||||
if str(column_name) not in physical:
|
||||
report.errors.append(f"元数据引用了不存在的字段：{table_name}.{column_name}")
|
||||
|
||||
cur.execute("SELECT table_name FROM _jiangchang_tables")
|
||||
for (table_name,) in cur.fetchall():
|
||||
if table_name in METADATA_TABLES:
|
||||
continue
|
||||
cur.execute(
|
||||
"SELECT 1 FROM sqlite_master WHERE type='table' AND name = ?",
|
||||
(table_name,),
|
||||
)
|
||||
if not cur.fetchone():
|
||||
report.errors.append(f"元数据引用了不存在的表：{table_name!r}")
|
||||
|
||||
return report
|
||||
|
||||
|
||||
def validate_template_database(conn: sqlite3.Connection) -> ValidationReport:
|
||||
"""模板自检入口：校验展示元数据完整性，并扫描业务表 DDL 是否含 SQLite 外键。"""
|
||||
return validate_display_metadata(conn)
|
||||
193
scripts/db/seed_data.py
Normal file
193
scripts/db/seed_data.py
Normal file
@@ -0,0 +1,193 @@
|
||||
"""业务种子数据：经费标准库、税号白名单、节假日。
|
||||
|
||||
init-db 幂等写入（存在即跳过，不覆盖管理员修改）。
|
||||
标准数值为示例种子，实施时以学校现行《差旅费管理办法》等文件为准，
|
||||
可在数据管理中直接编辑；节假日以国务院办公厅当年通知为准。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import sqlite3
|
||||
|
||||
from util.timeutil import now_unix
|
||||
|
||||
# (category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause)
|
||||
_SEED_STANDARDS: tuple[tuple[str, str, str, str, float, str, str, str], ...] = (
|
||||
(
|
||||
"住宿费",
|
||||
"一线城市",
|
||||
"北京,上海,广州,深圳",
|
||||
"副高及以上",
|
||||
550.0,
|
||||
"元/人/天",
|
||||
"《差旅费管理办法》",
|
||||
"第十条",
|
||||
),
|
||||
(
|
||||
"住宿费",
|
||||
"一线城市",
|
||||
"北京,上海,广州,深圳",
|
||||
"中级及以下",
|
||||
480.0,
|
||||
"元/人/天",
|
||||
"《差旅费管理办法》",
|
||||
"第十条",
|
||||
),
|
||||
(
|
||||
"住宿费",
|
||||
"省会及计划单列市",
|
||||
"哈尔滨,长春,沈阳,石家庄,太原,西安,济南,杭州,郑州,合肥,南昌,福州,长沙,武汉,海口,贵阳,成都,昆明,西宁,兰州,台北,呼和浩特,南宁,拉萨,银川,乌鲁木齐,大连,青岛,宁波,厦门",
|
||||
"副高及以上",
|
||||
480.0,
|
||||
"元/人/天",
|
||||
"《差旅费管理办法》",
|
||||
"第十条",
|
||||
),
|
||||
(
|
||||
"住宿费",
|
||||
"省会及计划单列市",
|
||||
"哈尔滨,长春,沈阳,石家庄,太原,西安,济南,杭州,郑州,合肥,南昌,福州,长沙,武汉,海口,贵阳,成都,昆明,西宁,兰州,台北,呼和浩特,南宁,拉萨,银川,乌鲁木齐,大连,青岛,宁波,厦门",
|
||||
"中级及以下",
|
||||
400.0,
|
||||
"元/人/天",
|
||||
"《差旅费管理办法》",
|
||||
"第十条",
|
||||
),
|
||||
(
|
||||
"住宿费",
|
||||
"其他地区",
|
||||
"",
|
||||
"副高及以上",
|
||||
400.0,
|
||||
"元/人/天",
|
||||
"《差旅费管理办法》",
|
||||
"第十条",
|
||||
),
|
||||
(
|
||||
"住宿费",
|
||||
"其他地区",
|
||||
"",
|
||||
"中级及以下",
|
||||
350.0,
|
||||
"元/人/天",
|
||||
"《差旅费管理办法》",
|
||||
"第十条",
|
||||
),
|
||||
(
|
||||
"交通费",
|
||||
"通用",
|
||||
"",
|
||||
"副高及以上",
|
||||
2.0,
|
||||
"席位等级代码（1=二等座/硬席/经济舱，2=一等座/软卧/公务舱，3=商务座/头等舱）",
|
||||
"《差旅费管理办法》",
|
||||
"第十二条",
|
||||
),
|
||||
(
|
||||
"交通费",
|
||||
"通用",
|
||||
"",
|
||||
"中级及以下",
|
||||
1.0,
|
||||
"席位等级代码（1=二等座/硬席/经济舱，2=一等座/软卧/公务舱，3=商务座/头等舱）",
|
||||
"《差旅费管理办法》",
|
||||
"第十二条",
|
||||
),
|
||||
)
|
||||
|
||||
# 校方购买方税号白名单种子（示例，实施时由财务处维护）
|
||||
_SEED_WHITELIST: tuple[tuple[str, str, str], ...] = (
|
||||
(
|
||||
"91440101MA9XXYYY0A",
|
||||
"广州华商学院",
|
||||
"示例种子税号；上线前请由财务处核准并替换为真实统一社会信用代码",
|
||||
),
|
||||
)
|
||||
|
||||
# 节假日种子（主体日期；连休范围请管理员按当年国务院办公厅通知补全）
|
||||
_SEED_HOLIDAYS: tuple[tuple[str, str], ...] = (
|
||||
("2026-01-01", "元旦"),
|
||||
("2026-02-16", "春节（除夕）"),
|
||||
("2026-02-17", "春节"),
|
||||
("2026-02-18", "春节"),
|
||||
("2026-02-19", "春节"),
|
||||
("2026-04-05", "清明节"),
|
||||
("2026-05-01", "劳动节"),
|
||||
("2026-06-19", "端午节"),
|
||||
("2026-09-25", "中秋节"),
|
||||
("2026-10-01", "国庆节"),
|
||||
("2026-10-02", "国庆节"),
|
||||
("2026-10-03", "国庆节"),
|
||||
)
|
||||
|
||||
|
||||
def seed_expense_standards(cursor: sqlite3.Cursor) -> int:
|
||||
"""幂等写入经费标准种子，返回新增条数。"""
|
||||
inserted = 0
|
||||
for row in _SEED_STANDARDS:
|
||||
category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause = row
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT COUNT(*) FROM expense_standards
|
||||
WHERE category = ? AND city_tier = ? AND staff_title = ?
|
||||
""",
|
||||
(category, city_tier, staff_title),
|
||||
)
|
||||
if int(cursor.fetchone()[0]) > 0:
|
||||
continue
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO expense_standards
|
||||
(category, city_tier, city_example, staff_title, max_amount, unit,
|
||||
basis_doc, clause, effective_date, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, NULL, ?, ?)
|
||||
""",
|
||||
(category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause, now_unix(), now_unix()),
|
||||
)
|
||||
inserted += 1
|
||||
return inserted
|
||||
|
||||
|
||||
def seed_taxid_whitelist(cursor: sqlite3.Cursor) -> int:
|
||||
"""幂等写入税号白名单种子，返回新增条数。"""
|
||||
inserted = 0
|
||||
for tax_no, org_name, note in _SEED_WHITELIST:
|
||||
cursor.execute("SELECT COUNT(*) FROM taxid_whitelist WHERE tax_no = ?", (tax_no,))
|
||||
if int(cursor.fetchone()[0]) > 0:
|
||||
continue
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO taxid_whitelist (tax_no, org_name, note, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
""",
|
||||
(tax_no, org_name, note, now_unix(), now_unix()),
|
||||
)
|
||||
inserted += 1
|
||||
return inserted
|
||||
|
||||
|
||||
def seed_holidays(cursor: sqlite3.Cursor) -> int:
|
||||
"""幂等写入节假日种子，返回新增条数。"""
|
||||
inserted = 0
|
||||
for holiday_date, holiday_name in _SEED_HOLIDAYS:
|
||||
cursor.execute("SELECT COUNT(*) FROM holiday_calendar WHERE holiday_date = ?", (holiday_date,))
|
||||
if int(cursor.fetchone()[0]) > 0:
|
||||
continue
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO holiday_calendar (holiday_date, holiday_name, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?)
|
||||
""",
|
||||
(holiday_date, holiday_name, now_unix(), now_unix()),
|
||||
)
|
||||
inserted += 1
|
||||
return inserted
|
||||
|
||||
|
||||
def seed_all(cursor: sqlite3.Cursor) -> dict[str, int]:
|
||||
"""写入全部业务种子，返回各类新增条数（供 init-db 输出与测试断言）。"""
|
||||
return {
|
||||
"expense_standards": seed_expense_standards(cursor),
|
||||
"taxid_whitelist": seed_taxid_whitelist(cursor),
|
||||
"holiday_calendar": seed_holidays(cursor),
|
||||
}
|
||||
50
scripts/db/timestamp_columns.py
Normal file
50
scripts/db/timestamp_columns.py
Normal file
@@ -0,0 +1,50 @@
|
||||
"""标准时间字段：Unix 秒级时间戳、默认值与 updated_at 自动维护。
|
||||
|
||||
业务表若包含 created_at / updated_at，在 CREATE TABLE 后调用
|
||||
``init_updated_at_trigger(cursor, table_name)``。模板默认不建业务表。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
STANDARD_TIMESTAMP_COLUMNS = frozenset({"created_at", "updated_at"})
|
||||
DATETIME_UNIX_SECONDS = "datetime_unix_seconds"
|
||||
|
||||
|
||||
def updated_at_trigger_name(table_name: str) -> str:
|
||||
return f"{table_name}_set_updated_at"
|
||||
|
||||
|
||||
def _quote_identifier(name: str) -> str:
|
||||
return '"' + name.replace('"', '""') + '"'
|
||||
|
||||
|
||||
def init_updated_at_trigger(cursor, table_name: str) -> None:
|
||||
"""幂等创建 {table}_set_updated_at：业务字段 UPDATE 后刷新 updated_at。"""
|
||||
trigger = updated_at_trigger_name(table_name)
|
||||
quoted = _quote_identifier(table_name)
|
||||
quoted_trigger = _quote_identifier(trigger)
|
||||
cursor.execute(
|
||||
f"""
|
||||
CREATE TRIGGER IF NOT EXISTS {quoted_trigger}
|
||||
AFTER UPDATE ON {quoted}
|
||||
FOR EACH ROW
|
||||
BEGIN
|
||||
UPDATE {quoted}
|
||||
SET updated_at = unixepoch()
|
||||
WHERE id = NEW.id AND updated_at = OLD.updated_at;
|
||||
END
|
||||
"""
|
||||
)
|
||||
|
||||
|
||||
def has_updated_at_trigger(cursor, table_name: str) -> bool:
|
||||
trigger_name = updated_at_trigger_name(table_name)
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT 1 FROM sqlite_master
|
||||
WHERE type = 'trigger' AND name = ? AND tbl_name = ?
|
||||
LIMIT 1
|
||||
""",
|
||||
(trigger_name, table_name),
|
||||
)
|
||||
return cursor.fetchone() is not None
|
||||
38
scripts/main.py
Normal file
38
scripts/main.py
Normal file
@@ -0,0 +1,38 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
skill-template CLI 入口。
|
||||
|
||||
复制为新技能后，请修改注释与常量，但保留当前分层结构：
|
||||
cli（argv）→ service（业务编排）→ db（持久化）→ util（通用工具）。
|
||||
|
||||
jiangchang_skill_core 来自宿主共享 Python runtime 安装的 jiangchang-platform-kit，
|
||||
不在技能仓库内 vendored。
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import sys
|
||||
|
||||
if sys.platform == "win32":
|
||||
import io
|
||||
|
||||
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", errors="replace")
|
||||
sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding="utf-8", errors="replace")
|
||||
|
||||
_scripts_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
if _scripts_dir not in sys.path:
|
||||
sys.path.insert(0, _scripts_dir)
|
||||
|
||||
from jiangchang_skill_core.runtime_env import apply_cli_local_defaults
|
||||
|
||||
apply_cli_local_defaults()
|
||||
|
||||
from util.config_bootstrap import bootstrap_skill_config
|
||||
|
||||
bootstrap_skill_config()
|
||||
|
||||
from cli.app import main
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
46
scripts/service/entitlement_service.py
Normal file
46
scripts/service/entitlement_service.py
Normal file
@@ -0,0 +1,46 @@
|
||||
"""在线鉴权（可选模板）。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Tuple
|
||||
|
||||
import requests
|
||||
from jiangchang_skill_core import config
|
||||
|
||||
|
||||
def check_entitlement(skill_slug: str) -> Tuple[bool, str]:
|
||||
auth_base = (config.get("JIANGCHANG_AUTH_BASE_URL") or "").strip().rstrip("/")
|
||||
if not auth_base:
|
||||
return True, ""
|
||||
user_id = (config.get("JIANGCHANG_USER_ID") or "").strip()
|
||||
if not user_id:
|
||||
return False, "鉴权失败：缺少用户身份（JIANGCHANG_USER_ID）"
|
||||
|
||||
auth_api_key = (config.get("JIANGCHANG_AUTH_API_KEY") or "").strip()
|
||||
timeout = int((config.get("JIANGCHANG_AUTH_TIMEOUT_SECONDS") or "5").strip())
|
||||
headers = {"Content-Type": "application/json"}
|
||||
if auth_api_key:
|
||||
headers["Authorization"] = f"Bearer {auth_api_key}"
|
||||
payload = {
|
||||
"user_id": user_id,
|
||||
"skill_slug": skill_slug,
|
||||
"trace_id": (config.get("JIANGCHANG_TRACE_ID") or "").strip(),
|
||||
"context": {"entry": "main.py"},
|
||||
}
|
||||
try:
|
||||
res = requests.post(f"{auth_base}/api/entitlements/check", json=payload, headers=headers, timeout=timeout)
|
||||
except requests.RequestException as exc:
|
||||
return False, f"鉴权请求失败：{exc}"
|
||||
if res.status_code != 200:
|
||||
return False, f"鉴权服务异常：HTTP {res.status_code}"
|
||||
try:
|
||||
body = res.json()
|
||||
except ValueError:
|
||||
return False, "鉴权服务异常：返回非 JSON"
|
||||
code = body.get("code")
|
||||
data = body.get("data") or {}
|
||||
if code != 200:
|
||||
return False, str(body.get("msg") or "鉴权失败")
|
||||
if not data.get("allow", False):
|
||||
return False, str(data.get("reason") or "未购买或已过期")
|
||||
return True, ""
|
||||
78
scripts/service/html_page_service.py
Normal file
78
scripts/service/html_page_service.py
Normal file
@@ -0,0 +1,78 @@
|
||||
"""技能 HTML 报告页：写入 `{skill_data}/html/*.html`（宿主扫描预览）。
|
||||
|
||||
契约要点（权威说明见 `references/ACTIONS.md`「HTML 页面报告」）：
|
||||
|
||||
- 目录：仅一层 `html/*.html`（非递归）
|
||||
- 重新生成 Action id 必须为 `render-html`（宿主硬编码）
|
||||
- 页面布局底线见 `references/ACTIONS.md`「宿主预览布局」；模板壳 `assets/html/example-report.html`
|
||||
- 成功后建议 `steps: refreshCatalog` 以重扫侧栏
|
||||
- 本模块只提供路径/模板/原子写；业务 KPI/表格由各技能自行填充
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import re
|
||||
from typing import Mapping
|
||||
|
||||
from util.logging_config import get_skill_logger
|
||||
from util.runtime_paths import get_html_dir, get_skill_root
|
||||
|
||||
# Host Data Management looks up this exact Skill Action id for「重新生成」.
|
||||
HOST_HTML_RENDER_ACTION_ID = "render-html"
|
||||
|
||||
_UNSAFE_NAME = re.compile(r"[\\/]|^\.\.|\.\.$")
|
||||
|
||||
|
||||
def _log():
|
||||
try:
|
||||
return get_skill_logger()
|
||||
except RuntimeError:
|
||||
return None
|
||||
|
||||
|
||||
def assert_html_filename(filename: str) -> str:
|
||||
name = (filename or "").strip()
|
||||
if not name.lower().endswith(".html"):
|
||||
raise ValueError(f"html filename must end with .html: {filename!r}")
|
||||
if _UNSAFE_NAME.search(name) or "/" in name or "\\" in name:
|
||||
raise ValueError(f"html filename must be a basename only: {filename!r}")
|
||||
if len(name) > 200:
|
||||
raise ValueError(f"html filename too long: {filename!r}")
|
||||
return name
|
||||
|
||||
|
||||
def html_page_path(filename: str, *, create_dir: bool = True) -> str:
|
||||
return os.path.join(get_html_dir(create=create_dir), assert_html_filename(filename))
|
||||
|
||||
|
||||
def load_html_template(relative_under_skill_root: str) -> str:
|
||||
"""Load UTF-8 template from skill root (e.g. ``assets/html/example-report.html``)."""
|
||||
rel = (relative_under_skill_root or "").strip().replace("\\", "/")
|
||||
if not rel or rel.startswith("/") or ".." in rel.split("/"):
|
||||
raise ValueError(f"invalid template path: {relative_under_skill_root!r}")
|
||||
path = os.path.join(get_skill_root(), *rel.split("/"))
|
||||
with open(path, encoding="utf-8") as f:
|
||||
return f.read()
|
||||
|
||||
|
||||
def render_html_placeholders(template: str, mapping: Mapping[str, str]) -> str:
|
||||
"""Replace ``{{KEY}}`` tokens. Unknown keys are left as-is."""
|
||||
page = template
|
||||
for key, value in mapping.items():
|
||||
page = page.replace("{{" + str(key) + "}}", str(value))
|
||||
return page
|
||||
|
||||
|
||||
def write_html_page(filename: str, html: str) -> str:
|
||||
"""Atomically write UTF-8 HTML under ``html/``. Returns absolute path."""
|
||||
dest = html_page_path(filename, create_dir=True)
|
||||
tmp = dest + ".tmp"
|
||||
text = html if isinstance(html, str) else str(html)
|
||||
with open(tmp, "w", encoding="utf-8", newline="\n") as f:
|
||||
f.write(text)
|
||||
os.replace(tmp, dest)
|
||||
log = _log()
|
||||
if log:
|
||||
log.info("html_page_written path=%s bytes=%s", dest, len(text.encode("utf-8")))
|
||||
return dest
|
||||
774
scripts/service/prereview_service.py
Normal file
774
scripts/service/prereview_service.py
Normal file
@@ -0,0 +1,774 @@
|
||||
"""票据合规预审规则引擎（唯一业务内核）。
|
||||
|
||||
四步审核（对齐《核心建设内容：各部门"Skill"深度需求规格书》财务处 Skill 001）：
|
||||
Step 1 字段完整性校验（F1-F4）
|
||||
Step 2 三方交叉验真（V1 真伪状态 / V2 购买方税号白名单 / V3 日期逻辑）
|
||||
Step 3 经费标准硬约束（S1 住宿费上限 / S2 交通等级 / S3 项目归属）
|
||||
Step 4 风控标记（R1 连号 / R2 同商家集中开票 / R3 节假日开票）
|
||||
|
||||
每条规则输出 {rule_id, name, level, detail, rule_ref}；level ∈ pass/warn/reject。
|
||||
总判定：任一 reject → 驳回；有 warn → 待确认；否则 通过。
|
||||
预审结果仅作提示，不构成最终报销结论；真伪以人工/官方查验为准。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import re
|
||||
import uuid
|
||||
from datetime import date, datetime, timedelta
|
||||
from typing import Any, Optional
|
||||
|
||||
from db.connection import get_conn, init_db
|
||||
from util.cny_amount import parse_cny_upper
|
||||
from util.logging_config import get_skill_logger
|
||||
from util.timeutil import now_unix
|
||||
|
||||
_VERDICT_PASS = "通过"
|
||||
_VERDICT_WARN = "待确认"
|
||||
_VERDICT_REJECT = "驳回"
|
||||
|
||||
_DATE_PATTERNS = ("%Y-%m-%d", "%Y/%m/%d", "%Y.%m.%d", "%Y年%m月%d日")
|
||||
|
||||
# 发票号码已知位数：数电票 20 位；增值税普/专票 8 位；旧版 10/12 位
|
||||
_KNOWN_NUMBER_LENGTHS = {8, 10, 12, 20}
|
||||
|
||||
_INVOICE_KEY_ALIASES = {
|
||||
"invoiceType": ("invoice_type", "发票类型", "type"),
|
||||
"invoiceCode": ("invoice_code", "发票代码"),
|
||||
"invoiceNumber": ("invoice_number", "invoice_no", "发票号码"),
|
||||
"issueDate": ("issue_date", "开票日期", "date"),
|
||||
"amount": ("amount", "金额", "total_amount", "价税合计"),
|
||||
"amountInWords": ("amount_in_words", "大写金额", "amount_words"),
|
||||
"taxRate": ("tax_rate", "税率"),
|
||||
"buyerName": ("buyer_name", "购买方名称", "buyer"),
|
||||
"buyerTaxNo": ("buyer_tax_no", "购买方税号", "buyer_tax_id"),
|
||||
"sellerName": ("seller_name", "销售方名称", "seller"),
|
||||
"sellerTaxNo": ("seller_tax_no", "销售方税号", "seller_tax_id"),
|
||||
"expenseCategory": ("expense_category", "category", "费用类别"),
|
||||
"city": ("city", "城市", "消费城市"),
|
||||
"days": ("days", "天数", "住宿天数"),
|
||||
"ticketClass": ("ticket_class", "座位等级", "席别", "舱位等级"),
|
||||
"manualVerifyResult": ("manual_verify_result", "查验结果", "人工查验结果"),
|
||||
}
|
||||
|
||||
_TICKET_CLASS_LEVELS = (
|
||||
(("商务座", "头等舱"), 3),
|
||||
(("一等座", "软卧", "公务舱", "一等舱"), 2),
|
||||
(("二等座", "硬座", "硬卧", "软座", "经济舱"), 1),
|
||||
)
|
||||
|
||||
_CHECK_PLATFORM_HINT = "全国增值税发票查验平台（inv-veri.chinatax.gov.cn）"
|
||||
|
||||
|
||||
def _log():
|
||||
try:
|
||||
return get_skill_logger()
|
||||
except RuntimeError:
|
||||
return None
|
||||
|
||||
|
||||
def _norm_invoice(raw: dict[str, Any]) -> dict[str, Any]:
|
||||
"""将输入发票字段归一化为 camelCase 键（兼容 snake_case 与常见中文键）。"""
|
||||
out: dict[str, Any] = {}
|
||||
for key, value in raw.items():
|
||||
camel = _camel(key)
|
||||
if camel in _INVOICE_KEY_ALIASES:
|
||||
out.setdefault(camel, value)
|
||||
continue
|
||||
matched = False
|
||||
for canonical, aliases in _INVOICE_KEY_ALIASES.items():
|
||||
if key in aliases:
|
||||
out.setdefault(canonical, value)
|
||||
matched = True
|
||||
break
|
||||
if not matched:
|
||||
out.setdefault(camel, value)
|
||||
return out
|
||||
|
||||
|
||||
def _camel(name: str) -> str:
|
||||
parts = re.split(r"[\s_]+", str(name).strip())
|
||||
if len(parts) <= 1:
|
||||
return str(name)
|
||||
head, *rest = parts
|
||||
return head + "".join(p[:1].upper() + p[1:] for p in rest)
|
||||
|
||||
|
||||
def _to_float(value: Any) -> Optional[float]:
|
||||
if value is None or value == "":
|
||||
return None
|
||||
if isinstance(value, (int, float)):
|
||||
return float(value)
|
||||
text = str(value).strip()
|
||||
text = re.sub(r"[¥￥元人民币\s,，]", "", text)
|
||||
try:
|
||||
return float(text)
|
||||
except ValueError:
|
||||
return None
|
||||
|
||||
|
||||
def _parse_date(value: Any) -> Optional[date]:
|
||||
if value is None or value == "":
|
||||
return None
|
||||
if isinstance(value, date):
|
||||
return value
|
||||
text = str(value).strip()
|
||||
for pattern in _DATE_PATTERNS:
|
||||
try:
|
||||
return datetime.strptime(text, pattern).date()
|
||||
except ValueError:
|
||||
continue
|
||||
return None
|
||||
|
||||
|
||||
def _fmt_date(value: Optional[date]) -> str:
|
||||
return value.isoformat() if value else ""
|
||||
|
||||
|
||||
def _load_whitelist_taxnos() -> set[str]:
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT tax_no FROM taxid_whitelist")
|
||||
return {str(row[0]).strip().upper() for row in cur.fetchall()}
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
def _load_holiday_dates() -> set[str]:
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT holiday_date FROM holiday_calendar")
|
||||
return {str(row[0]) for row in cur.fetchall()}
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
def _resolve_city_tier(city: str) -> Optional[tuple[str, str]]:
|
||||
"""按城市名归档城市等级，返回 (city_tier, basis)。"""
|
||||
if not city:
|
||||
return None
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute(
|
||||
"SELECT city_tier, city_example FROM expense_standards WHERE category = '住宿费' GROUP BY city_tier, city_example"
|
||||
)
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
for city_tier, city_example in rows:
|
||||
names = [n.strip() for n in str(city_example or "").split(",") if n.strip()]
|
||||
if city in names:
|
||||
return city_tier, "标准库城市示例"
|
||||
return None
|
||||
|
||||
|
||||
def _lookup_standard(category: str, city_tier: Optional[str], staff_title: Optional[str]) -> Optional[dict[str, Any]]:
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
sql = "SELECT city_tier, staff_title, max_amount, unit, basis_doc, clause FROM expense_standards WHERE category = ?"
|
||||
params: list[Any] = [category]
|
||||
if city_tier:
|
||||
sql += " AND city_tier = ?"
|
||||
params.append(city_tier)
|
||||
if staff_title:
|
||||
sql += " AND staff_title = ?"
|
||||
params.append(staff_title)
|
||||
cur.execute(sql + " LIMIT 1", params)
|
||||
row = cur.fetchone()
|
||||
finally:
|
||||
conn.close()
|
||||
if not row:
|
||||
return None
|
||||
return {
|
||||
"city_tier": row[0],
|
||||
"staff_title": row[1],
|
||||
"max_amount": float(row[2]),
|
||||
"unit": row[3] or "",
|
||||
"basis_doc": row[4] or "",
|
||||
"clause": row[5] or "",
|
||||
}
|
||||
|
||||
|
||||
def _ref(basis_doc: str, clause: str) -> str:
|
||||
if basis_doc and clause:
|
||||
return f"{basis_doc}{clause}"
|
||||
return basis_doc or clause or ""
|
||||
|
||||
|
||||
def _item(rule_id: str, name: str, level: str, detail: str, rule_ref: str = "") -> dict[str, str]:
|
||||
return {"rule_id": rule_id, "name": name, "level": level, "detail": detail, "rule_ref": rule_ref}
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Step 1 字段完整性校验
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _check_field_completeness(inv: dict[str, Any], today: date) -> list[dict[str, str]]:
|
||||
items: list[dict[str, str]] = []
|
||||
missing = [
|
||||
label
|
||||
for key, label in (
|
||||
("invoiceNumber", "发票号码"),
|
||||
("issueDate", "开票日期"),
|
||||
("amount", "金额"),
|
||||
("sellerName", "销售方名称"),
|
||||
)
|
||||
if inv.get(key) in (None, "")
|
||||
]
|
||||
if missing:
|
||||
items.append(
|
||||
_item(
|
||||
"F1",
|
||||
"必填字段齐全",
|
||||
"warn",
|
||||
f"以下关键字段缺失或为空：{'、'.join(missing)}。请重新拍摄/补录后再提交报销。",
|
||||
"票据预审·字段校验",
|
||||
)
|
||||
)
|
||||
|
||||
number = str(inv.get("invoiceNumber") or "").strip().replace(" ", "")
|
||||
if number and not number.isdigit():
|
||||
items.append(_item("F2", "发票号码格式", "warn", f"发票号码含非数字字符：{number}", "票据预审·字段校验"))
|
||||
elif number and len(number) not in _KNOWN_NUMBER_LENGTHS:
|
||||
items.append(
|
||||
_item(
|
||||
"F2",
|
||||
"发票号码格式",
|
||||
"warn",
|
||||
f"发票号码 {len(number)} 位，不属于已知位数（数电票 20 位、普/专票 8 位、旧版 10/12 位），疑似异常，请人工核验票面。",
|
||||
"票据预审·字段校验",
|
||||
)
|
||||
)
|
||||
|
||||
issue_date = _parse_date(inv.get("issueDate"))
|
||||
raw_date = str(inv.get("issueDate") or "").strip()
|
||||
if raw_date and issue_date is None:
|
||||
items.append(_item("F3", "开票日期合法性", "reject", f"开票日期无法解析：{raw_date}（应为 YYYY-MM-DD）", "票据预审·字段校验"))
|
||||
elif issue_date and issue_date > today:
|
||||
items.append(_item("F3", "开票日期合法性", "reject", f"开票日期 {issue_date.isoformat()} 晚于今日，票面异常。", "票据预审·字段校验"))
|
||||
elif issue_date:
|
||||
items.append(_item("F3", "开票日期合法性", "pass", f"开票日期 {issue_date.isoformat()} 合法。", "票据预审·字段校验"))
|
||||
|
||||
amount = _to_float(inv.get("amount"))
|
||||
words = str(inv.get("amountInWords") or "").strip()
|
||||
if amount is not None and words:
|
||||
words_value = parse_cny_upper(words)
|
||||
if words_value is None:
|
||||
items.append(_item("F4", "大小写金额一致", "warn", f"大写金额“{words}”无法解析，请人工核对大小写金额。", "票据预审·字段校验"))
|
||||
elif abs(words_value - amount) > 0.01:
|
||||
items.append(
|
||||
_item(
|
||||
"F4",
|
||||
"大小写金额一致",
|
||||
"reject",
|
||||
f"大写金额 {words_value:.2f} 元与小写金额 {amount:.2f} 元不一致，票面异常。",
|
||||
"票据预审·字段校验",
|
||||
)
|
||||
)
|
||||
else:
|
||||
items.append(_item("F4", "大小写金额一致", "pass", f"大小写金额一致（{amount:.2f} 元）。", "票据预审·字段校验"))
|
||||
return items
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Step 2 三方交叉验真
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _check_cross_verification(
|
||||
inv: dict[str, Any],
|
||||
context: dict[str, Any],
|
||||
whitelist: set[str],
|
||||
today: date,
|
||||
) -> list[dict[str, str]]:
|
||||
items: list[dict[str, str]] = []
|
||||
|
||||
verify = str(inv.get("manualVerifyResult") or "").strip().lower()
|
||||
if verify in ("fail", "false", "不符", "假票", "查验不符"):
|
||||
items.append(_item("V1", "发票真伪", "reject", "人工查验结果为“不符”，判定为问题票据。", _CHECK_PLATFORM_HINT))
|
||||
elif verify in ("pass", "true", "通过", "真票", "查验通过"):
|
||||
items.append(_item("V1", "发票真伪", "pass", "人工查验结果为“通过”。", _CHECK_PLATFORM_HINT))
|
||||
else:
|
||||
items.append(
|
||||
_item(
|
||||
"V1",
|
||||
"发票真伪",
|
||||
"warn",
|
||||
f"尚未查验真伪：请经办人到{_CHECK_PLATFORM_HINT}录入票面信息核验，并在本报告回填查验结果。",
|
||||
_CHECK_PLATFORM_HINT,
|
||||
)
|
||||
)
|
||||
|
||||
buyer_tax_no = str(inv.get("buyerTaxNo") or "").strip()
|
||||
if not buyer_tax_no:
|
||||
items.append(_item("V2", "购买方税号白名单", "warn", "票面缺少购买方税号，无法进行白名单比对，请补录抬头信息。", "票据预审·税号白名单"))
|
||||
elif buyer_tax_no.strip().upper() in whitelist:
|
||||
buyer_name = str(inv.get("buyerName") or "").strip()
|
||||
items.append(_item("V2", "购买方税号白名单", "pass", f"购买方税号在我校准予报销白名单内{('（' + buyer_name + '）') if buyer_name else ''}。", "票据预审·税号白名单"))
|
||||
else:
|
||||
items.append(
|
||||
_item(
|
||||
"V2",
|
||||
"购买方税号白名单",
|
||||
"reject",
|
||||
f"购买方税号 {buyer_tax_no} 不在我校准予报销税号白名单内，抬头开错，须换开后方可报销。",
|
||||
"票据预审·税号白名单",
|
||||
)
|
||||
)
|
||||
|
||||
issue_date = _parse_date(inv.get("issueDate"))
|
||||
trip_start = _parse_date(context.get("trip_start"))
|
||||
trip_end = _parse_date(context.get("trip_end"))
|
||||
if issue_date and trip_start:
|
||||
window_end = trip_end or trip_start
|
||||
if issue_date < trip_start:
|
||||
items.append(
|
||||
_item(
|
||||
"V3",
|
||||
"日期逻辑（先审批后出行）",
|
||||
"warn",
|
||||
f"开票日期 {issue_date.isoformat()} 早于出差开始日 {trip_start.isoformat()}，请核实行程与审批单（先审批后出行）。",
|
||||
"票据预审·日期逻辑",
|
||||
)
|
||||
)
|
||||
elif issue_date > window_end + timedelta(days=1):
|
||||
items.append(
|
||||
_item(
|
||||
"V3",
|
||||
"日期逻辑（先审批后出行）",
|
||||
"warn",
|
||||
f"开票日期 {issue_date.isoformat()} 晚于出差结束日 {window_end.isoformat()}（含 1 天宽限），请核实行程。",
|
||||
"票据预审·日期逻辑",
|
||||
)
|
||||
)
|
||||
else:
|
||||
items.append(
|
||||
_item(
|
||||
"V3",
|
||||
"日期逻辑（先审批后出行）",
|
||||
"pass",
|
||||
f"开票日期落在出差区间 [{trip_start.isoformat()}, {(trip_end or trip_start).isoformat()}] 内。",
|
||||
"票据预审·日期逻辑",
|
||||
)
|
||||
)
|
||||
return items
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Step 3 经费标准硬约束
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _check_expense_standards(inv: dict[str, Any], context: dict[str, Any]) -> list[dict[str, str]]:
|
||||
items: list[dict[str, str]] = []
|
||||
category = str(inv.get("expenseCategory") or "").strip()
|
||||
amount = _to_float(inv.get("amount"))
|
||||
staff_title = str(context.get("staff_title") or "").strip()
|
||||
|
||||
if "住宿" in category and amount is not None:
|
||||
city = str(context.get("trip_city") or inv.get("city") or "").strip()
|
||||
tier_info = _resolve_city_tier(city) if city else None
|
||||
if not city:
|
||||
items.append(
|
||||
_item(
|
||||
"S1",
|
||||
"住宿费标准比对",
|
||||
"warn",
|
||||
"未提供消费城市，无法匹配住宿费标准，请补录城市后重新预审或人工对照《差旅费管理办法》。",
|
||||
_ref("《差旅费管理办法》", "第十条"),
|
||||
)
|
||||
)
|
||||
elif not tier_info:
|
||||
items.append(
|
||||
_item(
|
||||
"S1",
|
||||
"住宿费标准比对",
|
||||
"warn",
|
||||
f"城市“{city}”不在标准库城市等级中，无法匹配住宿费标准，请人工对照《差旅费管理办法》或请管理员维护该城市。",
|
||||
_ref("《差旅费管理办法》", "第十条"),
|
||||
)
|
||||
)
|
||||
else:
|
||||
standard = _lookup_standard("住宿费", tier_info[0], staff_title or None)
|
||||
days = _to_float(inv.get("days")) or 1.0
|
||||
days = max(days, 1.0)
|
||||
if not standard:
|
||||
items.append(
|
||||
_item(
|
||||
"S1",
|
||||
"住宿费标准比对",
|
||||
"warn",
|
||||
f"未匹配到“{tier_info[0]}/{staff_title or '未提供职级'}”的住宿费标准，请人工对照《差旅费管理办法》。",
|
||||
_ref("《差旅费管理办法》", "第十条"),
|
||||
)
|
||||
)
|
||||
else:
|
||||
per_day = amount / days
|
||||
limit = standard["max_amount"]
|
||||
ref = _ref(standard["basis_doc"], standard["clause"])
|
||||
if per_day > limit + 0.001:
|
||||
items.append(
|
||||
_item(
|
||||
"S1",
|
||||
"住宿费标准比对",
|
||||
"reject",
|
||||
(
|
||||
f"住宿费折合 {per_day:.2f} 元/人/天，超过“{standard['city_tier']}/{standard['staff_title']}”"
|
||||
f"上限 {limit:.2f} 元/人/天，超标 {per_day - limit:.2f} 元。"
|
||||
),
|
||||
ref,
|
||||
)
|
||||
)
|
||||
else:
|
||||
items.append(
|
||||
_item(
|
||||
"S1",
|
||||
"住宿费标准比对",
|
||||
"pass",
|
||||
f"住宿费折合 {per_day:.2f} 元/人/天，未超“{standard['city_tier']}/{standard['staff_title']}”上限 {limit:.2f} 元/人/天。",
|
||||
ref,
|
||||
)
|
||||
)
|
||||
|
||||
if "交通" in category or "出租" in str(inv.get("invoiceType") or ""):
|
||||
ticket_class = str(inv.get("ticketClass") or "").strip()
|
||||
standard = _lookup_standard("交通费", None, staff_title or None)
|
||||
if ticket_class and standard:
|
||||
class_level = _ticket_class_level(ticket_class)
|
||||
ref = _ref(standard["basis_doc"], standard["clause"])
|
||||
if class_level is None:
|
||||
items.append(_item("S2", "交通席位等级", "warn", f"席位/舱位等级“{ticket_class}”无法识别，请人工对照职级可乘等级。", ref))
|
||||
elif class_level > standard["max_amount"]:
|
||||
items.append(
|
||||
_item(
|
||||
"S2",
|
||||
"交通席位等级",
|
||||
"warn",
|
||||
f"“{ticket_class}”等级高于“{standard['staff_title']}”可乘等级（代码 {standard['max_amount']:.0f}），请确认是否符合职级规定或已获审批。",
|
||||
ref,
|
||||
)
|
||||
)
|
||||
else:
|
||||
items.append(_item("S2", "交通席位等级", "pass", f"“{ticket_class}”符合“{standard['staff_title']}”可乘等级。", ref))
|
||||
|
||||
project_no = str(context.get("project_no") or "").strip()
|
||||
if project_no:
|
||||
items.append(
|
||||
_item(
|
||||
"S3",
|
||||
"项目经费归属",
|
||||
"pass",
|
||||
f"本笔将计入预算项目 {project_no}；项目余额校验待财务系统对接后启用。",
|
||||
"票据预审·项目归属",
|
||||
)
|
||||
)
|
||||
return items
|
||||
|
||||
|
||||
def _ticket_class_level(ticket_class: str) -> Optional[int]:
|
||||
for keywords, level in _TICKET_CLASS_LEVELS:
|
||||
if any(k in ticket_class for k in keywords):
|
||||
return level
|
||||
return None
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Step 4 风控标记
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _check_holiday(inv: dict[str, Any], holidays: set[str]) -> list[dict[str, str]]:
|
||||
items: list[dict[str, str]] = []
|
||||
issue_date = _parse_date(inv.get("issueDate"))
|
||||
if not issue_date:
|
||||
return items
|
||||
iso = issue_date.isoformat()
|
||||
if iso in holidays:
|
||||
items.append(
|
||||
_item("R3", "节假日开票", "warn", f"开票日期 {iso} 为法定节假日，请核实消费真实性。", "票据预审·风控标记")
|
||||
)
|
||||
elif issue_date.weekday() >= 5:
|
||||
items.append(
|
||||
_item("R3", "节假日开票", "warn", f"开票日期 {iso} 为周末（{'周六' if issue_date.weekday() == 5 else '周日'}），请核实消费真实性。", "票据预审·风控标记")
|
||||
)
|
||||
return items
|
||||
|
||||
|
||||
def _check_batch_risks(invoices: list[dict[str, Any]]) -> dict[int, list[dict[str, str]]]:
|
||||
"""R1 连号发票 + R2 同商家集中开票（批内）。返回 invoice index → items。"""
|
||||
extra: dict[int, list[dict[str, str]]] = {}
|
||||
|
||||
# R1：同发票代码且号码连续（差值 ≤ 2）
|
||||
numeric_by_code: dict[str, list[tuple[int, int]]] = {}
|
||||
for idx, inv in enumerate(invoices):
|
||||
code = str(inv.get("invoiceCode") or "").strip() or "-"
|
||||
number = str(inv.get("invoiceNumber") or "").strip()
|
||||
if number.isdigit():
|
||||
numeric_by_code.setdefault(code, []).append((idx, int(number)))
|
||||
for _code, entries in numeric_by_code.items():
|
||||
for idx, number in entries:
|
||||
peers = [n for j, n in entries if j != idx and abs(n - number) <= 2]
|
||||
if peers:
|
||||
extra.setdefault(idx, []).append(
|
||||
_item(
|
||||
"R1",
|
||||
"连号发票",
|
||||
"warn",
|
||||
f"与本批发票号码连号（相邻号码：{', '.join(str(p) for p in sorted(set(peers)))}），请核实是否拆分消费。",
|
||||
"票据预审·风控标记",
|
||||
)
|
||||
)
|
||||
|
||||
# R2：同商家集中开票（批内按销方税号，缺税号回退销方名称）
|
||||
sellers: dict[str, list[int]] = {}
|
||||
for idx, inv in enumerate(invoices):
|
||||
key = str(inv.get("sellerTaxNo") or "").strip().upper() or str(inv.get("sellerName") or "").strip()
|
||||
if key:
|
||||
sellers.setdefault(key, []).append(idx)
|
||||
for _key, idx_list in sellers.items():
|
||||
if len(idx_list) >= 3:
|
||||
sample_name = str(invoices[idx_list[0]].get("sellerName") or "同一销售方").strip()
|
||||
for idx in idx_list:
|
||||
extra.setdefault(idx, []).append(
|
||||
_item(
|
||||
"R2",
|
||||
"同商家集中开票",
|
||||
"warn",
|
||||
f"本批有 {len(idx_list)} 张发票来自“{sample_name}”，存在集中开票情形，请核实消费真实性。",
|
||||
"票据预审·风控标记",
|
||||
)
|
||||
)
|
||||
return extra
|
||||
|
||||
|
||||
def _check_seller_history(inv: dict[str, Any]) -> Optional[dict[str, str]]:
|
||||
"""R2 历史扩展：该销方近 30 天（含本张）累计 ≥ 5 张 → 黄灯。"""
|
||||
seller_tax_no = str(inv.get("sellerTaxNo") or "").strip()
|
||||
if not seller_tax_no:
|
||||
return None
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT COUNT(*) FROM invoice_records
|
||||
WHERE seller_tax_no = ? AND created_at >= ?
|
||||
""",
|
||||
(seller_tax_no, now_unix() - 30 * 24 * 3600),
|
||||
)
|
||||
history = int(cur.fetchone()[0])
|
||||
finally:
|
||||
conn.close()
|
||||
if history + 1 >= 5:
|
||||
seller_name = str(inv.get("sellerName") or seller_tax_no).strip()
|
||||
return _item(
|
||||
"R2",
|
||||
"同商家集中开票",
|
||||
"warn",
|
||||
f"销售方“{seller_name}”近 30 天累计开票 {history + 1} 张（含本张），存在集中开票情形，请核实消费真实性。",
|
||||
"票据预审·风控标记",
|
||||
)
|
||||
return None
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 内核入口
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _decide(items: list[dict[str, str]]) -> str:
|
||||
levels = {i["level"] for i in items}
|
||||
if "reject" in levels:
|
||||
return _VERDICT_REJECT
|
||||
if "warn" in levels:
|
||||
return _VERDICT_WARN
|
||||
return _VERDICT_PASS
|
||||
|
||||
|
||||
def _summarize(verdict: str, invoice_number: str, items: list[dict[str, str]]) -> str:
|
||||
rejects = [i for i in items if i["level"] == "reject"]
|
||||
warns = [i for i in items if i["level"] == "warn"]
|
||||
if verdict == _VERDICT_REJECT:
|
||||
return f"发票 {invoice_number} 预审驳回：{'；'.join(i['name'] for i in rejects[:3])}"
|
||||
if verdict == _VERDICT_WARN:
|
||||
return f"发票 {invoice_number} 待确认：{'；'.join(i['name'] for i in warns[:3])}"
|
||||
return f"发票 {invoice_number} 预审通过，全部检查项通过。"
|
||||
|
||||
|
||||
def _validate_input(invoices: list[dict[str, Any]]) -> None:
|
||||
if not invoices:
|
||||
raise ValueError("发票字段为空：请提供至少一张发票的结构化字段 JSON")
|
||||
if len(invoices) > 50:
|
||||
raise ValueError(f"单批发票数量过多（{len(invoices)} 张），单批上限 50 张")
|
||||
for idx, inv in enumerate(invoices):
|
||||
if not isinstance(inv, dict):
|
||||
raise ValueError(f"第 {idx + 1} 张发票字段不是 JSON 对象")
|
||||
|
||||
|
||||
def run_prereview(invoices_raw: list[dict[str, Any]], context: dict[str, Any]) -> dict[str, Any]:
|
||||
"""预审内核：四步审核 → 写库 → 生成《初审意见单》HTML。"""
|
||||
from service.report_service import write_report_for_kernel
|
||||
|
||||
log = _log()
|
||||
init_db()
|
||||
invoices = [_norm_invoice(inv) for inv in invoices_raw]
|
||||
_validate_input(invoices)
|
||||
|
||||
batch_no = now_unix().to_bytes(5, "big").hex() + uuid.uuid4().hex[:6]
|
||||
today = date.today()
|
||||
whitelist = _load_whitelist_taxnos()
|
||||
holidays = _load_holiday_dates()
|
||||
|
||||
if log:
|
||||
log.info("prereview_kernel_start batch_no=%s count=%s stage=kernel", batch_no, len(invoices))
|
||||
|
||||
batch_extra = _check_batch_risks(invoices)
|
||||
results: list[dict[str, Any]] = []
|
||||
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
for idx, inv in enumerate(invoices):
|
||||
items: list[dict[str, str]] = []
|
||||
items.extend(_check_field_completeness(inv, today))
|
||||
items.extend(_check_cross_verification(inv, context, whitelist, today))
|
||||
items.extend(_check_expense_standards(inv, context))
|
||||
items.extend(_check_holiday(inv, holidays))
|
||||
items.extend(batch_extra.get(idx, []))
|
||||
|
||||
seller_key = str(inv.get("sellerTaxNo") or "").strip()
|
||||
if seller_key:
|
||||
history_item = _check_seller_history(inv)
|
||||
if history_item and not any(i["rule_id"] == "R2" for i in items):
|
||||
items.append(history_item)
|
||||
|
||||
verdict = _decide(items)
|
||||
invoice_number = str(inv.get("invoiceNumber") or "").strip() or f"未提供-{idx + 1}"
|
||||
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO invoice_records
|
||||
(batch_no, invoice_type, invoice_code, invoice_number, issue_date, amount,
|
||||
amount_in_words, tax_rate, buyer_name, buyer_tax_no, seller_name, seller_tax_no,
|
||||
expense_category, city, days, purpose, project_no, trip_context_json)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(
|
||||
batch_no,
|
||||
str(inv.get("invoiceType") or ""),
|
||||
str(inv.get("invoiceCode") or ""),
|
||||
invoice_number,
|
||||
_fmt_date(_parse_date(inv.get("issueDate"))),
|
||||
_to_float(inv.get("amount")),
|
||||
str(inv.get("amountInWords") or ""),
|
||||
_to_float(inv.get("taxRate")),
|
||||
str(inv.get("buyerName") or ""),
|
||||
str(inv.get("buyerTaxNo") or ""),
|
||||
str(inv.get("sellerName") or ""),
|
||||
str(inv.get("sellerTaxNo") or ""),
|
||||
str(inv.get("expenseCategory") or ""),
|
||||
str(inv.get("city") or ""),
|
||||
int(_to_float(inv.get("days")) or 0),
|
||||
str(context.get("purpose") or ""),
|
||||
str(context.get("project_no") or ""),
|
||||
json.dumps(context, ensure_ascii=False),
|
||||
),
|
||||
)
|
||||
invoice_id = int(cur.lastrowid)
|
||||
|
||||
summary = _summarize(verdict, invoice_number, items)
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO prereview_reports
|
||||
(batch_no, invoice_id, invoice_number, seller_name, amount, verdict,
|
||||
pass_json, warn_json, reject_json, summary)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(
|
||||
batch_no,
|
||||
invoice_id,
|
||||
invoice_number,
|
||||
str(inv.get("sellerName") or ""),
|
||||
_to_float(inv.get("amount")),
|
||||
verdict,
|
||||
json.dumps([i for i in items if i["level"] == "pass"], ensure_ascii=False),
|
||||
json.dumps([i for i in items if i["level"] == "warn"], ensure_ascii=False),
|
||||
json.dumps([i for i in items if i["level"] == "reject"], ensure_ascii=False),
|
||||
summary,
|
||||
),
|
||||
)
|
||||
report_id = int(cur.lastrowid)
|
||||
|
||||
report_view = {
|
||||
"id": report_id,
|
||||
"batch_no": batch_no,
|
||||
"invoice_number": invoice_number,
|
||||
"seller_name": str(inv.get("sellerName") or ""),
|
||||
"amount": _to_float(inv.get("amount")),
|
||||
"verdict": verdict,
|
||||
"summary": summary,
|
||||
"verify_status": "待查验",
|
||||
"created_at": now_unix(),
|
||||
"pass_json": json.dumps([i for i in items if i["level"] == "pass"], ensure_ascii=False),
|
||||
"warn_json": json.dumps([i for i in items if i["level"] == "warn"], ensure_ascii=False),
|
||||
"reject_json": json.dumps([i for i in items if i["level"] == "reject"], ensure_ascii=False),
|
||||
}
|
||||
invoice_view = {
|
||||
"invoice_type": str(inv.get("invoiceType") or ""),
|
||||
"invoice_code": str(inv.get("invoiceCode") or ""),
|
||||
"invoice_number": invoice_number,
|
||||
"issue_date": _fmt_date(_parse_date(inv.get("issueDate"))),
|
||||
"amount": _to_float(inv.get("amount")),
|
||||
"buyer_name": str(inv.get("buyerName") or ""),
|
||||
"buyer_tax_no": str(inv.get("buyerTaxNo") or ""),
|
||||
"seller_name": str(inv.get("sellerName") or ""),
|
||||
"seller_tax_no": str(inv.get("sellerTaxNo") or ""),
|
||||
"expense_category": str(inv.get("expenseCategory") or ""),
|
||||
"city": str(inv.get("city") or ""),
|
||||
}
|
||||
report_file = write_report_for_kernel(
|
||||
report_id=report_id,
|
||||
report=report_view,
|
||||
invoice=invoice_view,
|
||||
context=context,
|
||||
)
|
||||
cur.execute(
|
||||
"UPDATE prereview_reports SET report_file = ? WHERE id = ?",
|
||||
(report_file, report_id),
|
||||
)
|
||||
|
||||
results.append(
|
||||
{
|
||||
"report_id": report_id,
|
||||
"batch_no": batch_no,
|
||||
"invoice_id": invoice_id,
|
||||
"invoice_number": invoice_number,
|
||||
"verdict": verdict,
|
||||
"summary": summary,
|
||||
"items": items,
|
||||
"report_file": report_file,
|
||||
}
|
||||
)
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
counts = {v: sum(1 for r in results if r["verdict"] == v) for v in (_VERDICT_PASS, _VERDICT_WARN, _VERDICT_REJECT)}
|
||||
payload = {
|
||||
"ok": True,
|
||||
"batch_no": batch_no,
|
||||
"count": len(results),
|
||||
"verdict_counts": counts,
|
||||
"disclaimer": "预审结果仅作提示，不构成最终报销结论；发票真伪以官方查验结果为准。",
|
||||
"reports": results,
|
||||
}
|
||||
if log:
|
||||
log.info(
|
||||
"prereview_kernel_done batch_no=%s pass=%s warn=%s reject=%s stage=kernel",
|
||||
batch_no,
|
||||
counts[_VERDICT_PASS],
|
||||
counts[_VERDICT_WARN],
|
||||
counts[_VERDICT_REJECT],
|
||||
)
|
||||
return payload
|
||||
182
scripts/service/report_service.py
Normal file
182
scripts/service/report_service.py
Normal file
@@ -0,0 +1,182 @@
|
||||
"""《初审意见单》HTML 报告组装。
|
||||
|
||||
写入 `{skill_data}/html/report-{report_id}.html`，宿主数据管理侧栏可预览；
|
||||
重新生成走固定 Action id `render-html`（见 references/ACTIONS.md）。
|
||||
`{{BODY}}` 内数据一律经 html.escape 转义。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import html
|
||||
import json
|
||||
from typing import Any
|
||||
|
||||
from service.html_page_service import load_html_template, render_html_placeholders, write_html_page
|
||||
from util.timeutil import unix_to_iso
|
||||
|
||||
_TEMPLATE_REL = "assets/html/prereview-report.html"
|
||||
|
||||
_LEVEL_BADGE = {"pass": "✅", "warn": "⚠️", "reject": "❌"}
|
||||
|
||||
|
||||
def _esc(value: Any) -> str:
|
||||
return html.escape(str(value if value is not None else ""), quote=True)
|
||||
|
||||
|
||||
def _items_rows(items_json: str) -> str:
|
||||
try:
|
||||
items = json.loads(items_json) if items_json else []
|
||||
except (TypeError, ValueError):
|
||||
items = []
|
||||
if not items:
|
||||
return '<tr><td colspan="3" class="muted">无</td></tr>'
|
||||
rows = []
|
||||
for item in items:
|
||||
badge = _LEVEL_BADGE.get(str(item.get("level") or ""), "•")
|
||||
ref = str(item.get("rule_ref") or "")
|
||||
ref_cell = f'<span class="ref">依据：{_esc(ref)}</span>' if ref else ""
|
||||
rows.append(
|
||||
"<tr>"
|
||||
f'<td class="nowrap">{badge} {_esc(item.get("rule_id"))}</td>'
|
||||
f'<td>{_esc(item.get("name"))}<br/><span class="muted">{_esc(item.get("detail"))}</span>{ref_cell}</td>'
|
||||
f'<td class="nowrap">{_esc(item.get("level"))}</td>'
|
||||
"</tr>"
|
||||
)
|
||||
return "".join(rows)
|
||||
|
||||
|
||||
def _info_table(inv: dict[str, Any]) -> str:
|
||||
fields = (
|
||||
("发票类型", "invoice_type"),
|
||||
("发票代码", "invoice_code"),
|
||||
("发票号码", "invoice_number"),
|
||||
("开票日期", "issue_date"),
|
||||
("金额（元）", "amount"),
|
||||
("购买方名称", "buyer_name"),
|
||||
("购买方税号", "buyer_tax_no"),
|
||||
("销售方名称", "seller_name"),
|
||||
("销售方税号", "seller_tax_no"),
|
||||
("费用类别", "expense_category"),
|
||||
("消费城市", "city"),
|
||||
)
|
||||
rows = "".join(
|
||||
f"<tr><th>{_esc(label)}</th><td>{_esc(inv.get(key) if inv.get(key) not in (None, '') else '—')}</td></tr>"
|
||||
for label, key in fields
|
||||
)
|
||||
return f'<table class="info">{rows}</table>'
|
||||
|
||||
|
||||
def _context_table(context: dict[str, Any]) -> str:
|
||||
fields = (
|
||||
("报销事由", "purpose"),
|
||||
("预算项目编号", "project_no"),
|
||||
("出差开始", "trip_start"),
|
||||
("出差结束", "trip_end"),
|
||||
("目的地", "trip_city"),
|
||||
("职级", "staff_title"),
|
||||
)
|
||||
rows = []
|
||||
for label, key in fields:
|
||||
value = str(context.get(key) or "").strip()
|
||||
if value:
|
||||
rows.append(f"<tr><th>{_esc(label)}</th><td>{_esc(value)}</td></tr>")
|
||||
if not rows:
|
||||
return ""
|
||||
return f'<table class="info">{"".join(rows)}</table>'
|
||||
|
||||
|
||||
def build_report_html(
|
||||
*,
|
||||
report: dict[str, Any],
|
||||
invoice: dict[str, Any],
|
||||
context: dict[str, Any],
|
||||
) -> str:
|
||||
"""组装完整《初审意见单》HTML 文本。"""
|
||||
verdict = str(report.get("verdict") or "")
|
||||
badge = {"通过": "verdict-pass", "待确认": "verdict-warn", "驳回": "verdict-reject"}.get(verdict, "verdict-warn")
|
||||
mark = {"通过": "✅ 通过", "待确认": "⚠️ 待确认", "驳回": "❌ 驳回"}.get(verdict, verdict)
|
||||
|
||||
verify_status = str(report.get("verify_status") or "待查验")
|
||||
verify_note = str(report.get("verify_note") or "")
|
||||
created = unix_to_iso(report.get("created_at"))
|
||||
|
||||
body = f"""
|
||||
<div class="verdict-line">
|
||||
<span class="verdict {badge}">{mark}</span>
|
||||
<span class="muted">批次 {_esc(report.get("batch_no"))} · 报告编号 {_esc(report.get("id"))} · 生成于 {_esc(created or "")}</span>
|
||||
</div>
|
||||
<p class="summary">{_esc(report.get("summary"))}</p>
|
||||
<div class="section"><h2>发票信息</h2>{_info_table(invoice)}</div>
|
||||
<div class="section"><h2>报销上下文</h2>{_context_table(context) or '<p class="muted">未提供差旅/项目上下文。</p>'}</div>
|
||||
<div class="section"><h2>❌ 驳回项（必须整改）</h2><table><tr><th>规则</th><th>说明</th><th>级别</th></tr>{_items_rows(report.get("reject_json") or "")}</table></div>
|
||||
<div class="section"><h2>⚠️ 待确认项（请核实）</h2><table><tr><th>规则</th><th>说明</th><th>级别</th></tr>{_items_rows(report.get("warn_json") or "")}</table></div>
|
||||
<div class="section"><h2>✅ 通过项</h2><table><tr><th>规则</th><th>说明</th><th>级别</th></tr>{_items_rows(report.get("pass_json") or "")}</table></div>
|
||||
<div class="section"><h2>真伪查验</h2>
|
||||
<p>当前状态：<strong>{_esc(verify_status)}</strong>{(" · 备注：" + _esc(verify_note)) if verify_note else ""}</p>
|
||||
<p class="muted">请经办人到全国增值税发票查验平台（inv-veri.chinatax.gov.cn）核验票面真伪后，在数据管理“初审意见”表回填查验结果。</p>
|
||||
</div>
|
||||
<footer>本《初审意见单》由票据合规预审助手自动生成，预审结果不构成最终报销结论，请以财务处人工审核为准。</footer>
|
||||
"""
|
||||
|
||||
template = load_html_template(_TEMPLATE_REL)
|
||||
return render_html_placeholders(
|
||||
template,
|
||||
{
|
||||
"TITLE": f"初审意见单 · 发票 {report.get('invoice_number') or ''}",
|
||||
"GENERATED_AT": created or "",
|
||||
"BODY": body,
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
def build_report_for_invoice(report_id: int, batch_no: str = "") -> str:
|
||||
"""按报告编号从库中取数并写出 HTML，返回文件绝对路径（重建入口用）。"""
|
||||
from db.connection import get_conn
|
||||
from util.logging_config import get_skill_logger
|
||||
|
||||
log = get_skill_logger()
|
||||
log.info("report_render_start report_id=%s stage=render", report_id)
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT * FROM prereview_reports WHERE id = ?", (report_id,))
|
||||
row = cur.fetchone()
|
||||
if not row:
|
||||
raise ValueError(f"初审报告不存在：{report_id}")
|
||||
report = _row_to_dict(cur, "prereview_reports", row)
|
||||
invoice = {}
|
||||
if report.get("invoice_id"):
|
||||
cur.execute("SELECT * FROM invoice_records WHERE id = ?", (report["invoice_id"],))
|
||||
irow = cur.fetchone()
|
||||
if irow:
|
||||
invoice = _row_to_dict(cur, "invoice_records", irow)
|
||||
context: dict[str, Any] = {}
|
||||
try:
|
||||
context = json.loads(invoice.get("trip_context_json") or "{}")
|
||||
except (TypeError, ValueError):
|
||||
context = {}
|
||||
html_text = build_report_html(report=report, invoice=invoice, context=context)
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
filename = f"report-{report_id}.html"
|
||||
path = write_html_page(filename, html_text)
|
||||
log.info("report_render_done report_id=%s path=%s stage=render", report_id, path)
|
||||
return path
|
||||
|
||||
|
||||
def write_report_for_kernel(
|
||||
*,
|
||||
report_id: int,
|
||||
report: dict[str, Any],
|
||||
invoice: dict[str, Any],
|
||||
context: dict[str, Any],
|
||||
) -> str:
|
||||
"""内核路径：用事务内内存数据直接组装并写出 HTML（不回查未提交的行）。"""
|
||||
html_text = build_report_html(report=report, invoice=invoice, context=context)
|
||||
return write_html_page(f"report-{report_id}.html", html_text)
|
||||
|
||||
|
||||
def _row_to_dict(cursor, table_name: str, row) -> dict[str, Any]:
|
||||
columns = [desc[0] for desc in cursor.description]
|
||||
return dict(zip(columns, row))
|
||||
57
scripts/service/sibling_bridge.py
Normal file
57
scripts/service/sibling_bridge.py
Normal file
@@ -0,0 +1,57 @@
|
||||
"""兄弟技能 CLI 调用工具。
|
||||
|
||||
通过子进程调用同级其他 skill 的 main.py，并解析 JSON 输出。
|
||||
具体调用哪些兄弟技能，由复制后的业务 skill 在 service 层按需决定。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import os
|
||||
import subprocess
|
||||
import sys
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from util.logging_config import subprocess_env_with_trace
|
||||
from util.runtime_paths import get_skills_root
|
||||
|
||||
|
||||
def call_sibling_json(skill_slug: str, args: List[str]) -> Optional[Dict[str, Any]]:
|
||||
"""通用兄弟技能调用器。
|
||||
|
||||
Args:
|
||||
skill_slug: 兄弟技能的 slug（即同级目录名）。
|
||||
args: 传给 main.py 的命令行参数列表。
|
||||
|
||||
Returns:
|
||||
若兄弟技能输出可解析的 JSON 对象则返回 dict；否则返回 None。
|
||||
"""
|
||||
script_path = os.path.join(get_skills_root(), skill_slug, "scripts", "main.py")
|
||||
if not os.path.isfile(script_path):
|
||||
return None
|
||||
proc = subprocess.run(
|
||||
[sys.executable, script_path, *args],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
encoding="utf-8",
|
||||
errors="replace",
|
||||
env=subprocess_env_with_trace(),
|
||||
)
|
||||
raw = (proc.stdout or "").strip()
|
||||
if not raw or raw.startswith("ERROR"):
|
||||
return None
|
||||
try:
|
||||
data = json.loads(raw)
|
||||
except json.JSONDecodeError:
|
||||
return None
|
||||
return data if isinstance(data, dict) else None
|
||||
|
||||
|
||||
def get_sibling_main_path(skill_slug: str) -> str:
|
||||
"""获取兄弟技能的 main.py 绝对路径。
|
||||
|
||||
使用示例（在 task_service.py 中）：
|
||||
from service.sibling_bridge import get_sibling_main_path, call_sibling_json
|
||||
result = call_sibling_json("account-manager", ["list", "--limit", "10"])
|
||||
"""
|
||||
return os.path.join(get_skills_root(), skill_slug, "scripts", "main.py")
|
||||
366
scripts/service/standards_service.py
Normal file
366
scripts/service/standards_service.py
Normal file
@@ -0,0 +1,366 @@
|
||||
"""经费标准库 / 税号白名单 / 节假日 / 初审记录 的查询与维护。
|
||||
|
||||
预审内核只读本模块数据；维护操作（改标准、增删白名单、查验回填）在此实现。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import sqlite3
|
||||
from typing import Any, Optional
|
||||
|
||||
from db.connection import get_conn, init_db
|
||||
from util.logging_config import get_skill_logger
|
||||
from util.timeutil import now_unix
|
||||
|
||||
VERIFY_PENDING = "待查验"
|
||||
VERIFY_PASS = "已查验通过"
|
||||
VERIFY_FAIL = "已查验不符"
|
||||
|
||||
|
||||
def _log():
|
||||
try:
|
||||
return get_skill_logger()
|
||||
except RuntimeError:
|
||||
return None
|
||||
|
||||
|
||||
def list_standards(category: Optional[str] = None) -> list[dict[str, Any]]:
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
if category:
|
||||
cur.execute(
|
||||
"SELECT id, category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause, effective_date FROM expense_standards WHERE category = ? ORDER BY category, city_tier, staff_title",
|
||||
(category,),
|
||||
)
|
||||
else:
|
||||
cur.execute(
|
||||
"SELECT id, category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause, effective_date FROM expense_standards ORDER BY category, city_tier, staff_title"
|
||||
)
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
return [
|
||||
{
|
||||
"id": r[0],
|
||||
"category": r[1],
|
||||
"city_tier": r[2],
|
||||
"city_example": r[3],
|
||||
"staff_title": r[4],
|
||||
"max_amount": r[5],
|
||||
"unit": r[6],
|
||||
"basis_doc": r[7],
|
||||
"clause": r[8],
|
||||
"effective_date": r[9],
|
||||
}
|
||||
for r in rows
|
||||
]
|
||||
|
||||
|
||||
def upsert_standard(
|
||||
*,
|
||||
category: str,
|
||||
city_tier: str,
|
||||
staff_title: str,
|
||||
max_amount: float,
|
||||
unit: str = "",
|
||||
basis_doc: str = "",
|
||||
clause: str = "",
|
||||
city_example: str = "",
|
||||
) -> dict[str, Any]:
|
||||
"""按 (类别, 城市等级, 职级) 幂等更新标准值。"""
|
||||
if not category or not city_tier or not staff_title:
|
||||
raise ValueError("类别 / 城市等级 / 职级 均不能为空")
|
||||
if max_amount <= 0:
|
||||
raise ValueError("标准值必须大于 0")
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute(
|
||||
"SELECT id FROM expense_standards WHERE category = ? AND city_tier = ? AND staff_title = ?",
|
||||
(category, city_tier, staff_title),
|
||||
)
|
||||
row = cur.fetchone()
|
||||
if row:
|
||||
cur.execute(
|
||||
"""
|
||||
UPDATE expense_standards
|
||||
SET max_amount = ?, unit = ?, basis_doc = ?, clause = ?,
|
||||
city_example = COALESCE(NULLIF(?, ''), city_example), updated_at = ?
|
||||
WHERE id = ?
|
||||
""",
|
||||
(max_amount, unit, basis_doc, clause, city_example, now_unix(), int(row[0])),
|
||||
)
|
||||
standard_id = int(row[0])
|
||||
action = "updated"
|
||||
else:
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO expense_standards
|
||||
(category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(category, city_tier, city_example, staff_title, max_amount, unit, basis_doc, clause, now_unix(), now_unix()),
|
||||
)
|
||||
standard_id = int(cur.lastrowid)
|
||||
action = "created"
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
log = _log()
|
||||
if log:
|
||||
log.info(
|
||||
"standard_upserted standard_id=%s action=%s category=%s tier=%s title=%s stage=standard",
|
||||
standard_id,
|
||||
action,
|
||||
category,
|
||||
city_tier,
|
||||
staff_title,
|
||||
)
|
||||
return {"ok": True, "id": standard_id, "action": action, "category": category, "city_tier": city_tier, "staff_title": staff_title, "max_amount": max_amount}
|
||||
|
||||
|
||||
def list_whitelist() -> list[dict[str, Any]]:
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT id, tax_no, org_name, note, created_at FROM taxid_whitelist ORDER BY id")
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
return [{"id": r[0], "tax_no": r[1], "org_name": r[2], "note": r[3], "created_at": r[4]} for r in rows]
|
||||
|
||||
|
||||
def add_whitelist(tax_no: str, org_name: str, note: str = "") -> dict[str, Any]:
|
||||
tax_no = (tax_no or "").strip()
|
||||
if not tax_no:
|
||||
raise ValueError("税号不能为空")
|
||||
if not org_name:
|
||||
raise ValueError("单位名称不能为空")
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT id FROM taxid_whitelist WHERE tax_no = ?", (tax_no,))
|
||||
if cur.fetchone():
|
||||
raise ValueError(f"税号已存在：{tax_no}")
|
||||
cur.execute(
|
||||
"INSERT INTO taxid_whitelist (tax_no, org_name, note, created_at, updated_at) VALUES (?, ?, ?, ?, ?)",
|
||||
(tax_no, org_name, note, now_unix(), now_unix()),
|
||||
)
|
||||
whitelist_id = int(cur.lastrowid)
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
return {"ok": True, "id": whitelist_id, "tax_no": tax_no, "org_name": org_name}
|
||||
|
||||
|
||||
def remove_whitelist(tax_no: str) -> dict[str, Any]:
|
||||
tax_no = (tax_no or "").strip()
|
||||
if not tax_no:
|
||||
raise ValueError("税号不能为空")
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("DELETE FROM taxid_whitelist WHERE tax_no = ?", (tax_no,))
|
||||
deleted = cur.rowcount
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
if not deleted:
|
||||
raise ValueError(f"白名单中不存在该税号：{tax_no}")
|
||||
return {"ok": True, "removed": tax_no}
|
||||
|
||||
|
||||
def list_holidays(year: Optional[str] = None) -> list[dict[str, Any]]:
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
if year:
|
||||
cur.execute(
|
||||
"SELECT id, holiday_date, holiday_name FROM holiday_calendar WHERE holiday_date LIKE ? ORDER BY holiday_date",
|
||||
(f"{year}%",),
|
||||
)
|
||||
else:
|
||||
cur.execute("SELECT id, holiday_date, holiday_name FROM holiday_calendar ORDER BY holiday_date")
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
return [{"id": r[0], "holiday_date": r[1], "holiday_name": r[2]} for r in rows]
|
||||
|
||||
|
||||
def list_reports(limit: int = 20, verdict: Optional[str] = None) -> list[dict[str, Any]]:
|
||||
init_db()
|
||||
limit = max(1, min(int(limit), 200))
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
if verdict:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT id, batch_no, invoice_number, seller_name, amount, verdict, verify_status, summary, report_file, created_at
|
||||
FROM prereview_reports WHERE verdict = ? ORDER BY id DESC LIMIT ?
|
||||
""",
|
||||
(verdict, limit),
|
||||
)
|
||||
else:
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT id, batch_no, invoice_number, seller_name, amount, verdict, verify_status, summary, report_file, created_at
|
||||
FROM prereview_reports ORDER BY id DESC LIMIT ?
|
||||
""",
|
||||
(limit,),
|
||||
)
|
||||
rows = cur.fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
return [
|
||||
{
|
||||
"id": r[0],
|
||||
"batch_no": r[1],
|
||||
"invoice_number": r[2],
|
||||
"seller_name": r[3],
|
||||
"amount": r[4],
|
||||
"verdict": r[5],
|
||||
"verify_status": r[6],
|
||||
"summary": r[7],
|
||||
"report_file": r[8],
|
||||
"created_at": r[9],
|
||||
}
|
||||
for r in rows
|
||||
]
|
||||
|
||||
|
||||
def get_report(report_id: int) -> dict[str, Any]:
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT * FROM prereview_reports WHERE id = ?", (report_id,))
|
||||
row = cur.fetchone()
|
||||
if not row:
|
||||
raise ValueError(f"初审报告不存在：{report_id}")
|
||||
columns = [d[0] for d in cur.description]
|
||||
report = dict(zip(columns, row))
|
||||
invoice: dict[str, Any] = {}
|
||||
if report.get("invoice_id"):
|
||||
cur.execute("SELECT * FROM invoice_records WHERE id = ?", (report["invoice_id"],))
|
||||
irow = cur.fetchone()
|
||||
if irow:
|
||||
icolumns = [d[0] for d in cur.description]
|
||||
invoice = dict(zip(icolumns, irow))
|
||||
finally:
|
||||
conn.close()
|
||||
for key in ("pass_json", "warn_json", "reject_json"):
|
||||
try:
|
||||
report[key] = json.loads(report.get(key) or "[]")
|
||||
except (TypeError, ValueError):
|
||||
report[key] = []
|
||||
report["invoice"] = invoice
|
||||
return report
|
||||
|
||||
|
||||
def mark_verified(report_id: int, result: str, note: str = "") -> dict[str, Any]:
|
||||
"""回填人工查验结果；查验“不符”则结论升级为驳回。"""
|
||||
result = (result or "").strip().lower()
|
||||
if result not in ("pass", "fail"):
|
||||
raise ValueError("查验结果只接受 pass / fail")
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT id, verdict, reject_json FROM prereview_reports WHERE id = ?", (report_id,))
|
||||
row = cur.fetchone()
|
||||
if not row:
|
||||
raise ValueError(f"初审报告不存在：{report_id}")
|
||||
_report_id, verdict, reject_json = int(row[0]), str(row[1]), row[2] or "[]"
|
||||
|
||||
if result == "pass":
|
||||
verify_status = VERIFY_PASS
|
||||
else:
|
||||
verify_status = VERIFY_FAIL
|
||||
try:
|
||||
rejects = json.loads(reject_json)
|
||||
except (TypeError, ValueError):
|
||||
rejects = []
|
||||
rejects.append(
|
||||
{
|
||||
"rule_id": "V1",
|
||||
"name": "发票真伪",
|
||||
"level": "reject",
|
||||
"detail": "人工查验回填：票面信息与查验平台不符，判定为问题票据。",
|
||||
"rule_ref": "全国增值税发票查验平台（inv-veri.chinatax.gov.cn）",
|
||||
}
|
||||
)
|
||||
reject_json = json.dumps(rejects, ensure_ascii=False)
|
||||
|
||||
new_verdict = "驳回" if result == "fail" else verdict
|
||||
cur.execute(
|
||||
"""
|
||||
UPDATE prereview_reports
|
||||
SET verify_status = ?, verify_note = ?, verify_updated_at = ?,
|
||||
verdict = ?, reject_json = ?
|
||||
WHERE id = ?
|
||||
""",
|
||||
(verify_status, note, now_unix(), new_verdict, reject_json, report_id),
|
||||
)
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
log = _log()
|
||||
if log:
|
||||
log.info(
|
||||
"verify_marked report_id=%s result=%s verdict=%s stage=verify",
|
||||
report_id,
|
||||
result,
|
||||
new_verdict,
|
||||
)
|
||||
return {
|
||||
"ok": True,
|
||||
"report_id": report_id,
|
||||
"verify_status": verify_status,
|
||||
"verdict": new_verdict,
|
||||
"note": note,
|
||||
}
|
||||
|
||||
|
||||
def stats() -> dict[str, Any]:
|
||||
init_db()
|
||||
conn = get_conn()
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT verdict, COUNT(*) FROM prereview_reports GROUP BY verdict")
|
||||
by_verdict = {str(r[0]): int(r[1]) for r in cur.fetchall()}
|
||||
cur.execute("SELECT COUNT(*) FROM invoice_records")
|
||||
invoice_total = int(cur.fetchone()[0])
|
||||
cur.execute("SELECT COUNT(*) FROM taxid_whitelist")
|
||||
whitelist_total = int(cur.fetchone()[0])
|
||||
cur.execute("SELECT COUNT(*) FROM expense_standards")
|
||||
standards_total = int(cur.fetchone()[0])
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT seller_name, COUNT(*) AS cnt FROM invoice_records
|
||||
WHERE seller_name != ''
|
||||
GROUP BY seller_name ORDER BY cnt DESC LIMIT 5
|
||||
"""
|
||||
)
|
||||
top_sellers = [{"seller_name": r[0], "count": int(r[1])} for r in cur.fetchall()]
|
||||
finally:
|
||||
conn.close()
|
||||
total_reports = sum(by_verdict.values())
|
||||
return {
|
||||
"ok": True,
|
||||
"invoice_total": invoice_total,
|
||||
"report_total": total_reports,
|
||||
"verdict_counts": by_verdict,
|
||||
"reject_rate": round(by_verdict.get("驳回", 0) / total_reports, 4) if total_reports else 0.0,
|
||||
"whitelist_total": whitelist_total,
|
||||
"standards_total": standards_total,
|
||||
"top_sellers": top_sellers,
|
||||
}
|
||||
84
scripts/service/task_run_support.py
Normal file
84
scripts/service/task_run_support.py
Normal file
@@ -0,0 +1,84 @@
|
||||
"""任务运行辅助：录屏信息组装与 CLI 输出（通用模板，无业务逻辑）。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any, Dict, Optional
|
||||
|
||||
|
||||
def build_video_info(
|
||||
video_summary: dict[str, Any],
|
||||
compose_context_before: Optional[dict[str, Any]] = None,
|
||||
compose_context_after: Optional[dict[str, Any]] = None,
|
||||
) -> dict[str, Any]:
|
||||
"""将 RpaVideoSession.summary() 转为 finish() / CLI 可用的 video 块。"""
|
||||
return {
|
||||
"enabled": video_summary.get("enabled", False),
|
||||
"path": video_summary.get("path"),
|
||||
"capture_path": video_summary.get("capture_path"),
|
||||
"record_log_path": video_summary.get("record_log_path"),
|
||||
"warnings": list(video_summary.get("warnings") or []),
|
||||
"voiceover_path": video_summary.get("voiceover_path"),
|
||||
"music_path": video_summary.get("music_path"),
|
||||
"audio_warnings": list(video_summary.get("audio_warnings") or []),
|
||||
"compose_context": {
|
||||
**(compose_context_before or {}),
|
||||
"after_compose": compose_context_after or {},
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
def merge_video_into_result_summary(
|
||||
payload: dict[str, Any],
|
||||
video_info: Optional[dict[str, Any]],
|
||||
) -> dict[str, Any]:
|
||||
"""把 video artifact 字段写入 finish() 结果对象（video / video_path 等）。"""
|
||||
if not video_info:
|
||||
return payload
|
||||
payload = dict(payload)
|
||||
payload["video"] = video_info
|
||||
payload["video_path"] = video_info.get("path")
|
||||
payload["raw_video"] = video_info.get("capture_path")
|
||||
payload["video_log"] = video_info.get("record_log_path")
|
||||
payload["video_warnings"] = list(video_info.get("warnings") or [])
|
||||
payload["voiceover_path"] = video_info.get("voiceover_path")
|
||||
payload["music_path"] = video_info.get("music_path")
|
||||
payload["audio_warnings"] = list(video_info.get("audio_warnings") or [])
|
||||
return payload
|
||||
|
||||
|
||||
def _print_video_summary(video_block: Optional[dict[str, Any]]) -> None:
|
||||
"""任务完成后在 CLI 打印录屏路径、日志与诊断信息。"""
|
||||
if not isinstance(video_block, dict) or not video_block.get("enabled"):
|
||||
return
|
||||
record_log = video_block.get("record_log_path")
|
||||
if record_log:
|
||||
print(f"录屏日志：{record_log}")
|
||||
path = video_block.get("path")
|
||||
if path:
|
||||
print(f"录屏路径：{path}")
|
||||
else:
|
||||
print("录屏路径：未生成")
|
||||
warnings = video_block.get("warnings") or []
|
||||
if warnings:
|
||||
print(f"视频诊断：{'; '.join(str(w) for w in warnings)}")
|
||||
capture = video_block.get("capture_path")
|
||||
if capture:
|
||||
print(f"原始录屏路径：{capture}")
|
||||
voiceover = video_block.get("voiceover_path")
|
||||
if voiceover:
|
||||
print(f"旁白路径：{voiceover}")
|
||||
music = video_block.get("music_path")
|
||||
if music:
|
||||
print(f"背景音乐：{music}")
|
||||
audio_warnings = video_block.get("audio_warnings") or []
|
||||
if audio_warnings:
|
||||
print(f"音频诊断：{'; '.join(str(w) for w in audio_warnings)}")
|
||||
compose_ctx = video_block.get("compose_context")
|
||||
if isinstance(compose_ctx, dict):
|
||||
print(
|
||||
"视频合成上下文："
|
||||
f"media_assets_root={compose_ctx.get('media_assets_root', '')}; "
|
||||
f"ffmpeg_path={compose_ctx.get('ffmpeg_path', '')}; "
|
||||
f"background_music_issue={compose_ctx.get('background_music_issue') or ''}; "
|
||||
f"background_music_sample_path={compose_ctx.get('background_music_sample_path', '')}"
|
||||
)
|
||||
699
scripts/service/task_service.py
Normal file
699
scripts/service/task_service.py
Normal file
@@ -0,0 +1,699 @@
|
||||
"""任务编排层：job_context / emit / finish、鉴权与错误转换。
|
||||
|
||||
领域业务在 ``service/prereview_service.py``（预审内核）、
|
||||
``service/standards_service.py``（标准库/白名单/查验回填）、
|
||||
``service/report_service.py``（HTML 意见单），由本模块调用；
|
||||
多入口（CLI / Action / 对话）复用同一内核，禁止按入口分叉业务。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import os
|
||||
import uuid
|
||||
from typing import Any, Optional
|
||||
|
||||
from jiangchang_skill_core import collect_runtime_diagnostics, config, format_runtime_health_lines
|
||||
from jiangchang_skill_core.activity import JobStopped, emit, finish, job_context
|
||||
|
||||
from jiangchang_skill_core.rpa.video_session import RpaVideoSession
|
||||
from service.entitlement_service import check_entitlement
|
||||
from service.task_run_support import (
|
||||
_print_video_summary,
|
||||
build_video_info,
|
||||
merge_video_into_result_summary,
|
||||
)
|
||||
from util.constants import LOG_LOGGER_NAME, PLATFORM_KIT_MIN_VERSION, SKILL_SLUG, SKILL_VERSION
|
||||
from util.logging_config import get_skill_logger, setup_skill_logging
|
||||
from util.runtime_paths import get_db_path, get_input_dir, get_skill_root
|
||||
from db.connection import init_db
|
||||
|
||||
|
||||
def _get_task_logger():
|
||||
try:
|
||||
return get_skill_logger()
|
||||
except RuntimeError:
|
||||
setup_skill_logging(SKILL_SLUG, LOG_LOGGER_NAME)
|
||||
return get_skill_logger()
|
||||
|
||||
|
||||
def _print_json(payload: dict[str, Any]) -> None:
|
||||
print(json.dumps(payload, ensure_ascii=False))
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 模板演示入口（保留：满足 POLICY-RPA-003 录屏集成与 CLI run 契约；不承载本技能业务）
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
async def _run_template_demo(target: Optional[str], input_id: Optional[str]) -> tuple[int, dict[str, Any]]:
|
||||
"""最小 RpaVideoSession 示范：不启动浏览器、不执行业务，仅演示录屏包裹范式。"""
|
||||
batch_id = uuid.uuid4().hex[:12]
|
||||
data_dir = get_skill_root()
|
||||
|
||||
async with RpaVideoSession(
|
||||
skill_slug=SKILL_SLUG,
|
||||
skill_data_dir=data_dir,
|
||||
batch_id=batch_id,
|
||||
title="开始执行示例任务",
|
||||
closing_title="示例任务执行完成",
|
||||
) as video:
|
||||
video.add_step("准备执行示例任务")
|
||||
if target:
|
||||
video.add_step(f"接收目标参数：{target}")
|
||||
if input_id:
|
||||
video.add_step(f"接收输入标识：{input_id}")
|
||||
video.add_step("示例任务执行完成")
|
||||
|
||||
video_summary = video.summary()
|
||||
video_info = build_video_info(video_summary, {}, {})
|
||||
return 1, video_info
|
||||
|
||||
|
||||
def cmd_run(target: Optional[str] = None, input_id: Optional[str] = None) -> int:
|
||||
"""模板演示入口：本技能主能力请使用 prereview 等业务命令。"""
|
||||
log = _get_task_logger()
|
||||
task_type = "demo"
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info(
|
||||
"task_start task_type=%s target_id=%s input_id=%s stage=run",
|
||||
task_type,
|
||||
target,
|
||||
input_id,
|
||||
)
|
||||
emit("开始处理任务", skill=SKILL_SLUG, stage="run")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning(
|
||||
"entitlement_failed task_type=%s stage=auth error_code=ENTITLEMENT_DENIED",
|
||||
task_type,
|
||||
)
|
||||
emit("鉴权失败，已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="auth")
|
||||
finish(
|
||||
status="failed",
|
||||
message=reason,
|
||||
skill=SKILL_SLUG,
|
||||
error_code="ENTITLEMENT_DENIED",
|
||||
stage="auth",
|
||||
)
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
|
||||
log.info("entitlement_ok task_type=%s stage=auth", task_type)
|
||||
log.info("rpa_start stage=demo_prepare task_type=%s", task_type)
|
||||
log.info("task_run_demo_start target_id=%s input_id=%s stage=demo_prepare", target, input_id)
|
||||
rc, video_info = asyncio.run(_run_template_demo(target, input_id))
|
||||
log.info("rpa_done stage=demo_prepare status=ok")
|
||||
log.info(
|
||||
"task_run_demo_done target_id=%s input_id=%s stage=demo_finish status=failed",
|
||||
target,
|
||||
input_id,
|
||||
)
|
||||
_print_video_summary(video_info)
|
||||
|
||||
summary_payload = merge_video_into_result_summary(
|
||||
{
|
||||
"template_demo": True,
|
||||
"target": target,
|
||||
"input_id": input_id,
|
||||
},
|
||||
video_info,
|
||||
)
|
||||
|
||||
log.warning(
|
||||
"task_failed task_type=%s stage=demo_finish error_code=TEMPLATE_STUB",
|
||||
task_type,
|
||||
)
|
||||
finish(
|
||||
status="failed",
|
||||
message="模板演示入口未承载本技能业务",
|
||||
skill=SKILL_SLUG,
|
||||
error_code="TEMPLATE_STUB",
|
||||
**summary_payload,
|
||||
)
|
||||
print("ERROR:TEMPLATE_STUB 这是模板演示入口，请使用 prereview 命令执行票据合规预审。")
|
||||
return rc
|
||||
except JobStopped:
|
||||
raise
|
||||
except Exception:
|
||||
log.exception(
|
||||
"task_failed task_type=%s target_id=%s input_id=%s",
|
||||
task_type,
|
||||
target,
|
||||
input_id,
|
||||
)
|
||||
emit("任务失败，已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="run")
|
||||
finish(
|
||||
status="failed",
|
||||
message="任务执行异常，详见统一日志",
|
||||
skill=SKILL_SLUG,
|
||||
stage="run",
|
||||
)
|
||||
return 1
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 基础命令
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def cmd_config_path() -> int:
|
||||
example_path = os.path.join(get_skill_root(), ".env.example")
|
||||
env_path = config.get_env_file_path() or ""
|
||||
_print_json(
|
||||
{
|
||||
"skill": SKILL_SLUG,
|
||||
"env_path": env_path,
|
||||
"example_path": os.path.abspath(example_path),
|
||||
}
|
||||
)
|
||||
return 0
|
||||
|
||||
|
||||
def cmd_health() -> int:
|
||||
runtime = collect_runtime_diagnostics(
|
||||
skill_slug=SKILL_SLUG,
|
||||
platform_kit_min_version=PLATFORM_KIT_MIN_VERSION,
|
||||
skill_root=get_skill_root(),
|
||||
)
|
||||
example_path = os.path.join(get_skill_root(), ".env.example")
|
||||
env_path = config.get_env_file_path() or ""
|
||||
env_exists = bool(env_path and os.path.isfile(env_path))
|
||||
|
||||
health_status = "failed" if runtime.has_fatal_issues else "ok"
|
||||
lines = [
|
||||
f"{SKILL_SLUG} health: {health_status}",
|
||||
*format_runtime_health_lines(runtime),
|
||||
f"env_path: {env_path}",
|
||||
f"env_exists: {env_exists}",
|
||||
f"example_path: {os.path.abspath(example_path)}",
|
||||
]
|
||||
for line in lines:
|
||||
print(line)
|
||||
return 1 if runtime.has_fatal_issues else 0
|
||||
|
||||
|
||||
def cmd_version() -> int:
|
||||
_print_json({"version": SKILL_VERSION, "skill": SKILL_SLUG})
|
||||
return 0
|
||||
|
||||
|
||||
def cmd_init_db() -> int:
|
||||
"""幂等创建/迁移本地 SQLite 与数据管理元数据（宿主 install/upgrade 可调用）。"""
|
||||
log = _get_task_logger()
|
||||
try:
|
||||
seeded = init_db()
|
||||
except Exception as exc: # noqa: BLE001 — CLI 边界，统一成可机读失败
|
||||
print(
|
||||
json.dumps(
|
||||
{"ok": False, "skill": SKILL_SLUG, "error": str(exc)},
|
||||
ensure_ascii=False,
|
||||
)
|
||||
)
|
||||
return 1
|
||||
_print_json(
|
||||
{
|
||||
"ok": True,
|
||||
"skill": SKILL_SLUG,
|
||||
"db_path": get_db_path(),
|
||||
"seeded": seeded,
|
||||
}
|
||||
)
|
||||
return 0
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 预审主命令
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _load_invoice_payload(invoice_json: str) -> list[dict[str, Any]]:
|
||||
"""解析 --invoice-json：支持 '-'（stdin）、文件路径、内联 JSON。"""
|
||||
from service.prereview_service import _norm_invoice
|
||||
|
||||
raw = (invoice_json or "").strip()
|
||||
if not raw:
|
||||
raise ValueError("缺少发票字段 JSON（--invoice-json）")
|
||||
if raw == "-":
|
||||
import sys
|
||||
|
||||
raw = sys.stdin.read().strip()
|
||||
elif raw.startswith(("[", "{")):
|
||||
pass
|
||||
elif os.path.isfile(raw):
|
||||
with open(raw, encoding="utf-8") as f:
|
||||
raw = f.read().strip()
|
||||
else:
|
||||
raise ValueError(f"--invoice-json 不是有效文件路径或 JSON：{raw[:80]}")
|
||||
|
||||
try:
|
||||
payload = json.loads(raw)
|
||||
except json.JSONDecodeError as exc:
|
||||
raise ValueError(f"发票字段 JSON 解析失败：{exc}") from exc
|
||||
|
||||
if isinstance(payload, dict):
|
||||
# 兼容 {"invoices": [...]} 包装
|
||||
if isinstance(payload.get("invoices"), list):
|
||||
payload = payload["invoices"]
|
||||
else:
|
||||
payload = [payload]
|
||||
if not isinstance(payload, list):
|
||||
raise ValueError("发票字段 JSON 应为对象或对象数组")
|
||||
return [_norm_invoice(item) for item in payload]
|
||||
|
||||
|
||||
def cmd_prereview(
|
||||
*,
|
||||
invoice_json: str,
|
||||
purpose: str = "",
|
||||
project_no: str = "",
|
||||
trip_start: str = "",
|
||||
trip_end: str = "",
|
||||
trip_city: str = "",
|
||||
staff_title: str = "",
|
||||
) -> int:
|
||||
"""票据合规预审主命令：四步审核 → JSON 结果 + 写库 + 《初审意见单》。"""
|
||||
log = _get_task_logger()
|
||||
from service.prereview_service import run_prereview
|
||||
|
||||
context = {
|
||||
"purpose": purpose,
|
||||
"project_no": project_no,
|
||||
"trip_start": trip_start,
|
||||
"trip_end": trip_end,
|
||||
"trip_city": trip_city,
|
||||
"staff_title": staff_title,
|
||||
}
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=prereview stage=prereview")
|
||||
emit("开始票据合规预审", skill=SKILL_SLUG, stage="prereview")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning("entitlement_failed task_type=prereview stage=auth error_code=ENTITLEMENT_DENIED")
|
||||
emit("鉴权失败，已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="auth")
|
||||
finish(status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth")
|
||||
log.error("prereview_denied stage=auth")
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
|
||||
log.info("entitlement_ok task_type=prereview stage=auth")
|
||||
try:
|
||||
invoices = _load_invoice_payload(invoice_json)
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=prereview stage=input error_code=INVOICE_INVALID detail=%s", exc)
|
||||
finish(
|
||||
status="failed",
|
||||
message=str(exc),
|
||||
skill=SKILL_SLUG,
|
||||
error_code="INVOICE_INVALID",
|
||||
stage="input",
|
||||
)
|
||||
log.error("prereview_invalid_input stage=input")
|
||||
print(f"ERROR:INVOICE_INVALID {exc}")
|
||||
return 1
|
||||
|
||||
result = run_prereview(invoices, context)
|
||||
_print_json(result)
|
||||
counts = result["verdict_counts"]
|
||||
emit(
|
||||
f"预审完成：通过 {counts.get('通过', 0)}，待确认 {counts.get('待确认', 0)}，驳回 {counts.get('驳回', 0)}",
|
||||
skill=SKILL_SLUG,
|
||||
stage="prereview",
|
||||
)
|
||||
finish(
|
||||
status="success",
|
||||
message=(
|
||||
f"预审完成 {result['count']} 张：通过 {counts.get('通过', 0)}，"
|
||||
f"待确认 {counts.get('待确认', 0)}，驳回 {counts.get('驳回', 0)}"
|
||||
),
|
||||
skill=SKILL_SLUG,
|
||||
stage="prereview",
|
||||
batch_no=result["batch_no"],
|
||||
)
|
||||
return 0
|
||||
except JobStopped:
|
||||
raise
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=prereview stage=prereview")
|
||||
emit("预审失败，已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="prereview")
|
||||
finish(status="failed", message="预审执行异常，详见统一日志", skill=SKILL_SLUG, stage="prereview")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_prereview_batch_input() -> int:
|
||||
"""批量预审 input/ 目录下的发票 JSON 文件（async Action 入口，任务中心可见进度）。"""
|
||||
log = _get_task_logger()
|
||||
from service.prereview_service import run_prereview
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=prereview_batch stage=batch")
|
||||
emit("开始批量预审", skill=SKILL_SLUG, stage="batch")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning("entitlement_failed task_type=prereview_batch stage=auth")
|
||||
finish(status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth")
|
||||
log.error("prereview_batch_denied stage=auth")
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
|
||||
input_dir = get_input_dir()
|
||||
files = sorted(
|
||||
os.path.join(input_dir, name)
|
||||
for name in os.listdir(input_dir)
|
||||
if name.lower().endswith(".json")
|
||||
)
|
||||
if not files:
|
||||
message = f"input 目录没有可预审的 JSON 文件：{input_dir}"
|
||||
log.warning("task_failed task_type=prereview_batch stage=batch error_code=INVOICE_INPUT_MISSING")
|
||||
finish(status="failed", message=message, skill=SKILL_SLUG, error_code="INVOICE_INPUT_MISSING", stage="batch")
|
||||
log.error("prereview_batch_no_input stage=batch")
|
||||
print(f"ERROR:INVOICE_INPUT_MISSING {message}")
|
||||
return 1
|
||||
|
||||
all_results = []
|
||||
for index, path in enumerate(files, start=1):
|
||||
emit(f"处理 {index}/{len(files)}：{os.path.basename(path)}", skill=SKILL_SLUG, stage="batch")
|
||||
with open(path, encoding="utf-8") as f:
|
||||
payload = json.load(f)
|
||||
if isinstance(payload, dict):
|
||||
invoices = payload.get("invoices") or [payload]
|
||||
context = payload.get("context") or {}
|
||||
else:
|
||||
invoices, context = payload, {}
|
||||
result = run_prereview(invoices, {k: str(v or "") for k, v in context.items()})
|
||||
all_results.append({"file": os.path.basename(path), "batch_no": result["batch_no"], "verdict_counts": result["verdict_counts"]})
|
||||
|
||||
_print_json({"ok": True, "files": len(all_results), "results": all_results})
|
||||
finish(status="success", message=f"批量预审完成，共处理 {len(all_results)} 个文件", skill=SKILL_SLUG, stage="batch")
|
||||
return 0
|
||||
except JobStopped:
|
||||
raise
|
||||
except json.JSONDecodeError as exc:
|
||||
log.exception("task_failed task_type=prereview_batch stage=batch error_code=INVOICE_INVALID")
|
||||
finish(status="failed", message=f"input 目录 JSON 解析失败：{exc}", skill=SKILL_SLUG, error_code="INVOICE_INVALID", stage="batch")
|
||||
print(f"ERROR:INVOICE_INVALID input 目录 JSON 解析失败：{exc}")
|
||||
return 1
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=prereview_batch stage=batch")
|
||||
emit("批量预审失败，已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="batch")
|
||||
finish(status="failed", message="批量预审执行异常，详见统一日志", skill=SKILL_SLUG, stage="batch")
|
||||
return 1
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 报告与查验回填
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def cmd_render_html(report_id: Optional[int] = None) -> int:
|
||||
"""重建《初审意见单》HTML；缺省重建最近 10 份。"""
|
||||
log = _get_task_logger()
|
||||
from service.report_service import build_report_for_invoice
|
||||
from service.standards_service import list_reports
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=render_html report_id=%s stage=render", report_id)
|
||||
try:
|
||||
if report_id:
|
||||
targets = [int(report_id)]
|
||||
else:
|
||||
targets = [r["id"] for r in list_reports(limit=10)]
|
||||
if not targets:
|
||||
log.warning("task_failed task_type=render_html error_code=REPORT_NOT_FOUND stage=render")
|
||||
finish(status="failed", message="暂无可重建的初审报告", skill=SKILL_SLUG, error_code="REPORT_NOT_FOUND", stage="render")
|
||||
print("ERROR:REPORT_NOT_FOUND 暂无可重建的初审报告")
|
||||
return 1
|
||||
written = []
|
||||
for rid in targets:
|
||||
written.append({"report_id": rid, "path": build_report_for_invoice(rid, "")})
|
||||
_print_json({"ok": True, "count": len(written), "reports": written})
|
||||
finish(status="success", message=f"已重建 {len(written)} 份初审意见单", skill=SKILL_SLUG, stage="render")
|
||||
return 0
|
||||
except JobStopped:
|
||||
raise
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=render_html error_code=REPORT_NOT_FOUND detail=%s", exc)
|
||||
finish(status="failed", message=str(exc), skill=SKILL_SLUG, error_code="REPORT_NOT_FOUND", stage="render")
|
||||
print(f"ERROR:REPORT_NOT_FOUND {exc}")
|
||||
return 1
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=render_html stage=render")
|
||||
finish(status="failed", message="意见单重建异常，详见统一日志", skill=SKILL_SLUG, stage="render")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_list_reports(limit: int = 20, verdict: Optional[str] = None) -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import list_reports
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=list_reports limit=%s verdict=%s", limit, verdict)
|
||||
try:
|
||||
reports = list_reports(limit=limit, verdict=verdict)
|
||||
_print_json({"ok": True, "count": len(reports), "reports": reports})
|
||||
return 0
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=list_reports")
|
||||
print("ERROR:QUERY_FAILED 初审记录查询异常，详见统一日志")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_show_report(report_id: int) -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import get_report
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=show_report report_id=%s", report_id)
|
||||
try:
|
||||
report = get_report(int(report_id))
|
||||
_print_json({"ok": True, "report": report})
|
||||
return 0
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=show_report error_code=REPORT_NOT_FOUND detail=%s", exc)
|
||||
print(f"ERROR:REPORT_NOT_FOUND {exc}")
|
||||
return 1
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=show_report")
|
||||
print("ERROR:QUERY_FAILED 初审报告查询异常，详见统一日志")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_mark_verified(report_id: int, result: str, note: str = "") -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import mark_verified
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=mark_verified report_id=%s result=%s", report_id, result)
|
||||
emit("开始回填查验结果", skill=SKILL_SLUG, stage="verify")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning("entitlement_failed task_type=mark_verified stage=auth")
|
||||
finish(status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth")
|
||||
log.error("mark_verified_denied stage=auth")
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
payload = mark_verified(int(report_id), result, note)
|
||||
_print_json(payload)
|
||||
finish(
|
||||
status="success",
|
||||
message=f"报告 {report_id} 查验结果已回填：{payload['verify_status']}",
|
||||
skill=SKILL_SLUG,
|
||||
stage="verify",
|
||||
)
|
||||
return 0
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=mark_verified error_code=VERIFY_INPUT_INVALID detail=%s", exc)
|
||||
finish(status="failed", message=str(exc), skill=SKILL_SLUG, error_code="VERIFY_INPUT_INVALID", stage="verify")
|
||||
print(f"ERROR:VERIFY_INPUT_INVALID {exc}")
|
||||
return 1
|
||||
except JobStopped:
|
||||
raise
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=mark_verified stage=verify")
|
||||
finish(status="failed", message="查验回填异常，详见统一日志", skill=SKILL_SLUG, stage="verify")
|
||||
return 1
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 标准库 / 白名单 / 节假日 / 统计
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def cmd_view_standards(category: Optional[str] = None) -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import list_standards
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=view_standards category=%s", category)
|
||||
try:
|
||||
standards = list_standards(category)
|
||||
_print_json({"ok": True, "count": len(standards), "standards": standards})
|
||||
return 0
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=view_standards")
|
||||
print("ERROR:QUERY_FAILED 标准库查询异常，详见统一日志")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_set_standard(
|
||||
*,
|
||||
category: str,
|
||||
city_tier: str,
|
||||
staff_title: str,
|
||||
max_amount: float,
|
||||
unit: str = "",
|
||||
basis_doc: str = "",
|
||||
clause: str = "",
|
||||
city_example: str = "",
|
||||
) -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import upsert_standard
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=set_standard category=%s tier=%s title=%s", category, city_tier, staff_title)
|
||||
emit("开始更新经费标准", skill=SKILL_SLUG, stage="standard")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning("entitlement_failed task_type=set_standard stage=auth")
|
||||
finish(status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth")
|
||||
log.error("set_standard_denied stage=auth")
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
payload = upsert_standard(
|
||||
category=category,
|
||||
city_tier=city_tier,
|
||||
staff_title=staff_title,
|
||||
max_amount=float(max_amount),
|
||||
unit=unit,
|
||||
basis_doc=basis_doc,
|
||||
clause=clause,
|
||||
city_example=city_example,
|
||||
)
|
||||
_print_json(payload)
|
||||
finish(status="success", message=f"标准已更新：{category}/{city_tier}/{staff_title}", skill=SKILL_SLUG, stage="standard")
|
||||
return 0
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=set_standard error_code=STANDARD_INPUT_INVALID detail=%s", exc)
|
||||
finish(status="failed", message=str(exc), skill=SKILL_SLUG, error_code="STANDARD_INPUT_INVALID", stage="standard")
|
||||
print(f"ERROR:STANDARD_INPUT_INVALID {exc}")
|
||||
return 1
|
||||
except JobStopped:
|
||||
raise
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=set_standard stage=standard")
|
||||
finish(status="failed", message="标准更新异常，详见统一日志", skill=SKILL_SLUG, stage="standard")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_list_whitelist() -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import list_whitelist
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=list_whitelist")
|
||||
try:
|
||||
items = list_whitelist()
|
||||
_print_json({"ok": True, "count": len(items), "items": items})
|
||||
return 0
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=list_whitelist")
|
||||
print("ERROR:QUERY_FAILED 白名单查询异常，详见统一日志")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_add_whitelist(tax_no: str, org_name: str, note: str = "") -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import add_whitelist
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=add_whitelist tax_no=%s", tax_no)
|
||||
emit("开始新增白名单税号", skill=SKILL_SLUG, stage="whitelist")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning("entitlement_failed task_type=add_whitelist stage=auth")
|
||||
finish(status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth")
|
||||
log.error("add_whitelist_denied stage=auth")
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
payload = add_whitelist(tax_no, org_name, note)
|
||||
_print_json(payload)
|
||||
finish(status="success", message=f"已新增税号 {payload['tax_no']}", skill=SKILL_SLUG, stage="whitelist")
|
||||
return 0
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=add_whitelist error_code=WHITELIST_INPUT_INVALID detail=%s", exc)
|
||||
finish(status="failed", message=str(exc), skill=SKILL_SLUG, error_code="WHITELIST_INPUT_INVALID", stage="whitelist")
|
||||
print(f"ERROR:WHITELIST_INPUT_INVALID {exc}")
|
||||
return 1
|
||||
except JobStopped:
|
||||
raise
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=add_whitelist stage=whitelist")
|
||||
finish(status="failed", message="白名单新增异常，详见统一日志", skill=SKILL_SLUG, stage="whitelist")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_remove_whitelist(tax_no: str) -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import remove_whitelist
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=remove_whitelist tax_no=%s", tax_no)
|
||||
emit("开始删除白名单税号", skill=SKILL_SLUG, stage="whitelist")
|
||||
try:
|
||||
ok, reason = check_entitlement(SKILL_SLUG)
|
||||
if not ok:
|
||||
log.warning("entitlement_failed task_type=remove_whitelist stage=auth")
|
||||
finish(status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth")
|
||||
log.error("remove_whitelist_denied stage=auth")
|
||||
print(f"ERROR:ENTITLEMENT_DENIED {reason}")
|
||||
return 1
|
||||
payload = remove_whitelist(tax_no)
|
||||
_print_json(payload)
|
||||
finish(status="success", message=f"已删除税号 {payload['removed']}", skill=SKILL_SLUG, stage="whitelist")
|
||||
return 0
|
||||
except ValueError as exc:
|
||||
log.warning("task_failed task_type=remove_whitelist error_code=WHITELIST_INPUT_INVALID detail=%s", exc)
|
||||
finish(status="failed", message=str(exc), skill=SKILL_SLUG, error_code="WHITELIST_INPUT_INVALID", stage="whitelist")
|
||||
print(f"ERROR:WHITELIST_INPUT_INVALID {exc}")
|
||||
return 1
|
||||
except JobStopped:
|
||||
raise
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=remove_whitelist stage=whitelist")
|
||||
finish(status="failed", message="白名单删除异常，详见统一日志", skill=SKILL_SLUG, stage="whitelist")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_list_holidays(year: Optional[str] = None) -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import list_holidays
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=list_holidays year=%s", year)
|
||||
try:
|
||||
items = list_holidays(year)
|
||||
_print_json({"ok": True, "count": len(items), "holidays": items})
|
||||
return 0
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=list_holidays")
|
||||
print("ERROR:QUERY_FAILED 节假日查询异常，详见统一日志")
|
||||
return 1
|
||||
|
||||
|
||||
def cmd_stats() -> int:
|
||||
log = _get_task_logger()
|
||||
from service.standards_service import stats
|
||||
|
||||
with job_context(skill=SKILL_SLUG):
|
||||
log.info("task_start task_type=stats")
|
||||
try:
|
||||
_print_json(stats())
|
||||
return 0
|
||||
except Exception:
|
||||
log.exception("task_failed task_type=stats")
|
||||
print("ERROR:QUERY_FAILED 统计查询异常，详见统一日志")
|
||||
return 1
|
||||
12
scripts/util/argparse_zh.py
Normal file
12
scripts/util/argparse_zh.py
Normal file
@@ -0,0 +1,12 @@
|
||||
"""argparse 中文错误说明。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
|
||||
|
||||
class ZhArgumentParser(argparse.ArgumentParser):
|
||||
def error(self, message: str) -> None:
|
||||
print(f"参数错误：{message}\n请执行：python main.py -h 查看帮助", file=sys.stderr)
|
||||
self.exit(2)
|
||||
102
scripts/util/cny_amount.py
Normal file
102
scripts/util/cny_amount.py
Normal file
@@ -0,0 +1,102 @@
|
||||
"""中文大写金额与阿拉伯金额互转（发票大小写金额一致性校验用）。
|
||||
|
||||
支持 "壹仟陆佰伍拾元整"、"壹佰贰拾叁元肆角伍分"、"拾元整" 等常见发票大写写法；
|
||||
同时兼容 "¥1,234.50" / "1234.50元" 等阿拉伯写法的清洗。
|
||||
解析失败返回 None，由调用方降级为"待确认"，不做硬性驳回。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import re
|
||||
from typing import Optional
|
||||
|
||||
_DIGITS = {
|
||||
"零": 0,
|
||||
"壹": 1,
|
||||
"贰": 2,
|
||||
"叁": 3,
|
||||
"肆": 4,
|
||||
"伍": 5,
|
||||
"陆": 6,
|
||||
"柒": 7,
|
||||
"捌": 8,
|
||||
"玖": 9,
|
||||
}
|
||||
_SMALL_UNITS = {"拾": 10, "佰": 100, "仟": 1000}
|
||||
_GROUP_UNITS = {"万": 10**4, "亿": 10**8}
|
||||
_ALL_KEYS = r"零壹贰叁肆伍陆柒捌玖拾佰仟万亿元圆角分整"
|
||||
|
||||
_TRAILING_CHARS = re.compile(r"[人民币¥￥\s,，元圆整]|人民币$")
|
||||
|
||||
|
||||
def clean_amount_text(text: str) -> str:
|
||||
"""清洗阿拉伯金额文本，返回可 float() 的字符串（失败返回原文本去掉常见缀符）。"""
|
||||
cleaned = _TRAILING_CHARS.sub("", (text or "").strip())
|
||||
return cleaned
|
||||
|
||||
|
||||
def parse_cny_upper(text: str) -> Optional[float]:
|
||||
"""解析中文大写金额为数值；无法解析时返回 None。"""
|
||||
if not text:
|
||||
return None
|
||||
raw = (text or "").strip()
|
||||
raw = raw.replace("人民币", "").replace("￥", "").replace("¥", "").strip()
|
||||
raw = re.sub(r"[\s,，]", "", raw)
|
||||
if not raw:
|
||||
return None
|
||||
if re.search(r"[0-9]", raw):
|
||||
return None # 纯阿拉伯数字写法不属于大写金额
|
||||
|
||||
if not re.fullmatch(rf"[{_ALL_KEYS}]+", raw):
|
||||
return None
|
||||
|
||||
int_part, frac_part = raw, ""
|
||||
for sep in ("元", "圆"):
|
||||
if sep in raw:
|
||||
int_part, frac_part = raw.split(sep, 1)
|
||||
break
|
||||
|
||||
try:
|
||||
yuan = _parse_integer_section(int_part) if int_part else 0.0
|
||||
except ValueError:
|
||||
return None
|
||||
|
||||
jiao = fen = 0
|
||||
m_jiao = re.search(r"([零壹贰叁肆伍陆柒捌玖])角", frac_part)
|
||||
if m_jiao:
|
||||
jiao = _DIGITS[m_jiao.group(1)]
|
||||
m_fen = re.search(r"([零壹贰叁肆伍陆柒捌玖])分", frac_part)
|
||||
if m_fen:
|
||||
fen = _DIGITS[m_fen.group(1)]
|
||||
return round(yuan + jiao / 10 + fen / 100, 2)
|
||||
|
||||
|
||||
def _parse_integer_section(section: str) -> float:
|
||||
"""解析整数段（元之前的部分），如 壹仟陆佰伍拾 / 拾贰万零壹佰。"""
|
||||
if not section:
|
||||
return 0.0
|
||||
if section == "零":
|
||||
return 0.0
|
||||
|
||||
total = 0.0
|
||||
group_value = 0
|
||||
number = 0
|
||||
for ch in section:
|
||||
if ch in _DIGITS:
|
||||
number = _DIGITS[ch]
|
||||
elif ch in _SMALL_UNITS:
|
||||
unit = _SMALL_UNITS[ch]
|
||||
group_value += (number if number else 1) * unit
|
||||
number = 0
|
||||
elif ch in _GROUP_UNITS:
|
||||
unit = _GROUP_UNITS[ch]
|
||||
group_value = (group_value + number) * unit
|
||||
total += group_value
|
||||
group_value = 0
|
||||
number = 0
|
||||
else:
|
||||
raise ValueError(f"非法大写字符: {ch}")
|
||||
total += group_value + number
|
||||
if total <= 0:
|
||||
raise ValueError("大写金额解析为 0")
|
||||
return total
|
||||
19
scripts/util/config_bootstrap.py
Normal file
19
scripts/util/config_bootstrap.py
Normal file
@@ -0,0 +1,19 @@
|
||||
"""技能配置初始化：.env.example 落盘与用户 .env 缺失项合并。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
|
||||
from jiangchang_skill_core import config
|
||||
|
||||
from util.constants import SKILL_SLUG, SKILL_VERSION
|
||||
from util.runtime_paths import get_skill_root
|
||||
|
||||
|
||||
def bootstrap_skill_config() -> str:
|
||||
"""确保用户数据目录 .env 存在，并追加 .env.example 中的新配置项。"""
|
||||
example_path = os.path.join(get_skill_root(), ".env.example")
|
||||
env_path = config.ensure_env_file(SKILL_SLUG, example_path)
|
||||
comment = f"{SKILL_SLUG} v{SKILL_VERSION}" if SKILL_VERSION else None
|
||||
config.merge_missing_env_keys(example_path, env_path, comment_skill=comment)
|
||||
return env_path
|
||||
6
scripts/util/constants.py
Normal file
6
scripts/util/constants.py
Normal file
@@ -0,0 +1,6 @@
|
||||
"""技能标识、版本与平台公共库约束（票据合规预审助手）。"""
|
||||
|
||||
SKILL_SLUG = "verify-invoice-default"
|
||||
SKILL_VERSION = "1.0.0"
|
||||
LOG_LOGGER_NAME = "openclaw.skill.verify_invoice_default"
|
||||
PLATFORM_KIT_MIN_VERSION = "1.2.11"
|
||||
21
scripts/util/logging_config.py
Normal file
21
scripts/util/logging_config.py
Normal file
@@ -0,0 +1,21 @@
|
||||
"""Re-export unified logging (implementation: jiangchang_skill_core.unified_logging)."""
|
||||
|
||||
from jiangchang_skill_core.unified_logging import (
|
||||
attach_unified_file_handler,
|
||||
ensure_trace_for_process,
|
||||
get_skill_log_file_path,
|
||||
get_skill_logger,
|
||||
get_unified_logs_dir,
|
||||
setup_skill_logging,
|
||||
subprocess_env_with_trace,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
"attach_unified_file_handler",
|
||||
"ensure_trace_for_process",
|
||||
"get_skill_log_file_path",
|
||||
"get_skill_logger",
|
||||
"get_unified_logs_dir",
|
||||
"setup_skill_logging",
|
||||
"subprocess_env_with_trace",
|
||||
]
|
||||
155
scripts/util/runtime_paths.py
Normal file
155
scripts/util/runtime_paths.py
Normal file
@@ -0,0 +1,155 @@
|
||||
"""数据根、技能目录、兄弟技能根路径与用户数据子目录解析。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
|
||||
from jiangchang_skill_core.runtime_env import get_data_root, get_sibling_skills_root, get_user_id
|
||||
|
||||
from util.constants import SKILL_SLUG
|
||||
|
||||
_SCRIPTS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
# 用户可见的上行/下行目录（数据管理「输入/输出」文件浏览器）
|
||||
STANDARD_SUBDIR_INPUT = "input"
|
||||
STANDARD_SUBDIR_OUTPUT = "output"
|
||||
|
||||
# 宿主扫描的 HTML 报告页（数据管理侧栏 htmlPages，非 input/output 文件浏览器）
|
||||
STANDARD_SUBDIR_HTML = "html"
|
||||
|
||||
# 内部/可选目录（不上侧栏）
|
||||
STANDARD_SUBDIR_UPLOADS = "uploads"
|
||||
STANDARD_SUBDIR_CACHE = "cache"
|
||||
STANDARD_SUBDIR_TEMP = "temp"
|
||||
STANDARD_SUBDIR_RPA_ARTIFACTS = "rpa-artifacts"
|
||||
STANDARD_SUBDIR_VIDEOS = "videos"
|
||||
|
||||
CONFIG_KEY_INPUT_DIR = "SKILL_INPUT_DIR"
|
||||
CONFIG_KEY_OUTPUT_DIR = "SKILL_OUTPUT_DIR"
|
||||
CONFIG_KEY_UPLOAD_DIR = "SKILL_UPLOAD_DIR"
|
||||
CONFIG_KEY_CACHE_DIR = "SKILL_CACHE_DIR"
|
||||
CONFIG_KEY_TEMP_DIR = "SKILL_TEMP_DIR"
|
||||
|
||||
|
||||
def get_skill_root() -> str:
|
||||
return os.path.dirname(_SCRIPTS_DIR)
|
||||
|
||||
|
||||
def get_openclaw_root() -> str:
|
||||
return get_sibling_skills_root(_SCRIPTS_DIR)
|
||||
|
||||
|
||||
def get_skills_root() -> str:
|
||||
return get_sibling_skills_root(_SCRIPTS_DIR)
|
||||
|
||||
|
||||
def get_skill_data_dir() -> str:
|
||||
path = os.path.join(get_data_root(), get_user_id(), SKILL_SLUG)
|
||||
os.makedirs(path, exist_ok=True)
|
||||
return path
|
||||
|
||||
|
||||
def get_db_path(filename: str | None = None) -> str:
|
||||
name = filename or f"{SKILL_SLUG}.db"
|
||||
return os.path.join(get_skill_data_dir(), name)
|
||||
|
||||
|
||||
def _normalize_subdir(value: str) -> str:
|
||||
cleaned = value.strip().strip("/\\")
|
||||
if not cleaned or cleaned.startswith(".."):
|
||||
raise ValueError(f"invalid skill data subdir: {value!r}")
|
||||
return cleaned.replace("/", os.sep).replace("\\", os.sep)
|
||||
|
||||
|
||||
def resolve_data_path(
|
||||
config_key: str | None,
|
||||
default_subdir: str,
|
||||
*,
|
||||
create: bool = True,
|
||||
) -> str:
|
||||
"""解析技能拥有路径：空配置 → 数据目录下 default_subdir；相对 → 相对数据目录；绝对 → 原样。"""
|
||||
from jiangchang_skill_core import config
|
||||
|
||||
base = get_skill_data_dir()
|
||||
raw = (config.get(config_key) or "").strip() if config_key else ""
|
||||
if not raw:
|
||||
path = os.path.join(base, _normalize_subdir(default_subdir))
|
||||
elif os.path.isabs(raw):
|
||||
path = os.path.abspath(os.path.expanduser(raw))
|
||||
else:
|
||||
path = os.path.join(base, _normalize_subdir(raw))
|
||||
|
||||
if create:
|
||||
os.makedirs(path, exist_ok=True)
|
||||
return path
|
||||
|
||||
|
||||
def resolve_input_path(raw_path: str, *, create_parent: bool = False) -> str:
|
||||
"""解析 CLI 输入路径：绝对路径原样；相对路径相对 {skill_data_dir}/input/。"""
|
||||
text = (raw_path or "").strip()
|
||||
if not text:
|
||||
return text
|
||||
expanded = os.path.expanduser(text)
|
||||
if os.path.isabs(expanded):
|
||||
return os.path.abspath(expanded)
|
||||
input_dir = get_input_dir(create=create_parent)
|
||||
return os.path.abspath(os.path.join(input_dir, os.path.normpath(expanded)))
|
||||
|
||||
|
||||
def get_input_dir(*, create: bool = True) -> str:
|
||||
"""上行：用户/Agent 放入的待处理文件（清单、CSV、产品表等）。"""
|
||||
return resolve_data_path(CONFIG_KEY_INPUT_DIR, STANDARD_SUBDIR_INPUT, create=create)
|
||||
|
||||
|
||||
def get_output_dir(*, create: bool = True) -> str:
|
||||
"""下行：技能/宿主交给用户拿走的文件（视频、报表、导出 CSV 等）。"""
|
||||
return resolve_data_path(CONFIG_KEY_OUTPUT_DIR, STANDARD_SUBDIR_OUTPUT, create=create)
|
||||
|
||||
|
||||
def get_uploads_dir(*, create: bool = True) -> str:
|
||||
return resolve_data_path(CONFIG_KEY_UPLOAD_DIR, STANDARD_SUBDIR_UPLOADS, create=create)
|
||||
|
||||
|
||||
def get_cache_dir(*, create: bool = True) -> str:
|
||||
return resolve_data_path(CONFIG_KEY_CACHE_DIR, STANDARD_SUBDIR_CACHE, create=create)
|
||||
|
||||
|
||||
def get_temp_dir(*, create: bool = True) -> str:
|
||||
return resolve_data_path(CONFIG_KEY_TEMP_DIR, STANDARD_SUBDIR_TEMP, create=create)
|
||||
|
||||
|
||||
def get_rpa_artifacts_dir(batch_id: str, *, create: bool = True) -> str:
|
||||
batch = (batch_id or "").strip() or "default"
|
||||
path = os.path.join(get_skill_data_dir(), STANDARD_SUBDIR_RPA_ARTIFACTS, batch)
|
||||
if create:
|
||||
os.makedirs(path, exist_ok=True)
|
||||
return path
|
||||
|
||||
|
||||
def get_videos_dir(*, create: bool = True) -> str:
|
||||
path = os.path.join(get_skill_data_dir(), STANDARD_SUBDIR_VIDEOS)
|
||||
if create:
|
||||
os.makedirs(path, exist_ok=True)
|
||||
return path
|
||||
|
||||
|
||||
def get_html_dir(*, create: bool = True) -> str:
|
||||
"""HTML 报告页目录：宿主扫描 `{skill_data}/html/*.html`（一层、非递归）。"""
|
||||
path = os.path.join(get_skill_data_dir(), STANDARD_SUBDIR_HTML)
|
||||
if create:
|
||||
os.makedirs(path, exist_ok=True)
|
||||
return path
|
||||
|
||||
|
||||
def list_resolved_data_paths() -> dict[str, str]:
|
||||
"""供 health / config-path 输出已解析的数据子目录（不创建目录）。"""
|
||||
return {
|
||||
"skill_data_dir": get_skill_data_dir(),
|
||||
"input_dir": get_input_dir(create=False),
|
||||
"output_dir": get_output_dir(create=False),
|
||||
"html_dir": get_html_dir(create=False),
|
||||
"uploads_dir": get_uploads_dir(create=False),
|
||||
"cache_dir": get_cache_dir(create=False),
|
||||
"temp_dir": get_temp_dir(create=False),
|
||||
"videos_dir": get_videos_dir(create=False),
|
||||
}
|
||||
227
scripts/util/slug_naming.py
Normal file
227
scripts/util/slug_naming.py
Normal file
@@ -0,0 +1,227 @@
|
||||
"""技能 slug 语义校验（verb-noun-platform 规范，见 development/NAMING.md）。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import re
|
||||
|
||||
_KEBAB_SLUG = re.compile(r"^[a-z0-9]+(?:-[a-z0-9]+)*$")
|
||||
_MAX_SLUG_LENGTH = 48
|
||||
_MIN_SEGMENTS = 3
|
||||
_MAX_SEGMENTS = 5
|
||||
_MAX_NOUN_SEGMENTS = 3
|
||||
|
||||
_SKILL_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
_NAMING_DIR = os.path.join(_SKILL_ROOT, "assets", "naming")
|
||||
|
||||
_BUILTIN_VERBS = frozenset(
|
||||
{
|
||||
"scrape",
|
||||
"receive",
|
||||
"download",
|
||||
"export",
|
||||
"add",
|
||||
"fill",
|
||||
"enter",
|
||||
"bind",
|
||||
"withdraw",
|
||||
"reconcile",
|
||||
"verify",
|
||||
"submit",
|
||||
"query",
|
||||
"process",
|
||||
"reply",
|
||||
"generate",
|
||||
"draft",
|
||||
"review",
|
||||
"confirm",
|
||||
"track",
|
||||
"upload",
|
||||
"match",
|
||||
"triage",
|
||||
"ship",
|
||||
"reprice",
|
||||
}
|
||||
)
|
||||
_BUILTIN_PLATFORMS = frozenset(
|
||||
{
|
||||
"amazon",
|
||||
"shopee",
|
||||
"xinghang",
|
||||
"alibaba",
|
||||
"kingdee",
|
||||
"icbc",
|
||||
"pingpong",
|
||||
"worldfirst",
|
||||
"paypal",
|
||||
"lianlian",
|
||||
"single-window",
|
||||
"export-rebate",
|
||||
"etax",
|
||||
}
|
||||
)
|
||||
_BUILTIN_SCOPES = frozenset({"all", "batch", "multi", "default"})
|
||||
|
||||
LEGACY_EXEMPT_SLUGS: frozenset[str] = frozenset(
|
||||
{
|
||||
"account-manager",
|
||||
"skill-template",
|
||||
"your-skill-slug",
|
||||
"your_skill_slug",
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
def load_wordlist(filename: str) -> frozenset[str]:
|
||||
"""从 assets/naming/{filename} 加载词表；文件不存在时 fallback 内置集合。"""
|
||||
fallbacks = {
|
||||
"verbs.txt": _BUILTIN_VERBS,
|
||||
"platforms.txt": _BUILTIN_PLATFORMS,
|
||||
"scopes.txt": _BUILTIN_SCOPES,
|
||||
}
|
||||
fallback = fallbacks.get(filename, frozenset())
|
||||
path = os.path.join(_NAMING_DIR, filename)
|
||||
if not os.path.isfile(path):
|
||||
return fallback
|
||||
words: set[str] = set()
|
||||
with open(path, encoding="utf-8") as f:
|
||||
for line in f:
|
||||
word = line.strip()
|
||||
if word and not word.startswith("#"):
|
||||
words.add(word)
|
||||
return frozenset(words) if words else fallback
|
||||
|
||||
|
||||
def parse_slug_segments(slug: str) -> list[str]:
|
||||
"""按 '-' 分割；非法字符由上层 kebab 校验处理。"""
|
||||
slug = slug.strip()
|
||||
if not slug:
|
||||
return []
|
||||
return slug.split("-")
|
||||
|
||||
|
||||
def _match_platform_suffix(
|
||||
segments: list[str],
|
||||
platforms: frozenset[str],
|
||||
) -> tuple[str, int] | None:
|
||||
"""返回 (platform_name, platform_part_count) 或 None。"""
|
||||
for platform in sorted(platforms, key=lambda item: (-item.count("-"), item)):
|
||||
parts = platform.split("-")
|
||||
part_count = len(parts)
|
||||
if len(segments) < part_count + 2:
|
||||
continue
|
||||
if segments[-part_count:] == parts:
|
||||
return platform, part_count
|
||||
return None
|
||||
|
||||
|
||||
def classify_slug_form(segments: list[str], *, slug: str = "") -> str:
|
||||
"""
|
||||
返回: 'legacy_exempt' | 'standard' | 'scope' | 'invalid'
|
||||
|
||||
- legacy_exempt: slug in LEGACY_EXEMPT_SLUGS
|
||||
- standard: 末段匹配 platforms, 首段 in verbs, 3<=len<=5
|
||||
- scope: 末段 in scopes, 首段 in verbs, 3<=len<=5
|
||||
- invalid: 其他
|
||||
"""
|
||||
if slug in LEGACY_EXEMPT_SLUGS:
|
||||
return "legacy_exempt"
|
||||
|
||||
if not segments or not (_MIN_SEGMENTS <= len(segments) <= _MAX_SEGMENTS):
|
||||
return "invalid"
|
||||
|
||||
verbs = load_wordlist("verbs.txt")
|
||||
platforms = load_wordlist("platforms.txt")
|
||||
scopes = load_wordlist("scopes.txt")
|
||||
|
||||
if segments[0] not in verbs:
|
||||
return "invalid"
|
||||
|
||||
if segments[-1] in scopes:
|
||||
return "scope"
|
||||
|
||||
if _match_platform_suffix(segments, platforms) is not None:
|
||||
return "standard"
|
||||
|
||||
return "invalid"
|
||||
|
||||
|
||||
def validate_slug_semantics(
|
||||
slug: str,
|
||||
*,
|
||||
strict: bool = True,
|
||||
) -> tuple[list[str], list[str]]:
|
||||
"""返回 (errors, warnings)。"""
|
||||
slug = slug.strip()
|
||||
if slug in LEGACY_EXEMPT_SLUGS:
|
||||
return [], []
|
||||
|
||||
errors: list[str] = []
|
||||
warnings: list[str] = []
|
||||
|
||||
if not _KEBAB_SLUG.fullmatch(slug):
|
||||
msg = f"slug 必须为 kebab-case（小写字母、数字、连字符）：{slug!r}"
|
||||
(errors if strict else warnings).append(msg)
|
||||
return errors, warnings
|
||||
|
||||
if len(slug) > _MAX_SLUG_LENGTH:
|
||||
msg = f"slug 长度不得超过 {_MAX_SLUG_LENGTH} 字符（当前 {len(slug)}）：{slug!r}"
|
||||
(errors if strict else warnings).append(msg)
|
||||
|
||||
segments = parse_slug_segments(slug)
|
||||
segment_count = len(segments)
|
||||
|
||||
if segment_count < _MIN_SEGMENTS:
|
||||
msg = (
|
||||
f"slug 段数应为 {_MIN_SEGMENTS}～{_MAX_SEGMENTS}（verb-noun-platform），"
|
||||
f"当前为 {segment_count} 段：{slug!r}；新技能禁止 2 段无平台形态（如 reconcile-finance）"
|
||||
)
|
||||
(errors if strict else warnings).append(msg)
|
||||
return errors, warnings
|
||||
|
||||
if segment_count > _MAX_SEGMENTS:
|
||||
msg = f"slug 段数不得超过 {_MAX_SEGMENTS}（当前 {segment_count}）：{slug!r}"
|
||||
(errors if strict else warnings).append(msg)
|
||||
|
||||
verbs = load_wordlist("verbs.txt")
|
||||
platforms = load_wordlist("platforms.txt")
|
||||
scopes = load_wordlist("scopes.txt")
|
||||
|
||||
if segments[0] in platforms and segments[0] not in verbs:
|
||||
msg = f"平台名不应放在 slug 最前：{slug!r}（应为 verb-noun-platform）"
|
||||
(errors if strict else warnings).append(msg)
|
||||
|
||||
if segments[0] not in verbs:
|
||||
msg = f"slug 首段应为动词白名单中的词：{segments[0]!r}（slug={slug!r}）"
|
||||
(errors if strict else warnings).append(msg)
|
||||
|
||||
form = classify_slug_form(segments, slug=slug)
|
||||
if form == "scope":
|
||||
if segments[-1] not in scopes:
|
||||
msg = f"slug 末段应为 scope 白名单中的词：{segments[-1]!r}（slug={slug!r}）"
|
||||
(errors if strict else warnings).append(msg)
|
||||
noun_count = segment_count - 2
|
||||
elif form == "standard":
|
||||
platform_match = _match_platform_suffix(segments, platforms)
|
||||
if platform_match is None:
|
||||
msg = f"slug 末段应匹配平台白名单：{slug!r}"
|
||||
(errors if strict else warnings).append(msg)
|
||||
noun_count = 0
|
||||
else:
|
||||
_platform_name, platform_part_count = platform_match
|
||||
noun_count = segment_count - 1 - platform_part_count
|
||||
else:
|
||||
if segment_count >= _MIN_SEGMENTS:
|
||||
if segments[-1] not in scopes:
|
||||
platform_match = _match_platform_suffix(segments, platforms)
|
||||
if platform_match is None:
|
||||
msg = f"slug 末段应匹配平台或 scope 白名单：{slug!r}"
|
||||
(errors if strict else warnings).append(msg)
|
||||
noun_count = 0
|
||||
|
||||
if noun_count > _MAX_NOUN_SEGMENTS:
|
||||
warnings.append(
|
||||
f"slug 中间名词段建议不超过 {_MAX_NOUN_SEGMENTS} 个词（当前 {noun_count}）：{slug!r}"
|
||||
)
|
||||
|
||||
return errors, warnings
|
||||
20
scripts/util/timeutil.py
Normal file
20
scripts/util/timeutil.py
Normal file
@@ -0,0 +1,20 @@
|
||||
"""时间戳与 ISO 展示。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import time
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
|
||||
|
||||
def now_unix() -> int:
|
||||
return int(time.time())
|
||||
|
||||
|
||||
def unix_to_iso(ts: Optional[int]) -> Optional[str]:
|
||||
if ts is None:
|
||||
return None
|
||||
try:
|
||||
return datetime.fromtimestamp(int(ts)).isoformat(timespec="seconds")
|
||||
except (ValueError, OSError, OverflowError):
|
||||
return None
|
||||
Reference in New Issue
Block a user