import boto3
import os
import json

# reference: https://aws.amazon.com/premiumsupport/knowledge-center/start-stop-lambda-eventbridge/

ec2 = boto3.client('ec2', region_name=os.environ['AWS_REGION'])

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']))
    else:
        resp = "Event action not provided"
    return resp