X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=src%2Fosm_ngsa%2Fdags%2Fmultivim_vim_status.py;h=d9f54bb12cfca254f99eb720a16b55d82f020b24;hb=HEAD;hp=53078790dbe83e1058926420a68514ebfead6822;hpb=9bea7378e9011b0f8835e97fd1a617da9dae3680;p=osm%2FNG-SA.git

diff --git a/src/osm_ngsa/dags/multivim_vim_status.py b/src/osm_ngsa/dags/multivim_vim_status.py
index 5307879..d9f54bb 100644
--- a/src/osm_ngsa/dags/multivim_vim_status.py
+++ b/src/osm_ngsa/dags/multivim_vim_status.py
@@ -85,7 +85,7 @@ def create_dag(dag_id, dag_number, dag_description, vim_id):
                 vim_type = vim_account["config"]["vim_type"].lower()
                 if vim_type == "vio" and "vrops_site" not in vim_account["config"]:
                     vim_type = "openstack"
-            if vim_type == "openstack":
+            if vim_type == "openstack" or vim_type == "vio":
                 return OpenStackCollector(vim_account)
             if vim_type == "gcp":
                 return GcpCollector(vim_account)
@@ -122,7 +122,8 @@ def create_dag(dag_id, dag_number, dag_description, vim_id):
             if collector:
                 status = collector.is_vim_ok()
                 logger.info(f"VIM status: {status}")
-                metric.labels(vim_id).set(1)
+                if status:
+                    metric.labels(vim_id).set(1)
             else:
                 logger.info("Error creating VIM collector")
             # Push to Prometheus