feat: resolve shortage names from Git-synced Kingdee material cache
This commit is contained in:
64
shared/runtime/material_name_cache.py
Normal file
64
shared/runtime/material_name_cache.py
Normal file
@@ -0,0 +1,64 @@
|
||||
"""Portable, read-only material-code/name lookup for delivery replies."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import hashlib
|
||||
import json
|
||||
from functools import lru_cache
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
|
||||
MATERIAL_NAME_CACHE_FILE = (
|
||||
Path(__file__).resolve().parents[2]
|
||||
/ "capabilities/am516-delivery-prediction/data/material_names.json"
|
||||
)
|
||||
|
||||
|
||||
class MaterialNameCacheError(ValueError):
|
||||
"""A cache error that contains no credentials or raw ERP records."""
|
||||
|
||||
|
||||
def material_names_hash(materials: dict[str, str]) -> str:
|
||||
content = json.dumps(materials, ensure_ascii=False, sort_keys=True, separators=(",", ":"))
|
||||
return hashlib.sha256(content.encode("utf-8")).hexdigest()
|
||||
|
||||
|
||||
def material_names_from_records(records: list[dict[str, Any]]) -> dict[str, str]:
|
||||
if not records:
|
||||
raise MaterialNameCacheError("物料查询结果为空,保留原缓存。")
|
||||
materials: dict[str, str] = {}
|
||||
for row in records:
|
||||
code, name = row.get("FNumber"), row.get("FName")
|
||||
if not isinstance(code, str) or not code.strip() or not isinstance(name, str) or not name.strip():
|
||||
raise MaterialNameCacheError("物料记录缺少有效编码或名称,保留原缓存。")
|
||||
code, name = code.strip(), " ".join(name.split())
|
||||
if code in materials and materials[code] != name:
|
||||
raise MaterialNameCacheError("同一物料编码存在不同名称,需核对后再刷新缓存。")
|
||||
materials[code] = name
|
||||
return dict(sorted(materials.items()))
|
||||
|
||||
|
||||
@lru_cache(maxsize=4)
|
||||
def load_material_names(path: Path = MATERIAL_NAME_CACHE_FILE) -> dict[str, str]:
|
||||
"""Load once per process; a Git update takes effect after the bot restarts."""
|
||||
try:
|
||||
payload = json.loads(path.read_text(encoding="utf-8"))
|
||||
except FileNotFoundError as exc:
|
||||
raise MaterialNameCacheError("本地物料名称缓存文件缺失") from exc
|
||||
except (OSError, UnicodeError, json.JSONDecodeError) as exc:
|
||||
raise MaterialNameCacheError("本地物料名称缓存文件不可读取或格式错误") from exc
|
||||
if not isinstance(payload, dict) or payload.get("schema_version") != 1:
|
||||
raise MaterialNameCacheError("本地物料名称缓存版本或结构无效")
|
||||
materials = payload.get("materials")
|
||||
if not isinstance(materials, dict) or not materials:
|
||||
raise MaterialNameCacheError("本地物料名称缓存为空或结构无效")
|
||||
if any(
|
||||
not isinstance(code, str) or not code.strip()
|
||||
or not isinstance(name, str) or not name.strip()
|
||||
for code, name in materials.items()
|
||||
):
|
||||
raise MaterialNameCacheError("本地物料名称缓存包含无效编码或名称")
|
||||
if payload.get("material_count") != len(materials) or payload.get("content_sha256") != material_names_hash(materials):
|
||||
raise MaterialNameCacheError("本地物料名称缓存计数或完整性校验失败")
|
||||
return materials
|
||||
Reference in New Issue
Block a user