style: minor changes

lambda/ec2-scheduler/lambda_function.py

@@ -4,13 +4,14 @@ import json
 
 # reference: https://aws.amazon.com/premiumsupport/knowledge-center/start-stop-lambda-eventbridge/
 
-ec2 = boto3.client('ec2', region_name=os.environ['region_name'])
-
 def lambda_handler(event, context):
-    if (event['action'] == 'start'):
-        resp = ec2.start_instances(InstanceIds=json.loads(os.environ['instances']))
-    elif (event['action'] == 'stop'):
-        resp = ec2.stop_instances(InstanceIds=json.loads(os.environ['instances']))
+    ec2 = boto3.client('ec2', region_name=os.environ['region_name'])
+    instances = json.loads(os.environ['instances'])
+    if event['action'] == 'start':
+        resp = ec2.start_instances(InstanceIds=instances)
+    elif event['action'] == 'stop':
+        resp = ec2.stop_instances(InstanceIds=instances)
     else:
-        resp = "Event action not provided"
+        raise ValueError("Invalid event action")
     return resp
+