Enable black in tox.ini
[osm/MON.git] / osm_mon / collector / infra_collectors / base_osinfra.py
index 018aa9a..8ab34e4 100644 (file)
@@ -66,7 +66,9 @@ class BaseOpenStackInfraCollector(BaseVimInfraCollector):
         metrics.append(vim_status_metric)
         vnfrs = self.common_db.get_vnfrs(vim_account_id=self.vim_account["_id"])
         if self.conf.get("collector", "vm_infra_metrics"):
-            vm_infra_metrics_enabled = str(self.conf.get("collector", "vm_infra_metrics")).lower() in ("yes", "true", "1")
+            vm_infra_metrics_enabled = str(
+                self.conf.get("collector", "vm_infra_metrics")
+            ).lower() in ("yes", "true", "1")
         else:
             vm_infra_metrics_enabled = True
         if vm_infra_metrics_enabled:
@@ -94,7 +96,7 @@ class BaseOpenStackInfraCollector(BaseVimInfraCollector):
                     }
                     try:
                         vm = self.nova.servers.get(resource_uuid)
-                        vm_status = (0 if (vm.status == 'ERROR') else 1)
+                        vm_status = 0 if (vm.status == "ERROR") else 1
                         vm_status_metric = Metric(tags, "vm_status", vm_status)
                     except Exception as e:
                         log.warning("VM status is not OK: %s" % e)
@@ -116,42 +118,104 @@ class BaseOpenStackInfraCollector(BaseVimInfraCollector):
             vimacc_resources = self.vim_account["resources"]
             # Compute resources
             try:
-                com_lim = self.nova.limits.get()._info['absolute']
-                if ("compute" in vimacc_resources) \
-                   and ((vimacc_resources["compute"]["ram"]["total"] != com_lim['maxTotalRAMSize'])
-                   or (vimacc_resources["compute"]["vcpus"]["total"] != com_lim['maxTotalCores'])
-                   or (vimacc_resources["compute"]["ram"]["used"] != com_lim['totalRAMUsed'])
-                   or (vimacc_resources["compute"]["vcpus"]["used"] != com_lim['totalCoresUsed'])
-                   or (vimacc_resources["compute"]["instances"]["total"] != com_lim['maxTotalInstances'])
-                   or (vimacc_resources["compute"]["instances"]["used"] != com_lim['totalInstancesUsed'])):
-                    update_dict = {"resources.compute": {"ram": {"total": com_lim['maxTotalRAMSize'],
-                                                                 "used": com_lim['totalRAMUsed']},
-                                                         "vcpus": {"total": com_lim['maxTotalCores'],
-                                                                   "used": com_lim['totalCoresUsed']},
-                                                         "instances": {"total": com_lim['maxTotalInstances'],
-                                                                       "used": com_lim['totalInstancesUsed']}}}
-                    suc_value = self.common_db.set_vim_account(str(self.vim_account['_id']), update_dict)
+                com_lim = self.nova.limits.get()._info["absolute"]
+                if ("compute" in vimacc_resources) and (
+                    (
+                        vimacc_resources["compute"]["ram"]["total"]
+                        != com_lim["maxTotalRAMSize"]
+                    )
+                    or (
+                        vimacc_resources["compute"]["vcpus"]["total"]
+                        != com_lim["maxTotalCores"]
+                    )
+                    or (
+                        vimacc_resources["compute"]["ram"]["used"]
+                        != com_lim["totalRAMUsed"]
+                    )
+                    or (
+                        vimacc_resources["compute"]["vcpus"]["used"]
+                        != com_lim["totalCoresUsed"]
+                    )
+                    or (
+                        vimacc_resources["compute"]["instances"]["total"]
+                        != com_lim["maxTotalInstances"]
+                    )
+                    or (
+                        vimacc_resources["compute"]["instances"]["used"]
+                        != com_lim["totalInstancesUsed"]
+                    )
+                ):
+                    update_dict = {
+                        "resources.compute": {
+                            "ram": {
+                                "total": com_lim["maxTotalRAMSize"],
+                                "used": com_lim["totalRAMUsed"],
+                            },
+                            "vcpus": {
+                                "total": com_lim["maxTotalCores"],
+                                "used": com_lim["totalCoresUsed"],
+                            },
+                            "instances": {
+                                "total": com_lim["maxTotalInstances"],
+                                "used": com_lim["totalInstancesUsed"],
+                            },
+                        }
+                    }
+                    suc_value = self.common_db.set_vim_account(
+                        str(self.vim_account["_id"]), update_dict
+                    )
                     log.info("Compute resources update in mongoDB  = %s" % suc_value)
             except Exception as e:
                 log.warning("Error in updating compute resources: %s" % e)
 
             # Volume resources
             try:
