community.general/test/integration/targets/aws_lambda/files/mini_lambda.py

from __future__ import print_function
import json
import os


def handler(event, context):
    """
    The handler function is the function which gets called each time
    the lambda is run.
    """
    # printing goes to the cloudwatch log allowing us to simply debug the lambda if we can find
    # the log entry.
    print("got event:\n" + json.dumps(event))

    # if the name parameter isn't present this can throw an exception
    # which will result in an amazon chosen failure from the lambda
    # which can be completely fine.

    name = event["name"]

    # we can use environment variables as part of the configuration of the lambda
    # which can change the behaviour of the lambda without needing a new upload

    extra = os.environ.get("EXTRA_MESSAGE")
    if extra is not None and len(extra) > 0:
        greeting = "hello {0}. {1}".format(name, extra)
    else:
        greeting = "hello " + name

    return {"message": greeting}


def main():
    """
    This main function will normally never be called during normal
    lambda use.  It is here for testing the lambda program only.
    """
    event = {"name": "james"}
    context = None
    print(handler(event, context))


if __name__ == '__main__':
    main()