新增注册表补全功能，修复Protocol执行失败

unilabos/app/register.py

@@ -13,17 +13,17 @@ def register_devices_and_resources(mqtt_client, lab_registry):
     """
     logger.info("[UniLab Register] 开始注册设备和资源...")
 
-    # 注册设备信息
-    for device_info in lab_registry.obtain_registry_device_info():
-        mqtt_client.publish_registry(device_info["id"], device_info, False)
-        logger.debug(f"[UniLab Register] 注册设备: {device_info['id']}")
-
-    # 注册资源信息
-    for resource_info in lab_registry.obtain_registry_resource_info():
-        mqtt_client.publish_registry(resource_info["id"], resource_info, False)
-        logger.debug(f"[UniLab Register] 注册资源: {resource_info['id']}")
-
-    time.sleep(10)
+    # # 注册设备信息
+    # for device_info in lab_registry.obtain_registry_device_info():
+    #     mqtt_client.publish_registry(device_info["id"], device_info, False)
+    #     logger.debug(f"[UniLab Register] 注册设备: {device_info['id']}")
+    #
+    # # 注册资源信息
+    # for resource_info in lab_registry.obtain_registry_resource_info():
+    #     mqtt_client.publish_registry(resource_info["id"], resource_info, False)
+    #     logger.debug(f"[UniLab Register] 注册资源: {resource_info['id']}")
+    #
+    # time.sleep(10)
 
     logger.info("[UniLab Register] 设备和资源注册完成.")
 
@@ -34,7 +34,7 @@ def main():
     """
     parser = argparse.ArgumentParser(description="注册设备和资源到 MQTT")
     parser.add_argument(
-        "--registry_path",
+        "--registry",
         type=str,
         default=None,
         action="append",
@@ -46,10 +46,16 @@ def main():
         default=None,
         help="配置文件路径，支持.py格式的Python配置文件",
     )
+    parser.add_argument(
+        "--complete_registry",
+        action="store_true",
+        default=False,
+        help="是否补全注册表",
+    )
     args = parser.parse_args()
 
     # 构建注册表
-    build_registry(args.registry_path)
+    build_registry(args.registry, args.complete_registry)
     load_config_from_file(args.config)
 
     from unilabos.app.mq import mqtt_client