Reformat POL to standardized format

[osm/POL.git] / osm_policy_module / common / common_db_client.py

diff --git a/osm_policy_module/common/common_db_client.py b/osm_policy_module/common/common_db_client.py
index 9ec183a..1bd4e76 100644 (file)
--- a/osm_policy_module/common/common_db_client.py
+++ b/osm_policy_module/common/common_db_client.py
@@ -29,47 +29,56 @@ from osm_policy_module.core.exceptions import VdurNotFound
 
 class CommonDbClient:
     def __init__(self, config: Config):
-        if config.get('database', 'driver') == "mongo":
+        if config.get("database", "driver") == "mongo":
             self.common_db = dbmongo.DbMongo()
-        elif config.get('database', 'driver') == "memory":
+        elif config.get("database", "driver") == "memory":
             self.common_db = dbmemory.DbMemory()
         else:
-            raise Exception("Unknown database driver {}".format(config.get('section', 'driver')))
+            raise Exception(
+                "Unknown database driver {}".format(config.get("section", "driver"))
+            )
         self.common_db.db_connect(config.get("database"))
 
     def get_vnfr(self, nsr_id: str, member_index: str):
-        vnfr = self.common_db.get_one("vnfrs",
-                                      {"nsr-id-ref": nsr_id, "member-vnf-index-ref": member_index})
+        vnfr = self.common_db.get_one(
+            "vnfrs", {"nsr-id-ref": nsr_id, "member-vnf-index-ref": member_index}
+        )
         return vnfr
 
     def get_vnfrs(self, nsr_id: str):
         # TODO: Change for multiple DF support
-        nsr_nsd_df = self.get_nsr(nsr_id)['nsd'].get('df', [{}])[0]
-        all_nsd_member_vnf_index = [vnf.get('id') for vnf in nsr_nsd_df.get('vnf-profile', [])]
-        return [self.get_vnfr(nsr_id, member_index) for member_index in all_nsd_member_vnf_index]
+        nsr_nsd_df = self.get_nsr(nsr_id)["nsd"].get("df", [{}])[0]
+        all_nsd_member_vnf_index = [
+            vnf.get("id") for vnf in nsr_nsd_df.get("vnf-profile", [])
+        ]
+        return [
+            self.get_vnfr(nsr_id, member_index)
+            for member_index in all_nsd_member_vnf_index
+        ]
 
     def get_vnfd(self, vnfd_id: str):
-        vnfr = self.common_db.get_one("vnfds",
-                                      {"_id": vnfd_id})
+        vnfr = self.common_db.get_one("vnfds", {"_id": vnfd_id})
         return vnfr
 
     def get_nsr(self, nsr_id: str):
-        nsr = self.common_db.get_one("nsrs",
-                                     {"id": nsr_id})
+        nsr = self.common_db.get_one("nsrs", {"id": nsr_id})
         return nsr
 
     def get_nslcmop(self, nslcmop_id):
-        nslcmop = self.common_db.get_one("nslcmops",
-                                         {"_id": nslcmop_id})
+        nslcmop = self.common_db.get_one("nslcmops", {"_id": nslcmop_id})
         return nslcmop
 
     def get_vdur(self, nsr_id, member_index, vdur_name):
         vnfr = self.get_vnfr(nsr_id, member_index)
-        for vdur in vnfr['vdur']:
-            if vdur['name'] == vdur_name:
+        for vdur in vnfr["vdur"]:
+            if vdur["name"] == vdur_name:
                 return vdur
-        raise VdurNotFound('vdur not found for nsr-id %s, member_index %s and vdur_name %s', nsr_id, member_index,
-                           vdur_name)
+        raise VdurNotFound(
+            "vdur not found for nsr-id %s, member_index %s and vdur_name %s",
+            nsr_id,
+            member_index,
+            vdur_name,
+        )
 
     def create_nslcmop(self, nslcmop):
         self.common_db.create("nslcmops", nslcmop)