334 lines
11 KiB
Python
334 lines
11 KiB
Python
"""
|
||
陆运选项规则：运输类型、线路类别、运输分类。
|
||
|
||
本文件职责：从销售原话收取标准项、校验是否在名单内、生成补问清单（含填写样例表）、判断「确定查价」意图。
|
||
三项各自必填，组合不拦截查价。拉群必须对上填写样例。运输分类句里的「中港/东南亚」不得冒充类型或线路。
|
||
禁止：改六态；发明枚举；把选项当货物字段；再要单独的中港装载方式。
|
||
线程：纯函数，无共享可变状态，可在 Worker 槽调用。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from pathlib import Path
|
||
from typing import Any
|
||
|
||
TRANSPORT_TYPES = (
|
||
"国内运输拼车",
|
||
"国内运输整车",
|
||
"跨境集拼",
|
||
"跨境整车",
|
||
"中港整车",
|
||
"中港零担/集拼",
|
||
)
|
||
ROUTES = (
|
||
"国内长途/零担",
|
||
"国内城配/拖车/打包",
|
||
"中港/中亚/中欧",
|
||
"东南亚",
|
||
)
|
||
LAND_CLASSES = (
|
||
"普件(东南亚、中港常用线路/常规车型)",
|
||
"普件(东南亚、中亚、中欧涉及国内+国外段)",
|
||
"普件(东南亚、中亚、中欧涉及境外段清派等特殊要求)",
|
||
"急件",
|
||
)
|
||
|
||
# 填写样例，同时作为拉群门禁：对不上的类型+线路可以查价，但不能拉群。
|
||
SAMPLE_PAIRS = (
|
||
("国内运输拼车", "国内长途/零担"),
|
||
("国内运输拼车", "国内城配/拖车/打包"),
|
||
("国内运输整车", "国内长途/零担"),
|
||
("国内运输整车", "国内城配/拖车/打包"),
|
||
("跨境集拼", "中港/中亚/中欧"),
|
||
("跨境集拼", "东南亚"),
|
||
("跨境整车", "中港/中亚/中欧"),
|
||
("跨境整车", "东南亚"),
|
||
("中港整车", "中港/中亚/中欧"),
|
||
("中港零担/集拼", "中港/中亚/中欧"),
|
||
)
|
||
|
||
# 口语别名 → 标准运输类型。长词优先。禁止把单独「中港」收成类型。
|
||
_TYPE_ALIASES = (
|
||
("国内运输拼车", "国内运输拼车"),
|
||
("国内运输整车", "国内运输整车"),
|
||
("中港零担/集拼", "中港零担/集拼"),
|
||
("中港零担", "中港零担/集拼"),
|
||
("中港集拼", "中港零担/集拼"),
|
||
("中港整车", "中港整车"),
|
||
("国内拼车", "国内运输拼车"),
|
||
("国内整车", "国内运输整车"),
|
||
("跨境集拼", "跨境集拼"),
|
||
("跨境整车", "跨境整车"),
|
||
)
|
||
|
||
_ROUTE_ALIASES = (
|
||
("国内城配/拖车/打包", "国内城配/拖车/打包"),
|
||
("国内长途/零担", "国内长途/零担"),
|
||
("中港/中亚/中欧", "中港/中亚/中欧"),
|
||
("国内城配", "国内城配/拖车/打包"),
|
||
("国内长途", "国内长途/零担"),
|
||
("东南亚", "东南亚"),
|
||
)
|
||
|
||
# 分类全称从长到短；「急件」单独出现才算急件。
|
||
_CLASS_CANON = tuple(
|
||
sorted(LAND_CLASSES, key=len, reverse=True)
|
||
)
|
||
|
||
_CONFIRM_PHRASES = ("可以查了", "确定", "确认", "查价")
|
||
_CONFIRM_EXCLUDE = ("始发", "目的", "体积", "车型", "改成", "重量")
|
||
|
||
|
||
def canonicalize_transport_type(raw: str) -> str:
|
||
"""
|
||
把销售原话收成标准运输类型。
|
||
|
||
参数：任意一句。返回 TRANSPORT_TYPES 之一或空串。
|
||
只说「中港」不算选完，避免分类句里的中港把类型改掉。
|
||
副作用：无。先匹配更长别名。
|
||
"""
|
||
text = (raw or "").strip()
|
||
if not text:
|
||
return ""
|
||
for needle, canon in _TYPE_ALIASES:
|
||
if needle in text:
|
||
return canon
|
||
return ""
|
||
|
||
|
||
def canonicalize_route(raw: str) -> str:
|
||
"""把原话收成标准线路类别；对不上返回空串。不把单独「中港」收成中港/中亚/中欧。"""
|
||
text = (raw or "").strip()
|
||
if not text:
|
||
return ""
|
||
for needle, canon in _ROUTE_ALIASES:
|
||
if needle in text:
|
||
return canon
|
||
return ""
|
||
|
||
|
||
def canonicalize_land_class(raw: str) -> str:
|
||
"""
|
||
把原话收成标准运输分类。
|
||
|
||
先匹配完整分类句；整句只有「急件」才收急件，避免普件描述被截成急件。
|
||
"""
|
||
text = (raw or "").strip()
|
||
if not text:
|
||
return ""
|
||
for canon in _CLASS_CANON:
|
||
if canon == "急件":
|
||
continue
|
||
if canon in text:
|
||
return canon
|
||
stripped = text.replace("运输分类", "").replace("：", "").replace(":", "").strip()
|
||
if stripped == "急件" or text.strip() == "急件":
|
||
return "急件"
|
||
if "急件" in text and "普件" not in text:
|
||
return "急件"
|
||
return ""
|
||
|
||
|
||
def sample_pair_allowed(land_type: str, route: str) -> bool:
|
||
"""
|
||
类型+线路是否在填写样例里。
|
||
|
||
查价不调用本函数；点「拉产品进群协同」时必须为真，否则不建群。
|
||
入参先收成标准项。副作用：无。
|
||
"""
|
||
t = canonicalize_transport_type(land_type) or (land_type or "").strip()
|
||
r = canonicalize_route(route) or (route or "").strip()
|
||
return (t, r) in SAMPLE_PAIRS
|
||
|
||
|
||
def canonicalize_hk_load(raw: str) -> str:
|
||
"""
|
||
旧中港装载方式不再收取。
|
||
|
||
保留函数名避免旧调用处报缺符号；恒返回空串。新单用运输类型「中港整车」「中港零担/集拼」。
|
||
"""
|
||
_ = raw
|
||
return ""
|
||
|
||
|
||
def harvest_land_options(text: str, current: dict[str, str] | None = None) -> dict[str, str]:
|
||
"""
|
||
从一句原话收取陆运选项，合并到已有值上。
|
||
|
||
先收运输分类并把命中句从原文挖掉，再收类型/线路，避免分类里的东南亚、中港冒充选项。
|
||
已有非空项默认保留；本句明确写出另一标准项时才覆盖。
|
||
已有值对不上标准枚举则丢掉。不改货物字段。返回新 dict，不修改 current。
|
||
"""
|
||
merged = {
|
||
k: str(v).strip()
|
||
for k, v in dict(current or {}).items()
|
||
if str(v).strip()
|
||
}
|
||
_keep_canonical_options(merged)
|
||
blob = text or ""
|
||
found_class = canonicalize_land_class(blob)
|
||
remainder = blob.replace(found_class, " ") if found_class else blob
|
||
found_type = canonicalize_transport_type(remainder)
|
||
found_route = canonicalize_route(remainder)
|
||
if found_type:
|
||
merged["运输类型"] = found_type
|
||
if found_route:
|
||
merged["线路类别"] = found_route
|
||
if found_class:
|
||
merged["运输分类"] = found_class
|
||
merged.pop("中港装载方式", None)
|
||
return merged
|
||
|
||
|
||
def _keep_canonical_options(merged: dict[str, str]) -> None:
|
||
"""只留能对上枚举的选项；模型胡写的「陆运」不能冒充运输类型。旧中港两项作废。"""
|
||
t = canonicalize_transport_type(merged.get("运输类型") or "")
|
||
if t:
|
||
merged["运输类型"] = t
|
||
else:
|
||
merged.pop("运输类型", None)
|
||
r = canonicalize_route(merged.get("线路类别") or "")
|
||
if r:
|
||
merged["线路类别"] = r
|
||
else:
|
||
merged.pop("线路类别", None)
|
||
c = canonicalize_land_class(merged.get("运输分类") or "")
|
||
if c:
|
||
merged["运输分类"] = c
|
||
else:
|
||
merged.pop("运输分类", None)
|
||
merged.pop("中港装载方式", None)
|
||
|
||
|
||
def validate_land_options(opts: dict[str, str]) -> dict[str, Any]:
|
||
"""
|
||
校验三项是否都在名单内。组合不拦截。
|
||
|
||
返回 ok / missing / errors。对不上名单的项当缺项，不写联动纠正句。
|
||
副作用：无。
|
||
"""
|
||
src = {k: str(v).strip() for k, v in dict(opts or {}).items() if str(v).strip()}
|
||
t = canonicalize_transport_type(src.get("运输类型") or "")
|
||
r = canonicalize_route(src.get("线路类别") or "")
|
||
c = canonicalize_land_class(src.get("运输分类") or "")
|
||
|
||
missing: list[str] = []
|
||
if t not in TRANSPORT_TYPES:
|
||
missing.append("运输类型")
|
||
t = ""
|
||
if r not in ROUTES:
|
||
missing.append("线路类别")
|
||
r = ""
|
||
if c not in LAND_CLASSES:
|
||
missing.append("运输分类")
|
||
c = ""
|
||
|
||
ok = not missing
|
||
return {
|
||
"ok": ok,
|
||
"missing": missing,
|
||
"errors": [],
|
||
"normalized": {
|
||
"运输类型": t,
|
||
"线路类别": r,
|
||
"运输分类": c,
|
||
"中港装载方式": "",
|
||
},
|
||
}
|
||
|
||
|
||
def _class_prompt_lines() -> list[str]:
|
||
lines = ["运输分类（必填）：可选择回复，"]
|
||
for i, name in enumerate(LAND_CLASSES, start=1):
|
||
lines.append(f"{i}.{name}")
|
||
return lines
|
||
|
||
|
||
def option_prompt_needs_sample(*, current: dict[str, str]) -> bool:
|
||
"""运输类型或线路类别缺任一，就要发填写样例图。只缺分类不发。"""
|
||
src = {k: str(v).strip() for k, v in dict(current or {}).items() if str(v).strip()}
|
||
t = canonicalize_transport_type(src.get("运输类型") or "")
|
||
r = canonicalize_route(src.get("线路类别") or "")
|
||
return t not in TRANSPORT_TYPES or r not in ROUTES
|
||
|
||
|
||
def sample_image_path() -> Path:
|
||
"""填写样例表 PNG。和源码放一起，随 agent 发布包走。"""
|
||
return Path(__file__).resolve().parent.parent / "assets" / "land_sample_pairs.png"
|
||
|
||
|
||
def sample_fallback_text() -> str:
|
||
"""
|
||
图片发不出去时的文字兜底。
|
||
|
||
一行一种搭配，不再画框线。
|
||
"""
|
||
lines = ["填写样例"]
|
||
for i, (typ, route) in enumerate(SAMPLE_PAIRS, start=1):
|
||
lines.append(f"{i}.{typ} + {route}")
|
||
return "\n".join(lines)
|
||
|
||
|
||
def _sample_table() -> str:
|
||
"""兼容旧名：补问正文不再内嵌样例，只作图片失败兜底。"""
|
||
return sample_fallback_text()
|
||
|
||
|
||
def option_prompt(*, current: dict[str, str]) -> str:
|
||
"""
|
||
按已识别项生成补问清单。
|
||
|
||
缺哪项列哪项。运输类型或线路类别缺任一则提示见图，样例图另发出站。只缺分类不附图。
|
||
不修改 current。
|
||
"""
|
||
src = {k: str(v).strip() for k, v in dict(current or {}).items() if str(v).strip()}
|
||
t = canonicalize_transport_type(src.get("运输类型") or "")
|
||
r = canonicalize_route(src.get("线路类别") or "")
|
||
c = canonicalize_land_class(src.get("运输分类") or "")
|
||
need_type = t not in TRANSPORT_TYPES
|
||
need_route = r not in ROUTES
|
||
need_class = c not in LAND_CLASSES
|
||
lines: list[str] = []
|
||
|
||
if need_type and need_route and need_class:
|
||
lines.append("请一次选择：运输类型、线路类别、运输分类")
|
||
lines.append("")
|
||
lines.append("运输类型（必填）：可选择回复，" + "、".join(TRANSPORT_TYPES))
|
||
lines.append("")
|
||
lines.append("线路类别（必填）：可选择回复，" + "、".join(ROUTES))
|
||
lines.append("")
|
||
lines.extend(_class_prompt_lines())
|
||
lines.append("")
|
||
lines.append("填写样例见图。")
|
||
return "\n".join(lines).strip()
|
||
|
||
if need_type:
|
||
lines.append("运输类型（必填）：可选择回复，" + "、".join(TRANSPORT_TYPES))
|
||
if need_route:
|
||
if lines:
|
||
lines.append("")
|
||
lines.append("线路类别（必填）：可选择回复，" + "、".join(ROUTES))
|
||
if need_class:
|
||
if lines:
|
||
lines.append("")
|
||
lines.extend(_class_prompt_lines())
|
||
if need_type or need_route:
|
||
lines.append("")
|
||
lines.append("填写样例见图。")
|
||
return "\n".join(lines).strip()
|
||
|
||
|
||
def looks_like_confirm(text: str) -> bool:
|
||
"""
|
||
判断是否在表示确定查价。
|
||
|
||
「确定」「确认」「查价」「可以查了」及相近说法都算。
|
||
同一句像在改始发/目的/体积/车型/「改成」则不当确定，避免改内容被当成查价。
|
||
"""
|
||
raw = (text or "").strip()
|
||
if not raw:
|
||
return False
|
||
if any(x in raw for x in _CONFIRM_EXCLUDE):
|
||
return False
|
||
return any(p in raw for p in _CONFIRM_PHRASES)
|