Support structured logs from lambda (requires latest TF AWS provider)

src/lambda.py

@@ -1,5 +1,4 @@
 import logging
-from pythonjsonlogger import jsonlogger
 import urllib.parse
 import botocore
 import boto3
@@ -12,12 +11,17 @@
 import time
 import json
 
-# JSON Logs
-logger = logging.getLogger()
-logHandler = logging.StreamHandler()
-formatter = jsonlogger.JsonFormatter()
-logHandler.setFormatter(formatter)
-logger.addHandler(logHandler)
+# JSON Logs (Lambda handles this automatically, the import is for local execution)
+try:
+    from pythonjsonlogger import jsonlogger
+
+    logger = logging.getLogger()
+    logHandler = logging.StreamHandler()
+    formatter = jsonlogger.JsonFormatter()
+    logHandler.setFormatter(formatter)
+    logger.addHandler(logHandler)
+except ImportError:
+    pass
 
 # Logging Levels
 boto3.set_stream_logger(name="botocore", level=logging.WARN)