WIP

gmuloc · Jun 7, 2024 · 7932fb2 · 7932fb2
1 parent 5d50a3b
commit 7932fb2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/anta/cli/get/commands.py b/anta/cli/get/commands.py
@@ -50,11 +50,17 @@ def from_cvp(ctx: click.Context, output: Path, host: str, username: str, passwor
     TODO - handle get_inventory and get_devices_in_container failure
     """
     logger.info("Getting authentication token for user '%s' from CloudVision instance '%s'", username, host)
-    token = get_cv_token(cvp_ip=host, cvp_username=username, cvp_password=password, verify_cert=not ignore_cert)
+    is_cvaas = False
+    # TODO: This is not nice handling of authentication failure nor CVaaS detection
+    try:
+        token = get_cv_token(cvp_ip=host, cvp_username=username, cvp_password=password, verify_cert=not ignore_cert)
+    except Exception:
+        token = password
+        is_cvaas = True
 
     clnt = CvpClient()
     try:
-        clnt.connect(nodes=[host], username="", password="", api_token=token)
+        clnt.connect(nodes=[host], username="", password="", api_token=token, is_cvaas=is_cvaas)
     except CvpApiError as error:
         logger.error("Error connecting to CloudVision: %s", error)
         ctx.exit(ExitCode.USAGE_ERROR)