Firetail Python Lambda Middleware
Overview
The purpose of this module is to correctly log out the AWS Lambda event and response payload to allow the firetail extension to then send it on to the firetail logging api
The firetail_handler is a decorator that wraps around an event handler function in a AWS Lambda to extract the event and response payloads into a base64 logging message.
###Supported Lambda Runtimes
Installation
Install the module with using pip
pip install -U firetail-lambdaImplementing Middleware in lambda function
import json
from firetail_lambda import firetail_handler, firetail_app
app = firetail_app()
@firetail_handler(app)
def lambda_handler(event, context):
return {
"statusCode": 200,
"body": json.dumps({
"message": "Hello"
})
}Multiple Event handlers
import json
from firetail_lambda import firetail_handler, firetail_app
app = firetail_app()
@firetail_handler(app)
def lambda_handler(event, context):
return {
"statusCode": 200,
"body": json.dumps({
"message": "Hello"
})
}
@firetail_handler(app)
def lambda_handler_2(event, context):
return {
"statusCode": 200,
"body": json.dumps({
"message": "Hello 2"
})
}Custom Sanitization callback
import copy
import json
from firetail_lambda import firetail_handler, firetail_app
def sanitize_payloads(event, response):
new_event = copy.copy(event)
remove_headers = ['authorization','Authorization', 'x-api-key']
if 'headers' in event:
for header in remove_headers:
if header in event['headers']:
del new_event['headers'][header]
if 'multiValueHeaders' in event and header in event['multiValueHeaders']:
del new_event['multiValueHeaders'][header]
return new_event, response
app = firetail_app()
app.sanitization_callback = sanitize_payloads
@firetail_handler(app)
def lambda_handler(event, context):
return {
"statusCode": 200,
"body": json.dumps({
"message": "Hello"
})
}
