Invoice-Downloader/document_types.py at main · EthanYoQ/Invoice-Downloader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
"""
文档类型注册表 — 统一票据/订单分类配置

定义所有已支持的文档类型及其归档和校验规则。
新增类型只需在 DOCUMENT_TYPES 字典中加一个条目。
"""

MANUAL_REVIEW_FOLDER = "待人工复核"
NON_TARGET_COMPANY_FOLDER = "非目标公司发票"
PERSONAL_NON_REIMBURSEMENT_FOLDER = "个人非报销发票"

DOCUMENT_TYPES = {
    # 现有发票类型
    "打车":     {"exempt_from_purchaser_check": True,  "archive_folder": "打车"},
    "行程单":   {"exempt_from_purchaser_check": True,  "archive_folder": "打车"},
    "火车票":   {"exempt_from_purchaser_check": True,  "archive_folder": "火车票"},
    "机票":     {"exempt_from_purchaser_check": True,  "archive_folder": "机票"},
    "住宿发票": {"exempt_from_purchaser_check": False, "archive_folder": "住宿发票"},
    "住宿水单": {"exempt_from_purchaser_check": True,  "archive_folder": "住宿发票"},
    "餐饮":     {"exempt_from_purchaser_check": False, "archive_folder": "餐饮"},
    "过路费":   {"exempt_from_purchaser_check": True,  "archive_folder": "过路费"},
    "定额发票": {"exempt_from_purchaser_check": True,  "archive_folder": "定额发票"},
    "其他":     {"exempt_from_purchaser_check": False, "archive_folder": "其他"},
    # 国旅运通新增
    "航班行程单": {"exempt_from_purchaser_check": True, "archive_folder": "机票"},
    "住宿确认单": {"exempt_from_purchaser_check": True, "archive_folder": "住宿发票"},
    # 差旅服务费 (GBT Travel Services SCCT 发票)
    "差旅服务费": {"exempt_from_purchaser_check": True, "archive_folder": "差旅服务费"},
    # 隔离类型
    "非目标公司发票": {"exempt_from_purchaser_check": True, "archive_folder": NON_TARGET_COMPANY_FOLDER},
    "个人非报销发票": {"exempt_from_purchaser_check": True, "archive_folder": PERSONAL_NON_REIMBURSEMENT_FOLDER},
}


def get_document_type_names() -> list[str]:
    """Return the canonical document type vocabulary in declaration order."""
    return list(DOCUMENT_TYPES)


def normalize_document_type(doc_type: str) -> str:
    """Map free-form type text to the closest registered document type."""
    text = str(doc_type or "").strip()
    if text in DOCUMENT_TYPES:
        return text

    if "火车" in text or "高铁" in text:
        return "火车票"
    if "航班行程" in text:
        return "航班行程单"
    if "机票" in text or "航空" in text or "航班" in text:
        return "机票"
    if "水单" in text or "folio" in text.lower() or "账单" in text or "明细" in text:
        return "住宿水单"
    if "确认单" in text:
        return "住宿确认单"
    if "住宿" in text or "酒店" in text:
        return "住宿发票"
    if "行程单" in text or "报销单" in text:
        return "行程单"
    if "打车" in text or "滴滴" in text or "出租" in text or "高德" in text:
        return "打车"
    if "餐饮" in text or "餐" in text:
        return "餐饮"
    if "过路" in text or "高速" in text:
        return "过路费"
    if "定额" in text:
        return "定额发票"
    if "非目标" in text:
        return "非目标公司发票"
    if "个人非报销" in text:
        return "个人非报销发票"
    if "差旅服务" in text:
        return "差旅服务费"
    return "其他"


def is_exempt_type(doc_type: str) -> bool:
    """检查文档类型是否豁免购买方校验。"""
    entry = DOCUMENT_TYPES.get(doc_type)
    if entry:
        return entry["exempt_from_purchaser_check"]
    return False


def get_archive_folder(doc_type: str) -> str:
    """获取文档类型对应的归档目录。"""
    entry = DOCUMENT_TYPES.get(doc_type)
    if entry:
        return entry["archive_folder"]
    return doc_type or "其他"