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