-                vol_lim = self.cinder.limits.get()._info['absolute']
-                if ("storage" in vimacc_resources) and\
-                   ((vimacc_resources["storage"]["volumes"]["total"] != vol_lim['maxTotalVolumes'])
-                   or (vimacc_resources["storage"]["snapshots"]["total"] != vol_lim['maxTotalSnapshots'])
-                   or (vimacc_resources["storage"]["volumes"]["used"] != vol_lim['totalVolumesUsed'])
-                   or (vimacc_resources["storage"]["snapshots"]["used"] != vol_lim['totalSnapshotsUsed'])
-                   or (vimacc_resources["storage"]["storage"]["total"] != vol_lim['maxTotalVolumeGigabytes'])
-                   or (vimacc_resources["storage"]["storage"]["used"] != vol_lim['totalGigabytesUsed'])):
-                    update_dict = {"resources.storage": {"volumes": {"total": vol_lim['maxTotalVolumes'],
-                                                                     "used": vol_lim['totalVolumesUsed']},
-                                                         "snapshots": {"total": vol_lim['maxTotalSnapshots'],
-                                                                       "used": vol_lim['totalSnapshotsUsed']},
-                                                         "storage": {"total": vol_lim['maxTotalVolumeGigabytes'],
-                                                                     "used": vol_lim['totalGigabytesUsed']}}}
-                    suc_value = self.common_db.set_vim_account(str(self.vim_account['_id']), update_dict)
+                vol_lim = self.cinder.limits.get()._info["absolute"]
+                if ("storage" in vimacc_resources) and (
+                    (
+                        vimacc_resources["storage"]["volumes"]["total"]
+                        != vol_lim["maxTotalVolumes"]
+                    )
+                    or (
+                        vimacc_resources["storage"]["snapshots"]["total"]
+                        != vol_lim["maxTotalSnapshots"]
+                    )
+                    or (
+                        vimacc_resources["storage"]["volumes"]["used"]
+                        != vol_lim["totalVolumesUsed"]
+                    )
+                    or (
+                        vimacc_resources["storage"]["snapshots"]["used"]
+                        != vol_lim["totalSnapshotsUsed"]
+                    )
+                    or (
+                        vimacc_resources["storage"]["storage"]["total"]
+                        != vol_lim["maxTotalVolumeGigabytes"]
+                    )
+                    or (
+                        vimacc_resources["storage"]["storage"]["used"]
+                        != vol_lim["totalGigabytesUsed"]
+                    )
+                ):
+                    update_dict = {
+                        "resources.storage": {
+                            "volumes": {
+                                "total": vol_lim["maxTotalVolumes"],
+                                "used": vol_lim["totalVolumesUsed"],
+                            },
+                            "snapshots": {
+                                "total": vol_lim["maxTotalSnapshots"],
+                                "used": vol_lim["totalSnapshotsUsed"],
+                            },
+                            "storage": {
+                                "total": vol_lim["maxTotalVolumeGigabytes"],
+                                "used": vol_lim["totalGigabytesUsed"],
+                            },
+                        }
+                    }
+                    suc_value = self.common_db.set_vim_account(
+                        str(self.vim_account["_id"]), update_dict
+                    )
                     log.info("Volume resources update in mongoDB = %s" % suc_value)
             except Exception as e:
                 log.warning("Error in updating volume resources: %s" % e)
