Lambda import in ecs container

lambda_ecs.py

# import goes here

def lambda_handler(event, context):
  # lambda code goes here
  pass

if __name__ == '__main__':  # in case if we run it in a container
    if 'EVENT' not in os.environ:
        raise Exception('No event provided in os.environ.')
    else:
        eventstr = os.environ['EVENT']  # we always expect EVENT to be encoded as base64
        event = base64.b64decode(eventstr.encode('utf-8'))
        event = json.loads(event)
        results = lambda_handler(event, [])