194 lines
8.1 KiB
Python
194 lines
8.1 KiB
Python
# ============================================================
|
||
# Web 端演示数据播种（moduleId: state-web-demo, 可重生 ✅）
|
||
# 背景：Web 部署（容器 + PVC 持久卷 /app/server/data）首次打开工作台时，
|
||
# 希望直接看到一套完整演示项目，而不是空白工作区。
|
||
# 口径：
|
||
# - 仅 APS_MODE=web 且 APS_WEB_DEMO_SEED=1（镜像默认开启）时生效；
|
||
# 桌面端（~/.aps 本地文件）与测试默认不受影响。
|
||
# - 按租户懒播种：某租户首次快照且 0 项目时，为其创建演示项目；
|
||
# 落 marker 文件防重复（即使用户删光演示项目也不会回填）。
|
||
# - 世界数据落在 server/data 下（world_path_for 口径），随 PVC 持久化。
|
||
# ============================================================
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import os
|
||
import re
|
||
import tempfile
|
||
import threading
|
||
from pathlib import Path
|
||
from typing import Any
|
||
|
||
from server.aps_home import data_dir, mode
|
||
from server.auth.context import IdentityContext
|
||
from server.timeutil import add_minutes, fmt_date, today0
|
||
|
||
_ENV_FLAG = "APS_WEB_DEMO_SEED"
|
||
|
||
_lock = threading.Lock()
|
||
_attempted: set[str] = set() # 进程内防重：播种过/失败过的租户不再每次快照重试
|
||
|
||
DEMO_PROJECT_NAME = "演示项目 · 智能控制器装配排产"
|
||
DEMO_PROJECT_SCOPE = "Web 演示 · 固定产线 + 柔性能力池"
|
||
|
||
|
||
def demo_seed_enabled() -> bool:
|
||
"""是否允许播种演示数据：仅 Web 形态 + 显式开关（默认关，镜像 ENV 打开）。"""
|
||
if mode() != "web":
|
||
return False
|
||
raw = (os.environ.get(_ENV_FLAG) or "").strip().lower()
|
||
return raw in ("1", "true", "yes", "on")
|
||
|
||
|
||
def _safe(value: str, fallback: str = "platform") -> str:
|
||
cleaned = re.sub(r"[^\w\-]+", "_", (value or fallback).strip())[:64]
|
||
return cleaned or fallback
|
||
|
||
|
||
def marker_path(tenant_uuid: str) -> Path:
|
||
"""播种完成标记：与租户世界数据同区，随 PVC 持久化。"""
|
||
tenant = _safe(tenant_uuid)
|
||
if tenant == "platform":
|
||
return data_dir() / "demo-seeded.ok"
|
||
return data_dir() / "tenants" / tenant / "demo-seeded.ok"
|
||
|
||
|
||
def _write_json_atomic(path: Path, payload: dict[str, Any]) -> None:
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
fd, tmp = tempfile.mkstemp(dir=str(path.parent), suffix=".tmp")
|
||
try:
|
||
with os.fdopen(fd, "w", encoding="utf-8") as handle:
|
||
json.dump(payload, handle, ensure_ascii=False, indent=1)
|
||
os.replace(tmp, path)
|
||
except BaseException:
|
||
if os.path.exists(tmp):
|
||
os.unlink(tmp)
|
||
raise
|
||
|
||
|
||
def _tenant_project_count(tenant_uuid: str) -> int:
|
||
from sqlalchemy import func, select
|
||
|
||
from server.db.database import get_session
|
||
from server.db.models import WorkspaceProject
|
||
|
||
with get_session() as session:
|
||
return int(session.execute(select(func.count(WorkspaceProject.id)).where(
|
||
WorkspaceProject.tenant_uuid == tenant_uuid,
|
||
WorkspaceProject.deleted == 0,
|
||
)).scalar_one())
|
||
|
||
|
||
def _demo_messages() -> list[dict[str, Any]]:
|
||
"""一段引导式演示对话（日期相对播种当天，保证随时播种都自洽）。"""
|
||
base = today0()
|
||
due_vip = fmt_date(add_minutes(base, 4 * 24 * 60))
|
||
due_normal = fmt_date(add_minutes(base, 6 * 24 * 60))
|
||
return [
|
||
{
|
||
"role": "assistant",
|
||
"text": (
|
||
"欢迎使用 APS 智能排产工作台（Web 演示环境）。\n"
|
||
"本项目已内置一套完整演示数据：\n"
|
||
"· 固定产线层：2 车间 / 3 条装配线 / 13 个工位 / 5 道工序，3 款控制器产品与 BOM、工艺路线\n"
|
||
"· 柔性能力池：4 类共线产品（高压线束 / PDU / 充电枪 / 高压连接器），7 个能力池、4 台可跨区调配压接机\n"
|
||
"· 业务单据：7 张销售订单（含 1 张 VIP 急单）、10 张柔性订单、3 张预测订单、未来 30 天班次日历\n"
|
||
"· 冲突场景：后天上午 ASSY-01 设备季度保养\n\n"
|
||
"可以直接在下方对话框下达自然语言指令开始演示。"
|
||
),
|
||
},
|
||
{"role": "user", "text": "现在有哪些订单？哪些最紧急？"},
|
||
{
|
||
"role": "assistant",
|
||
"text": (
|
||
f"当前共 7 张销售订单。最紧急的是 VIP 急单（比亚迪，{due_vip} 交期，已标记插单），"
|
||
f"其次是海尔智家 VIP 订单（{due_normal} 交期）。\n"
|
||
"可以继续追问：「一键排产」「检查订单齐套性」「MRP 分解」「查看产能负载」。"
|
||
),
|
||
},
|
||
{"role": "user", "text": "一键排产"},
|
||
{
|
||
"role": "assistant",
|
||
"text": (
|
||
"排产前会先校验主数据与齐套性，并按冻结窗口（24h）保护已下达工单。\n"
|
||
"正式环境中该操作会生成 P2 确认卡，批准后才写入排产方案；"
|
||
"演示环境可直接确认，随后在「甘特 / 负载 / 方案对比」视图查看结果。"
|
||
),
|
||
},
|
||
]
|
||
|
||
|
||
def seed_demo_workspace(identity: IdentityContext, *, force: bool = False) -> bool:
|
||
"""为指定租户播种一套演示项目（幂等）。返回是否真正执行了播种。"""
|
||
tenant = _safe(identity.tenant_uuid)
|
||
marker = marker_path(tenant)
|
||
if marker.exists() and not force:
|
||
return False
|
||
if not force and _tenant_project_count(tenant) > 0:
|
||
# 已有真实项目：不播种，但落标记避免每次快照都查库
|
||
marker.parent.mkdir(parents=True, exist_ok=True)
|
||
marker.write_text("skipped: tenant already has projects\n", encoding="utf-8")
|
||
return False
|
||
|
||
from server.state.projects import ProjectStore
|
||
from server.state.seed import build_demo_world
|
||
from server.state.store import _safe_scope, _stores, world_path_for
|
||
|
||
store = ProjectStore(identity)
|
||
project_id = _safe(f"demo-{tenant}")[:64]
|
||
created = store.create_project(
|
||
DEMO_PROJECT_NAME,
|
||
scope_label=DEMO_PROJECT_SCOPE,
|
||
project_id=project_id,
|
||
)
|
||
project = created["project"]
|
||
session_id = created["session"]["id"]
|
||
|
||
# 世界数据：完整演示工厂（固定产线 + 柔性能力池 + 订单 + 日历）
|
||
world = build_demo_world()
|
||
world_path = Path(world_path_for(project_id, tenant))
|
||
_write_json_atomic(world_path, world)
|
||
# 若该 worldKey 已被进程内缓存（例如播种前被探测加载过），丢弃缓存保证读的是演示世界
|
||
_stores.pop((identity.tenant_uuid, _safe_scope(project_id, "default")), None)
|
||
|
||
# 引导式演示对话 + 示例文件记录
|
||
for message in _demo_messages():
|
||
store.append_message(session_id, message)
|
||
store.create_file(
|
||
project_id,
|
||
"锐扬APS演示数据（示例导入包）",
|
||
kind="import",
|
||
note="演示 CSV 包位于仓库 demo-data/锐扬APS演示数据，可在「导入」栏体验解析入库",
|
||
)
|
||
store.create_file(
|
||
project_id,
|
||
"排产方案报告（排产后生成）",
|
||
kind="schedule",
|
||
note="执行「一键排产」后可导出 Excel 工作计划表（工作计划 / 方案概览 / 冲突）",
|
||
)
|
||
|
||
marker.parent.mkdir(parents=True, exist_ok=True)
|
||
marker.write_text(f"seeded project={project_id}\n", encoding="utf-8")
|
||
return True
|
||
|
||
|
||
def maybe_seed_demo_workspace(identity: IdentityContext) -> bool:
|
||
"""快照入口的懒播种钩子：条件不满足时零成本返回 False。"""
|
||
if not demo_seed_enabled():
|
||
return False
|
||
if not identity or not identity.user_id:
|
||
return False # 系统/维护身份不播种，避免污染 platform 租户
|
||
tenant = _safe(identity.tenant_uuid)
|
||
with _lock:
|
||
if tenant in _attempted:
|
||
return False
|
||
_attempted.add(tenant)
|
||
try:
|
||
return seed_demo_workspace(identity)
|
||
except Exception:
|
||
# 播种失败（如数据目录暂不可写）不阻断主流程；本进程内不再重试，重启后可再试
|
||
import logging
|
||
|
||
logging.getLogger(__name__).warning("web demo seed failed for tenant %s", tenant, exc_info=True)
|
||
return False
|