完成在main中启动设备可视化

完成在main中启动设备可视化,并输出物料ID:mesh的对应关系resource_model

添加物料模型管理类,遍历物料与resource_model,完成TF数据收集
This commit is contained in:
zhangshixiang
2025-04-24 00:59:43 +08:00
parent 5b4f580a6f
commit 279c5ed519
40 changed files with 4678 additions and 43 deletions

View File

@@ -5,6 +5,7 @@ import sys
import json
import yaml
from copy import deepcopy
import threading
# 首先添加项目根目录到路径
current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -14,7 +15,7 @@ if ilabos_dir not in sys.path:
from unilabos.config.config import load_config, BasicConfig
from unilabos.utils.banner_print import print_status, print_unilab_banner
from unilabos.device_mesh.resource_visalization import ResourceVisualization
def parse_args():
"""解析命令行参数"""
@@ -102,7 +103,8 @@ def main():
print_unilab_banner(args_dict)
# 注册表
build_registry(args_dict["registry_path"])
registry_dict = build_registry(args_dict["registry_path"])
if args_dict["graph"] is not None:
import unilabos.resources.graphio as graph_res
@@ -150,8 +152,16 @@ def main():
signal.signal(signal.SIGTERM, _exit)
mqtt_client.start()
resource_visualization = ResourceVisualization(args_dict["devices_config"], args_dict["resources_config"],registry_dict)
start_backend(**args_dict)
start_server()
print('-'*100)
print(resource_visualization.resource_model)
print(json.dumps(args_dict["resources_config"], indent=4, ensure_ascii=False))
print('-'*100)
server_thread = threading.Thread(target=start_server)
server_thread.start()
resource_visualization.start()
if __name__ == "__main__":