@@ -159,20 +223,51 @@ class BaseOpenStackInfraCollector(BaseVimInfraCollector):
             # Network resources
             try:
                 net_lim = self.neutron.show_quota_details(self.tenant_id)["quota"]
-                if ("network" in vimacc_resources) and\
-                   ((vimacc_resources["network"]["networks"]["total"] != net_lim["network"]["limit"])
-                   or (vimacc_resources["network"]["networks"]["used"] != net_lim['network']['used'])
-                   or (vimacc_resources["network"]["subnets"]["total"] != net_lim['subnet']['limit'])
-                   or (vimacc_resources["network"]["subnets"]["used"] != net_lim['subnet']['used'])
-                   or (vimacc_resources["network"]["floating_ips"]["total"] != net_lim['floatingip']['limit'])
-                   or (vimacc_resources["network"]["floating_ips"]["used"] != net_lim['floatingip']['used'])):
-                    update_dict = {"resources.network": {"networks": {"total": net_lim['network']['limit'],
-                                                                      "used": net_lim['network']['used']},
-                                                         "subnets": {"total": net_lim['subnet']['limit'],
-                                                                     "used": net_lim['subnet']['used']},
-                                                         "floating_ips": {"total": net_lim['floatingip']['limit'],
-                                                                          "used": net_lim['floatingip']['used']}}}
-                    suc_value = self.common_db.set_vim_account(str(self.vim_account['_id']), update_dict)
+                if ("network" in vimacc_resources) and (
+                    (
+                        vimacc_resources["network"]["networks"]["total"]
+                        != net_lim["network"]["limit"]
+                    )
+                    or (
+                        vimacc_resources["network"]["networks"]["used"]
+                        != net_lim["network"]["used"]
+                    )
+                    or (
+                        vimacc_resources["network"]["subnets"]["total"]
+                        != net_lim["subnet"]["limit"]
+                    )
+                    or (
+                        vimacc_resources["network"]["subnets"]["used"]
+                        != net_lim["subnet"]["used"]
+                    )
+                    or (
+                        vimacc_resources["network"]["floating_ips"]["total"]
+                        != net_lim["floatingip"]["limit"]
+                    )
+                    or (
+                        vimacc_resources["network"]["floating_ips"]["used"]
+                        != net_lim["floatingip"]["used"]
+                    )
+                ):
+                    update_dict = {
+                        "resources.network": {
+                            "networks": {
+                                "total": net_lim["network"]["limit"],
+                                "used": net_lim["network"]["used"],
+                            },
+                            "subnets": {
+                                "total": net_lim["subnet"]["limit"],
+                                "used": net_lim["subnet"]["used"],
+                            },
+                            "floating_ips": {
+                                "total": net_lim["floatingip"]["limit"],
+                                "used": net_lim["floatingip"]["used"],
+                            },
+                        }
+                    }
+                    suc_value = self.common_db.set_vim_account(
+                        str(self.vim_account["_id"]), update_dict
+                    )
                     log.info("Network resources update in mongoDB = %s" % suc_value)
             except Exception as e:
                 log.warning("Error in updating network resources: %s" % e)
@@ -193,4 +288,7 @@ class BaseOpenStackInfraCollector(BaseVimInfraCollector):
     def _build_neutron_client(self, vim_account: dict) -> tuple:
         # sess = OpenstackUtils.get_session(vim_account)
         tenant_id = self.vim_session.get_project_id()
-        return neutron_client.Client("2", session=self.vim_session, timeout=10), tenant_id
+        return (
+            neutron_client.Client("2", session=self.vim_session, timeout=10),
+            tenant_id,
+        )