[issue-8] add support for ssm parameter with value starting with ami-

Author: baptiste-s

functions/lambda.py

@@ -59,11 +59,15 @@ class InstanceRefreshInProgress(LambdaError):
 
 def get_current_image_id():
     """Returns current AMI from SSM"""
+    param_value = ""
     param = ssm.get_parameter(Name=SSM_PARAMETER_NAME)
-    param_value = json.loads(param['Parameter']['Value'])
-    image_id = param_value['image_id']
+    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']
     logger.info('Newest image_id is "%s"', image_id)
-    return param_value['image_id']
+    return image_id
 
 
 def get_launch_template_name_and_auto_scaling_group():