添加环境检查功能，支持自动安装缺失的Python包；更新命令行参数以跳过环境检查；优化代码格式。

unilabos/app/main.py

@@ -43,10 +43,11 @@ def convert_argv_dashes_to_underscores(args: argparse.ArgumentParser):
     for i, arg in enumerate(sys.argv):
         for option_string in option_strings:
             if arg.startswith(option_string):
-                new_arg = arg[:2] + arg[2:len(option_string)].replace("-", "_") + arg[len(option_string):]
+                new_arg = arg[:2] + arg[2 : len(option_string)].replace("-", "_") + arg[len(option_string) :]
                 sys.argv[i] = new_arg
                 break
 
+
 def parse_args():
     """解析命令行参数"""
     parser = argparse.ArgumentParser(description="Start Uni-Lab Edge server.")
@@ -128,6 +129,11 @@ def parse_args():
         default="",
         help="实验室唯一ID，也可通过环境变量 UNILABOS_MQCONFIG_LABID 设置或传入--config设置",
     )
+    parser.add_argument(
+        "--skip_env_check",
+        action="store_true",
+        help="跳过启动时的环境依赖检查",
+    )
     return parser
 
 
@@ -138,6 +144,17 @@ def main():
     convert_argv_dashes_to_underscores(args)
     args_dict = vars(args.parse_args())
 
+    # 环境检查 - 检查并自动安装必需的包 (可选)
+    if not args_dict.get("skip_env_check", False):
+        from unilabos.utils.environment_check import check_environment
+
+        print_status("正在进行环境依赖检查...", "info")
+        if not check_environment(auto_install=True):
+            print_status("环境检查失败，程序退出", "error")
+            os._exit(1)
+    else:
+        print_status("跳过环境依赖检查", "warning")
+
     # 加载配置文件，优先加载config，然后从env读取
     config_path = args_dict.get("config")
     if os.getcwd().endswith("unilabos_data"):
@@ -151,17 +168,22 @@ def main():
             if not os.path.exists(config_path):
                 print_status(
                     f"当前工作目录 {working_dir} 未找到local_config.py，请通过 --config 传入 local_config.py 文件路径",
-                    "error")
+                    "error",
+                )
                 os._exit(1)
     elif os.path.exists(working_dir) and os.path.exists(os.path.join(working_dir, "local_config.py")):
         config_path = os.path.join(working_dir, "local_config.py")
-    elif not config_path and (not os.path.exists(working_dir) or not os.path.exists(os.path.join(working_dir, "local_config.py"))):
+    elif not config_path and (
+        not os.path.exists(working_dir) or not os.path.exists(os.path.join(working_dir, "local_config.py"))
+    ):
         print_status(f"未指定config路径，可通过 --config 传入 local_config.py 文件路径", "info")
         print_status(f"您是否为第一次使用？并将当前路径 {working_dir} 作为工作目录？ (Y/n)", "info")
         if input() != "n":
             os.makedirs(working_dir, exist_ok=True)
             config_path = os.path.join(working_dir, "local_config.py")
-            shutil.copy(os.path.join(os.path.dirname(os.path.dirname(__file__)), "config", "example_config.py"), config_path)
+            shutil.copy(
+                os.path.join(os.path.dirname(os.path.dirname(__file__)), "config", "example_config.py"), config_path
+            )
             print_status(f"已创建 local_config.py 路径： {config_path}", "info")
             print_status(f"请在文件夹中配置lab_id，放入下载的CA.crt、lab.crt、lab.key重新启动本程序", "info")
             os._exit(1)