315 lines
14 KiB
Python
315 lines
14 KiB
Python
# ============================================================
|
||
# 项目深度分析（moduleId: domain-project-analyze, 可重生 ✅）
|
||
# 工程目录 Excel/SQL → 写入当前项目世界 → 明细表 + 齐备度 + 知识库补洞 plan
|
||
# ============================================================
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
import re
|
||
from typing import Any
|
||
|
||
from server.aps_domain.readiness import check_readiness
|
||
|
||
World = dict[str, Any]
|
||
|
||
|
||
def _project_ctx(session_id: str | None) -> tuple[dict | None, str]:
|
||
try:
|
||
from server.state.projects import get_project_store
|
||
snap = get_project_store().snapshot(include_messages=False)
|
||
except Exception:
|
||
return None, ""
|
||
sessions = {s["id"]: s for s in (snap.get("sessions") or []) if isinstance(s, dict)}
|
||
projects = {p["id"]: p for p in (snap.get("projects") or []) if isinstance(p, dict)}
|
||
sess = sessions.get(session_id or "") or {}
|
||
proj = projects.get(sess.get("projectId") or "") if sess.get("projectId") else None
|
||
work = (proj.get("workDir") or "").strip() if proj else ""
|
||
return proj, work
|
||
|
||
|
||
def _fill_from_knowledge(world: World, product_codes: list[str]) -> list[dict[str, Any]]:
|
||
"""缺工艺时用行业模板/知识库自动补；返回动作清单。"""
|
||
actions: list[dict[str, Any]] = []
|
||
try:
|
||
from server.knowledge.routing_templates import recommend_templates, apply_template_to_product
|
||
except Exception:
|
||
return actions
|
||
routes = {r.get("productCode") for r in (world.get("flexRoutings") or [])}
|
||
mats = {m.get("code") for m in (world.get("flexMaterials") or [])}
|
||
for pc in product_codes:
|
||
if not pc:
|
||
continue
|
||
if pc not in mats:
|
||
world.setdefault("flexMaterials", []).append({
|
||
"id": len(world.get("flexMaterials") or []) + 1,
|
||
"code": pc, "name": pc, "type": "FINISHED_PRODUCT", "unit": "件",
|
||
"stock": 0, "inTransit": 0, "safetyStock": 0, "procurementLeadTime": 0,
|
||
})
|
||
actions.append({"type": "material_stub", "productCode": pc,
|
||
"detail": f"物料主数据缺失，已建产品档案占位 {pc}"})
|
||
if pc in routes:
|
||
continue
|
||
name = next((m.get("name") for m in (world.get("flexMaterials") or [])
|
||
if m.get("code") == pc), pc)
|
||
try:
|
||
recs = recommend_templates(f"{pc} {name}") or []
|
||
except Exception:
|
||
recs = []
|
||
if not recs:
|
||
actions.append({"type": "need_routing", "productCode": pc,
|
||
"detail": f"产品 {pc} 无工艺，知识库也无匹配模板，需客户提供工艺路线"})
|
||
continue
|
||
tpl = recs[0]
|
||
code = tpl.get("code") or tpl.get("templateCode")
|
||
try:
|
||
apply_template_to_product(world, code, pc)
|
||
actions.append({
|
||
"type": "template_applied", "productCode": pc,
|
||
"template": code, "templateName": tpl.get("name"),
|
||
"detail": f"产品 {pc} 缺工艺 → 已用知识库模板「{tpl.get('name') or code}」生成",
|
||
})
|
||
routes.add(pc)
|
||
except Exception as exc:
|
||
actions.append({"type": "template_failed", "productCode": pc,
|
||
"detail": f"模板 {code} 应用失败：{exc}"})
|
||
return actions
|
||
|
||
|
||
def analyze_project_deep(
|
||
world: World,
|
||
session_id: str | None = None,
|
||
*,
|
||
apply_sql: bool = True,
|
||
query: str = "",
|
||
) -> dict[str, Any]:
|
||
"""深度分析当前项目：目录包（SQL/Excel）+ 世界主数据 + 知识库补洞。"""
|
||
try:
|
||
from server.agent_core.progress import emit_thinking
|
||
except Exception:
|
||
def emit_thinking(*_a, **_k): # type: ignore
|
||
return None
|
||
|
||
proj, work = _project_ctx(session_id)
|
||
sources: list[str] = []
|
||
sql_preview: dict[str, Any] | None = None
|
||
applied: dict[str, int] = {}
|
||
kb_actions: list[dict[str, Any]] = []
|
||
|
||
emit_thinking("定位工程目录", (proj or {}).get("name") or "当前项目", pct=2)
|
||
if work and os.path.isdir(work):
|
||
names = sorted(os.listdir(work))
|
||
sqls = [n for n in names if n.lower().endswith(".sql") and not n.startswith("~$")]
|
||
xls = [n for n in names if re.search(r"\.(xlsx|xlsm|csv)$", n, re.I) and not n.startswith("~$")]
|
||
emit_thinking(
|
||
"清点目录文件",
|
||
f"SQL {len(sqls)} · Excel/CSV {len(xls)}",
|
||
pct=5,
|
||
)
|
||
if sqls:
|
||
from server.importers.sql_pack import preview_sql_file, apply_sql_pack_to_world
|
||
# 优先最大的 sql（完整库）
|
||
sqls_sorted = sorted(sqls, key=lambda n: os.path.getsize(os.path.join(work, n)), reverse=True)
|
||
path = os.path.join(work, sqls_sorted[0])
|
||
sources.append(f"SQL:{sqls_sorted[0]}")
|
||
try:
|
||
sql_preview = preview_sql_file(path)
|
||
if apply_sql and sql_preview.get("flex"):
|
||
emit_thinking("写入项目世界", "订单 / 物料 / 工艺 / 设备", pct=93)
|
||
# SQL 包是工程目录真相源，始终整包覆盖，避免旧错误工艺残留
|
||
applied = apply_sql_pack_to_world(
|
||
world, sql_preview["flex"], replace=True)
|
||
emit_thinking(
|
||
"入库完成",
|
||
"、".join(f"{k}{v}" for k, v in applied.items()) or "无新增",
|
||
status="done", pct=95,
|
||
)
|
||
except Exception as exc:
|
||
sql_preview = {"error": str(exc), "filename": sqls_sorted[0]}
|
||
emit_thinking("SQL 解析失败", str(exc), status="warn")
|
||
if xls:
|
||
sources.append(f"Excel×{len(xls)}")
|
||
try:
|
||
from server.aps_domain.folder_pack import analyze_work_dir
|
||
emit_thinking("核对 Excel/CSV", f"{len(xls)} 个表格", pct=96)
|
||
# 仅作文件清单；主数据以 SQL/世界为准
|
||
folder = analyze_work_dir(world, session_id)
|
||
except Exception:
|
||
folder = None
|
||
else:
|
||
folder = None
|
||
else:
|
||
folder = None
|
||
emit_thinking("未挂工程目录", "仅分析当前项目世界里的数据", status="warn", pct=10)
|
||
|
||
# 知识库补洞：缺工艺的产品
|
||
emit_thinking("检查齐备度与知识库", "缺工艺时自动匹配行业模板", pct=97)
|
||
active = [o for o in (world.get("flexOrders") or [])
|
||
if (o.get("status") or "") not in ("DONE", "CANCELLED")]
|
||
pcs = list({str(o.get("productCode") or "") for o in active})
|
||
kb_actions = _fill_from_knowledge(world, pcs)
|
||
if kb_actions:
|
||
emit_thinking(
|
||
"知识库补洞",
|
||
f"{len(kb_actions)} 项（模板/占位/待客户补）",
|
||
status="done", pct=98,
|
||
)
|
||
|
||
ready = check_readiness(world)
|
||
emit_thinking("整理分析报告", f"订单 {len(active)} 张", status="done", pct=99)
|
||
mats = world.get("flexMaterials") or []
|
||
routes = world.get("flexRoutings") or []
|
||
equip = [e for e in (world.get("flexEquipment") or []) if e.get("status") == "RUNNING"]
|
||
bom = world.get("flexBom") or []
|
||
|
||
order_rows = [{
|
||
"orderNo": o.get("orderNo"),
|
||
"productCode": o.get("productCode"),
|
||
"productName": o.get("productName") or next(
|
||
(m.get("name") for m in mats if m.get("code") == o.get("productCode")), ""),
|
||
"quantity": o.get("quantity"),
|
||
"dueDate": o.get("dueDate"),
|
||
"customerName": o.get("customerName") or "",
|
||
"ready": next((r["ready"] for r in ready["orders"] if r["orderNo"] == o.get("orderNo")), False),
|
||
} for o in active[:80]]
|
||
|
||
# 主数据抽样：订单涉及的产品优先
|
||
need_codes = {r["productCode"] for r in order_rows if r.get("productCode")}
|
||
mat_rows = []
|
||
for m in mats:
|
||
if need_codes and m.get("code") not in need_codes and len(mat_rows) >= 40:
|
||
continue
|
||
if m.get("code") in need_codes or len(mat_rows) < 40:
|
||
mat_rows.append({
|
||
"code": m.get("code"), "name": m.get("name"),
|
||
"type": m.get("type"), "unit": m.get("unit"), "stock": m.get("stock"),
|
||
})
|
||
if len(mat_rows) >= 60:
|
||
break
|
||
|
||
route_rows = []
|
||
for r in routes:
|
||
if need_codes and r.get("productCode") not in need_codes and len(route_rows) >= 40:
|
||
continue
|
||
route_rows.append({
|
||
"productCode": r.get("productCode"),
|
||
"seq": r.get("seq"),
|
||
"operationCode": r.get("operationCode"),
|
||
"operationName": r.get("operationName") or r.get("operationCode"),
|
||
"stdMin": r.get("stdTimePerUnit"),
|
||
})
|
||
if len(route_rows) >= 80:
|
||
break
|
||
|
||
equip_rows = [{
|
||
"code": e.get("code"), "name": e.get("name"),
|
||
"capabilities": "、".join((e.get("capabilities") or [])[:6]),
|
||
} for e in equip[:40]]
|
||
|
||
# 缺口 plan：先库后客户（同类缺口合并，避免刷屏）
|
||
plan: list[str] = []
|
||
need_rt = [a for a in kb_actions if a.get("type") == "need_routing"]
|
||
applied_tpl = [a for a in kb_actions if a.get("type") == "template_applied"]
|
||
other_kb = [a for a in kb_actions if a.get("type") not in ("need_routing", "template_applied")]
|
||
if applied_tpl:
|
||
plan.append(f"✓ 已用知识库模板补工艺 {len(applied_tpl)} 个产品")
|
||
for a in applied_tpl[:5]:
|
||
plan.append(f" · {a.get('productCode')} ← {a.get('templateName') or a.get('template')}")
|
||
if len(applied_tpl) > 5:
|
||
plan.append(f" · …另有 {len(applied_tpl) - 5} 个")
|
||
if need_rt:
|
||
samples = "、".join(str(a.get("productCode") or "") for a in need_rt[:8])
|
||
more = f" 等共 {len(need_rt)} 个" if len(need_rt) > 8 else f"（{len(need_rt)} 个）"
|
||
plan.append(
|
||
f"△ SQL/知识库仍未挂上工艺的产品：{samples}{more}。"
|
||
f"若现场库本有工艺，请重新「分析项目」（工艺按订单 craftl 关联，不是中间件料号）"
|
||
)
|
||
for a in other_kb[:6]:
|
||
plan.append(f"· {a['detail']}")
|
||
s = ready["summary"]
|
||
if s.get("blocked"):
|
||
plan.append(f"△ 仍有 {s['blocked']} 张单齐备度未过，见订单明细「是否可排」列")
|
||
if not active:
|
||
plan.append("△ 没有订单：请在工程目录放订单表/SQL，或说「新建订单」")
|
||
if not equip:
|
||
plan.append("△ 没有可用设备：请提供设备台账，或从 SQL 的 md_equipment 导入")
|
||
if not plan:
|
||
plan.append("✓ 主数据与知识库已齐，可直接说「给我排产」")
|
||
|
||
proj_name = (proj or {}).get("name") or "当前项目"
|
||
md_lines = [
|
||
f"**项目深度分析 · {proj_name}**",
|
||
"",
|
||
f"数据来源：{'、'.join(sources) if sources else '当前项目世界（未挂工程目录文件）'}",
|
||
]
|
||
if applied:
|
||
md_lines.append(
|
||
"已从 SQL 入库："
|
||
+ "、".join(f"{k} {v}" for k, v in applied.items())
|
||
)
|
||
if sql_preview and sql_preview.get("fieldMapText"):
|
||
md_lines.append(f"SQL 包识别：{sql_preview['fieldMapText']}")
|
||
md_lines += [
|
||
"",
|
||
f"订单 **{len(active)}** · 物料 **{len(mats)}** · 工艺步 **{len(routes)}** · "
|
||
f"设备 **{len(equip)}** · BOM **{len(bom)}**",
|
||
"",
|
||
"### 订单明细（本项目）",
|
||
"",
|
||
"| 订单号 | 产品 | 名称 | 数量 | 交期 | 客户 | 可排 |",
|
||
"| --- | --- | --- | ---: | --- | --- | --- |",
|
||
]
|
||
for r in order_rows[:30]:
|
||
md_lines.append(
|
||
f"| {r['orderNo']} | {r['productCode']} | {str(r.get('productName') or '')[:16]} "
|
||
f"| {r['quantity']} | {r['dueDate'] or '—'} | {str(r.get('customerName') or '')[:12]} "
|
||
f"| {'✓' if r['ready'] else '×'} |"
|
||
)
|
||
if len(order_rows) > 30:
|
||
md_lines.append(f"| … | 另有 {len(order_rows) - 30} 张 | | | | | |")
|
||
|
||
md_lines += [
|
||
"",
|
||
"### 主数据（产品/物料抽样）",
|
||
"",
|
||
"| 编码 | 名称 | 类型 | 单位 | 库存 |",
|
||
"| --- | --- | --- | --- | ---: |",
|
||
]
|
||
for m in mat_rows[:25]:
|
||
md_lines.append(
|
||
f"| {m['code']} | {str(m.get('name') or '')[:20]} | {m.get('type')} "
|
||
f"| {m.get('unit')} | {m.get('stock') or 0} |"
|
||
)
|
||
|
||
md_lines += ["", "### 补数 Plan（先知识库/数据库，再找客户）", ""]
|
||
for i, p in enumerate(plan[:12], 1):
|
||
md_lines.append(f"{i}. {p}")
|
||
|
||
return {
|
||
"ok": True,
|
||
"projectName": proj_name,
|
||
"workDir": work,
|
||
"sources": sources,
|
||
"applied": applied,
|
||
"sqlPreview": {
|
||
"filename": (sql_preview or {}).get("filename"),
|
||
"fieldMapText": (sql_preview or {}).get("fieldMapText"),
|
||
"tables": (sql_preview or {}).get("tables") or [],
|
||
"error": (sql_preview or {}).get("error"),
|
||
} if sql_preview else None,
|
||
"folder": {"files": (folder or {}).get("files") or []} if folder else None,
|
||
"orders": order_rows,
|
||
"materials": mat_rows,
|
||
"routings": route_rows,
|
||
"equipment": equip_rows,
|
||
"kbActions": kb_actions,
|
||
"plan": plan,
|
||
"readiness": ready,
|
||
"summary": {
|
||
"orders": len(active), "materials": len(mats),
|
||
"routings": len(routes), "equipment": len(equip), "bom": len(bom),
|
||
"ready": s.get("ready", 0), "blocked": s.get("blocked", 0),
|
||
},
|
||
"canSchedule": s.get("ready", 0) > 0 and not ready.get("globalIssues"),
|
||
"markdown": "\n".join(md_lines),
|
||
}
|