def server_handler()

in src/key_server.py [0:0]

• 19 lines of code
• 4 McCabe index (conditional complexity)

def server_handler(event, context):
    """
    This function is the entry point for the SPEKE reference key
    server Lambda. This is invoked from the API Gateway resource.
    """
    try:
        print(event)
        body = event['body']
        if event['isBase64Encoded']:
            body = base64.b64decode(body)
        cache = KeyCache(BUCKET_NAME, CLIENT_URL_PREFIX)
        generator = KeyGenerator()
        headers_from_event = event['headers']
        speke_version = headers_from_event.get('x-speke-version', '1.0')

        if speke_version == "2.0":
            response = ServerResponseBuilderV2(body, cache, generator).get_response()
        else:
            response = ServerResponseBuilder(body, cache, generator).get_response()
        
        print(response)
        return response
    except Exception as exception:
        print("EXCEPTION {}".format(exception))
        return {"isBase64Encoded": False, "statusCode": 500, "headers": {"Content-Type": "text/plain"}, "body": str(exception)}