Change Python logger format in AWS Lambda

lambda_logger_format.py

# Python logger in AWS Lambda has a preset format. To change the format of the logging statement, 
# remove the logging handler & add a new handler with the required format

import logging
import sys

def setup_logging():
    logger = logging.getLogger()
    for h in logger.handlers:
      logger.removeHandler(h)
    
    h = logging.StreamHandler()
    
    # use whatever format you want here
    log_format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
    h.setFormatter(logging.Formatter(log_format))
    logger.addHandler(h)
    logger.setLevel(logging.INFO)
    
    return logger

def lambda_handler(event, context):
    logger = setup_logging()
    logger.info("This is a test log  info statement!")
    logger.warning("This is a test log warning statement!")
    logger.error("This is a test log error statement")
    logger.critical("This is a test log critical error statement")
    return
    
# START RequestId: b8f58da0-d55b-11e7-ba6e-e5de6a0dbbc1 Version: $LATEST
# 2017-11-29 23:19:18,061 - root - INFO - This is a test log  info statement!
# 2017-11-29 23:19:18,061 - root - WARNING - This is a test log warning statement!
# 2017-11-29 23:19:18,061 - root - ERROR - This is a test log error statement
# 2017-11-29 23:19:18,061 - root - CRITICAL - This is a test log critical error statement
# END RequestId: b8f58da0-d55b-11e7-ba6e-e5de6a0dbbc1
# REPORT RequestId: b8f58da0-d55b-11e7-ba6e-e5de6a0dbbc1	Duration: 11.67 ms	Billed Duration: 100 ms 	Memory Size: 128 MB	Max Memory Used: 19 MB