feat: code reformat

Author: radimsuckr

functions/lambda.py

@@ -59,13 +59,14 @@ class InstanceRefreshInProgress(LambdaError):
 
 def get_current_image_id():
     """Returns current AMI from SSM"""
-    param_value = ""
+    param_value = None
     param = ssm.get_parameter(Name=SSM_PARAMETER_NAME)
-    if param['Parameter']['Value'].startswith('ami-'):
-      image_id = param['Parameter']['Value']
-    else: 
-      param_value = json.loads(param['Parameter']['Value'])
-      image_id = param_value['image_id']
+    value = param['Parameter']['Value']
+    if value.startswith('ami-'):
+        image_id = value
+    else:
+        param_value = json.loads(value)
+        image_id = param_value['image_id']
     logger.info('Newest image_id is "%s"', image_id)
     return image_id