mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2025-12-17 13:01:12 +00:00
139 lines
4.7 KiB
Python
139 lines
4.7 KiB
Python
"""
|
||
工作流工具模块
|
||
|
||
提供工作流上传等功能
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import uuid
|
||
from typing import Any, Dict, List, Optional
|
||
|
||
from unilabos.utils.banner_print import print_status
|
||
|
||
|
||
def _is_node_link_format(data: Dict[str, Any]) -> bool:
|
||
"""检查数据是否为 node-link 格式"""
|
||
return "nodes" in data and "edges" in data
|
||
|
||
|
||
def _convert_to_node_link(workflow_file: str, workflow_data: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""
|
||
将非 node-link 格式的工作流数据转换为 node-link 格式
|
||
|
||
Args:
|
||
workflow_file: 工作流文件路径（用于日志）
|
||
workflow_data: 原始工作流数据
|
||
|
||
Returns:
|
||
node-link 格式的工作流数据
|
||
"""
|
||
from unilabos.workflow.convert_from_json import convert_json_to_node_link
|
||
|
||
print_status(f"检测到非 node-link 格式，正在转换...", "info")
|
||
node_link_data = convert_json_to_node_link(workflow_data)
|
||
print_status(f"转换完成", "success")
|
||
return node_link_data
|
||
|
||
|
||
def upload_workflow(
|
||
workflow_file: str,
|
||
workflow_name: Optional[str] = None,
|
||
tags: Optional[List[str]] = None,
|
||
published: bool = False,
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
上传工作流到服务器
|
||
|
||
支持的输入格式：
|
||
1. node-link 格式: {"nodes": [...], "edges": [...]}
|
||
2. workflow/reagent 格式: {"workflow": [...], "reagent": {...}}
|
||
3. steps_info/labware_info 格式: {"steps_info": [...], "labware_info": [...]}
|
||
4. steps/labware 格式: {"steps": [...], "labware": [...]}
|
||
|
||
Args:
|
||
workflow_file: 工作流文件路径（JSON格式）
|
||
workflow_name: 工作流名称，如果不提供则从文件中读取或使用文件名
|
||
tags: 工作流标签列表，默认为空列表
|
||
published: 是否发布工作流，默认为False
|
||
|
||
Returns:
|
||
Dict: API响应数据
|
||
"""
|
||
# 延迟导入，避免在配置文件加载之前初始化 http_client
|
||
from unilabos.app.web import http_client
|
||
|
||
if not os.path.exists(workflow_file):
|
||
print_status(f"工作流文件不存在: {workflow_file}", "error")
|
||
return {"code": -1, "message": f"文件不存在: {workflow_file}"}
|
||
|
||
# 读取工作流文件
|
||
try:
|
||
with open(workflow_file, "r", encoding="utf-8") as f:
|
||
workflow_data = json.load(f)
|
||
except json.JSONDecodeError as e:
|
||
print_status(f"工作流文件JSON解析失败: {e}", "error")
|
||
return {"code": -1, "message": f"JSON解析失败: {e}"}
|
||
|
||
# 自动检测并转换格式
|
||
if not _is_node_link_format(workflow_data):
|
||
try:
|
||
workflow_data = _convert_to_node_link(workflow_file, workflow_data)
|
||
except Exception as e:
|
||
print_status(f"工作流格式转换失败: {e}", "error")
|
||
return {"code": -1, "message": f"格式转换失败: {e}"}
|
||
|
||
# 提取工作流数据
|
||
nodes = workflow_data.get("nodes", [])
|
||
edges = workflow_data.get("edges", [])
|
||
workflow_uuid_val = workflow_data.get("workflow_uuid", str(uuid.uuid4()))
|
||
wf_name_from_file = workflow_data.get("workflow_name", os.path.basename(workflow_file).replace(".json", ""))
|
||
|
||
# 确定工作流名称
|
||
final_name = workflow_name or wf_name_from_file
|
||
|
||
print_status(f"正在上传工作流: {final_name}", "info")
|
||
print_status(f" - 节点数量: {len(nodes)}", "info")
|
||
print_status(f" - 边数量: {len(edges)}", "info")
|
||
print_status(f" - 标签: {tags or []}", "info")
|
||
print_status(f" - 发布状态: {published}", "info")
|
||
|
||
# 调用 http_client 上传
|
||
result = http_client.workflow_import(
|
||
name=final_name,
|
||
workflow_uuid=workflow_uuid_val,
|
||
workflow_name=final_name,
|
||
nodes=nodes,
|
||
edges=edges,
|
||
tags=tags,
|
||
published=published,
|
||
)
|
||
|
||
if result.get("code") == 0:
|
||
data = result.get("data", {})
|
||
print_status("工作流上传成功！", "success")
|
||
print_status(f" - UUID: {data.get('uuid', 'N/A')}", "info")
|
||
print_status(f" - 名称: {data.get('name', 'N/A')}", "info")
|
||
else:
|
||
print_status(f"工作流上传失败: {result.get('message', '未知错误')}", "error")
|
||
|
||
return result
|
||
|
||
|
||
def handle_workflow_upload_command(args_dict: Dict[str, Any]) -> None:
|
||
"""
|
||
处理 workflow_upload 子命令
|
||
|
||
Args:
|
||
args_dict: 命令行参数字典
|
||
"""
|
||
workflow_file = args_dict.get("workflow_file")
|
||
workflow_name = args_dict.get("workflow_name")
|
||
tags = args_dict.get("tags", [])
|
||
published = args_dict.get("published", False)
|
||
|
||
if workflow_file:
|
||
upload_workflow(workflow_file, workflow_name, tags, published)
|
||
else:
|
||
print_status("未指定工作流文件路径，请使用 -f/--workflow_file 参数", "error")
|