228 lines
8.4 KiB
Python
228 lines
8.4 KiB
Python
|
# =============================================================================
|
|||
|
|
# IT智能服务台 — 审批卡片匹配引擎
|
|||
|
|
# =============================================================================
|
|||
|
|
# 职责：统一审批匹配逻辑，替代前端 APPROVAL_OPTIONS 和后端分散的匹配代码
|
|||
|
|
#
|
|||
|
|
# 匹配优先级：
|
|||
|
|
# 1. approval_type 精确匹配模板 ID（如 "zero_trust_vpn"）
|
|||
|
|
# 2. approval_type 关键词匹配模板 keywords（如 "VPN账号申请" 含 "VPN"）
|
|||
|
|
# 3. approval_type 匹配 category（如 "账号权限申请" → 返回该分类下所有模板）
|
|||
|
|
# 4. title 匹配模板 name（如 "VPN账号申请" → zero_trust_vpn）
|
|||
|
|
# 5. 降级：用户原始文本遍历所有 keywords
|
|||
|
|
#
|
|||
|
|
# 输出：标准化卡片数据 card_data，前端纯渲染
|
|||
|
|
# =============================================================================
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from typing import Optional
|
|||
|
|
from app.api.approval import APPROVAL_TEMPLATES
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ApprovalMatcher:
|
|||
|
|
"""审批卡片匹配器 — 后端统一匹配入口"""
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 公开方法
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
def match_and_build_card(
|
|||
|
|
self, approval_type: Optional[str], title: Optional[str] = None
|
|||
|
|
) -> Optional[dict]:
|
|||
|
|
"""一站式匹配 + 构建标准化卡片数据（Dify 正常路径）。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
approval_type: Dify 返回的审批类型（可能是 ID、中文分类名或具体名称）
|
|||
|
|
title: Dify 返回的具体审批名称（如"VPN账号申请"）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
标准化 card_data，或 None（匹配失败）
|
|||
|
|
"""
|
|||
|
|
if not approval_type:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 优先级 1：精确 ID 匹配
|
|||
|
|
template = self._match_by_id(approval_type)
|
|||
|
|
if template:
|
|||
|
|
return self._build_single_card(template)
|
|||
|
|
|
|||
|
|
# 优先级 2：关键词匹配
|
|||
|
|
template = self._match_by_keyword(approval_type)
|
|||
|
|
if template:
|
|||
|
|
return self._build_single_card(template)
|
|||
|
|
|
|||
|
|
# 优先级 3：category 匹配
|
|||
|
|
options = self._match_by_category(approval_type)
|
|||
|
|
if options:
|
|||
|
|
return self._build_multi_card(approval_type, options)
|
|||
|
|
|
|||
|
|
# 优先级 4：title 匹配
|
|||
|
|
if title:
|
|||
|
|
template = self._match_by_name(title)
|
|||
|
|
if template:
|
|||
|
|
return self._build_single_card(template)
|
|||
|
|
|
|||
|
|
# 优先级 5：文本 keywords 兜底
|
|||
|
|
template = self._match_by_keyword(approval_type)
|
|||
|
|
if template:
|
|||
|
|
return self._build_single_card(template)
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def match_by_keywords(self, user_text: str) -> Optional[dict]:
|
|||
|
|
"""Dify 不可用时，用用户原始文本关键词降级匹配。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
user_text: 用户输入的原始文本
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
标准化 card_data，或 None
|
|||
|
|
"""
|
|||
|
|
if not user_text:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
text_lower = user_text.lower()
|
|||
|
|
|
|||
|
|
# 遍历所有模板的 keywords
|
|||
|
|
best_template = None
|
|||
|
|
best_score = 0
|
|||
|
|
|
|||
|
|
for template_id, template in APPROVAL_TEMPLATES.items():
|
|||
|
|
keywords = template.get("keywords", [])
|
|||
|
|
score = 0
|
|||
|
|
for kw in keywords:
|
|||
|
|
if kw.lower() in text_lower:
|
|||
|
|
score += len(kw) # 关键词越长，匹配越准确
|
|||
|
|
if score > best_score:
|
|||
|
|
best_score = score
|
|||
|
|
best_template = template
|
|||
|
|
|
|||
|
|
if best_template and best_score > 0:
|
|||
|
|
logger.info(
|
|||
|
|
f"[ApprovalMatcher] 关键词降级匹配: text={user_text[:30]} -> "
|
|||
|
|
f"template={best_template['id']}, score={best_score}"
|
|||
|
|
)
|
|||
|
|
return self._build_single_card(best_template)
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 内部匹配方法
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
def _match_by_id(self, approval_type: str) -> Optional[dict]:
|
|||
|
|
"""精确匹配模板 ID（如 "zero_trust_vpn"）。"""
|
|||
|
|
return APPROVAL_TEMPLATES.get(approval_type)
|
|||
|
|
|
|||
|
|
def _match_by_keyword(self, approval_type: str) -> Optional[dict]:
|
|||
|
|
"""通过关键词匹配模板。
|
|||
|
|
|
|||
|
|
遍历所有模板的 keywords，检查 approval_type 是否包含任一关键词。
|
|||
|
|
使用最长匹配优先策略（避免"设备升级"误匹配到"升级"）。
|
|||
|
|
"""
|
|||
|
|
approval_lower = approval_type.lower()
|
|||
|
|
best_template = None
|
|||
|
|
best_len = 0
|
|||
|
|
|
|||
|
|
for template_id, template in APPROVAL_TEMPLATES.items():
|
|||
|
|
keywords = template.get("keywords", [])
|
|||
|
|
for kw in keywords:
|
|||
|
|
if kw.lower() in approval_lower:
|
|||
|
|
if len(kw) > best_len:
|
|||
|
|
best_len = len(kw)
|
|||
|
|
best_template = template
|
|||
|
|
|
|||
|
|
return best_template
|
|||
|
|
|
|||
|
|
def _match_by_category(self, approval_type: str) -> list[dict]:
|
|||
|
|
"""按 category 匹配，返回该分类下所有模板的卡片选项。"""
|
|||
|
|
result = []
|
|||
|
|
for template_id, template in APPROVAL_TEMPLATES.items():
|
|||
|
|
category = template.get("category", "")
|
|||
|
|
if category == approval_type:
|
|||
|
|
result.append(self._template_to_option(template))
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
def _match_by_name(self, title: str) -> Optional[dict]:
|
|||
|
|
"""通过模板 name 精确/模糊匹配（空格容错）。"""
|
|||
|
|
normalized = title.replace(" ", "") # 去除空格容错
|
|||
|
|
|
|||
|
|
for template_id, template in APPROVAL_TEMPLATES.items():
|
|||
|
|
name_clean = template["name"].replace(" ", "")
|
|||
|
|
if name_clean == normalized or normalized in name_clean:
|
|||
|
|
return template
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 卡片数据构建
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
def _build_single_card(self, template: dict) -> dict:
|
|||
|
|
"""构建单选项标准化卡片数据。"""
|
|||
|
|
return {
|
|||
|
|
"card_type": "single",
|
|||
|
|
"title": template["name"],
|
|||
|
|
"description": template.get("desc", ""),
|
|||
|
|
"options": [self._template_to_option(template)],
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def _build_multi_card(self, category_name: str, options: list[dict]) -> dict:
|
|||
|
|
"""构建多选项标准化卡片数据。"""
|
|||
|
|
return {
|
|||
|
|
"card_type": "multiple",
|
|||
|
|
"title": f"{category_name}（{len(options)}项）",
|
|||
|
|
"description": "请选择具体审批类型",
|
|||
|
|
"options": options,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def _template_to_option(self, template: dict) -> dict:
|
|||
|
|
"""将模板数据转换为前端卡片选项。"""
|
|||
|
|
return {
|
|||
|
|
"name": template["name"],
|
|||
|
|
"icon": template.get("icon", "orders-o"),
|
|||
|
|
"desc": template.get("desc", ""),
|
|||
|
|
"url": template.get("url", ""),
|
|||
|
|
"category": template.get("category", ""),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def get_all_categories(self) -> list[dict]:
|
|||
|
|
"""获取所有分类及其选项（供前端全部展示用）。"""
|
|||
|
|
categories = {}
|
|||
|
|
for template_id, template in APPROVAL_TEMPLATES.items():
|
|||
|
|
cat = template.get("category", "其他")
|
|||
|
|
if cat not in categories:
|
|||
|
|
categories[cat] = []
|
|||
|
|
categories[cat].append(self._template_to_option(template))
|
|||
|
|
|
|||
|
|
result = []
|
|||
|
|
for cat, options in categories.items():
|
|||
|
|
result.append({
|
|||
|
|
"category": cat,
|
|||
|
|
"options": options,
|
|||
|
|
})
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
def get_all_templates(self) -> list[dict]:
|
|||
|
|
"""获取所有模板（含新增字段）。"""
|
|||
|
|
return [
|
|||
|
|
{**template, "template_id": tid}
|
|||
|
|
for tid, template in APPROVAL_TEMPLATES.items()
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 单例工厂
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
_approval_matcher: Optional[ApprovalMatcher] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_approval_matcher() -> ApprovalMatcher:
|
|||
|
|
"""获取 ApprovalMatcher 单例。"""
|
|||
|
|
global _approval_matcher
|
|||
|
|
if _approval_matcher is None:
|
|||
|
|
_approval_matcher = ApprovalMatcher()
|
|||
|
|
return _approval_matcher
|