Merge pull request #677 from NVIDIA/am/click-cli

Re-work CLI implementation to use Click

Author: amaslenn

pyproject.toml

@@ -27,9 +27,10 @@ dependencies = [
   "jinja2==3.1.6",
   "websockets==15.0.1",
   "rich==14.0.0",
+  "click>=8.2.1",
 ]
 requires-python = ">=3.10"
-scripts = { cloudai = "cloudai.__main__:main" }
+scripts = { cloudai = "cloudai.cli:main" }
 
   [project.optional-dependencies]
   dev = [

src/cloudai/__main__.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
-# Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -14,62 +14,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import logging
-import logging.config
-
-from .cli import CloudAICLI
-from .handlers import handle_dry_run_and_run, handle_generate_report, handle_install_and_uninstall
-
-
-def setup_logging(log_file: str, log_level: str) -> None:
-    """
-    Configure logging for the application.
-
-    Args:
-        log_level (str): The logging level (e.g., DEBUG, INFO).
-        log_file (str): The name of the log file.
-    """
-    numeric_level = getattr(logging, log_level.upper(), None)
-    if not isinstance(numeric_level, int):
-        raise ValueError(f"Invalid log level: {log_level}")
-
-    LOGGING_CONFIG = {
-        "version": 1,
-        "disable_existing_loggers": True,
-        "formatters": {
-            "standard": {"format": "%(asctime)s - %(levelname)s - %(message)s"},
-            "short": {"format": "[%(levelname)s] %(message)s"},
-        },
-        "handlers": {
-            "default": {
-                "level": log_level.upper(),
-                "formatter": "short",
-                "class": "logging.StreamHandler",
-                "stream": "ext://sys.stdout",
-            },
-            "debug_file": {
-                "level": "DEBUG",
-                "formatter": "standard",
-                "class": "logging.FileHandler",
-                "filename": log_file,
-                "mode": "w",
-            },
-        },
-        "loggers": {
-            "": {
-                "handlers": ["default", "debug_file"],
-                "level": "DEBUG",
-                "propagate": False,
-            },
-        },
-    }
-    logging.config.dictConfig(LOGGING_CONFIG)
-
+from .cli import main, setup_logging
 
 __all__ = [
-    "CloudAICLI",
-    "handle_dry_run_and_run",
-    "handle_generate_report",
-    "handle_install_and_uninstall",
+    "main",
     "setup_logging",
 ]