Reformat POL to standardized format

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

diff --git a/osm_policy_module/common/lcm_client.py b/osm_policy_module/common/lcm_client.py
index 089e541..5085a86 100644 (file)
--- a/osm_policy_module/common/lcm_client.py
+++ b/osm_policy_module/common/lcm_client.py
@@ -47,7 +47,9 @@ class LcmClient:
             loop = asyncio.get_event_loop()
         self.loop = loop
 
-    async def scale(self, nsr_id: str, scaling_group_name: str, vnf_member_index: str, action: str):
+    async def scale(
+        self, nsr_id: str, scaling_group_name: str, vnf_member_index: str, action: str
+    ):
         """
         Sends scaling action to LCM through the message bus.
 
@@ -57,14 +59,25 @@ class LcmClient:
         :param action: Scaling action to be executed. Valid values: scale_in, scale_out
         :return:
         """
-        log.debug("scale %s %s %s %s", nsr_id, scaling_group_name, vnf_member_index, action)
+        log.debug(
+            "scale %s %s %s %s", nsr_id, scaling_group_name, vnf_member_index, action
+        )
         nsr = self.db_client.get_nsr(nsr_id)
-        nslcmop = self._generate_nslcmop(nsr_id, scaling_group_name, vnf_member_index, action, nsr['_admin'])
+        nslcmop = self._generate_nslcmop(
+            nsr_id, scaling_group_name, vnf_member_index, action, nsr["_admin"]
+        )
         self.db_client.create_nslcmop(nslcmop)
         log.debug("Sending scale action message: %s", json.dumps(nslcmop))
         await self.msg_bus.aiowrite("ns", "scale", nslcmop)
 
-    def _generate_nslcmop(self, nsr_id: str, scaling_group_name: str, vnf_member_index: str, action: str, admin: dict):
+    def _generate_nslcmop(
+        self,
+        nsr_id: str,
+        scaling_group_name: str,
+        vnf_member_index: str,
+        action: str,
+        admin: dict,
+    ):
         """
         Builds scaling nslcmop.
 
@@ -75,7 +88,14 @@ class LcmClient:
         :param admin: Dict corresponding to the _admin section of the nsr. Required keys: projects_read, projects_write.
         :return:
         """
-        log.debug("_generate_nslcmop %s %s %s %s %s", nsr_id, scaling_group_name, vnf_member_index, action, admin)
+        log.debug(
+            "_generate_nslcmop %s %s %s %s %s",
+            nsr_id,
+            scaling_group_name,
+            vnf_member_index,
+            action,
+            admin,
+        )
         _id = str(uuid.uuid4())
         now = time.time()
         params = {
@@ -84,10 +104,10 @@ class LcmClient:
                 "scaleVnfType": action.upper(),
                 "scaleByStepData": {
                     "scaling-group-descriptor": scaling_group_name,
-                    "member-vnf-index": vnf_member_index
-                }
+                    "member-vnf-index": vnf_member_index,
+                },
             },
-            "scaleTime": "{}Z".format(datetime.datetime.utcnow().isoformat())
+            "scaleTime": "{}Z".format(datetime.datetime.utcnow().isoformat()),
         }
 
         nslcmop = {
@@ -106,8 +126,8 @@ class LcmClient:
                 "nsInstance": "/osm/nslcm/v1/ns_instances/" + nsr_id,
             },
             "_admin": {
-                "projects_read": admin['projects_read'],
-                "projects_write": admin['projects_write']
-            }
+                "projects_read": admin["projects_read"],
+                "projects_write": admin["projects_write"],
+            },
         }
         return nslcmop