more logging improvements

Author: amancevice

lambda_gateway/__init__.py

@@ -0,0 +1,23 @@
+import logging
+
+
+def set_stream_logger(name, level=logging.DEBUG, format_string=None):
+    """
+    Adapted from boto3.set_stream_logger()
+    """
+    if format_string is None:
+        format_string = \
+            '%(addr)s - - [%(asctime)s] %(levelname)s - %(message)s'
+
+    logger = logging.getLogger(name)
+    handler = logging.StreamHandler()
+    formatter = logging.Formatter(format_string, '%-d/%b/%Y %H:%M:%S')
+    adapter = logging.LoggerAdapter(logger, dict(addr='::1'))
+    logger.setLevel(level)
+    handler.setLevel(level)
+    handler.setFormatter(formatter)
+    logger.addHandler(handler)
+    return adapter
+
+
+logger = set_stream_logger(__name__)

lambda_gateway/event_proxy.py

@@ -1,10 +1,9 @@
 import asyncio
 import json
 import os
-import sys
 from importlib.util import (spec_from_file_location, module_from_spec)
 
-from lambda_gateway import lambda_context
+from lambda_gateway import (lambda_context, logger)
 
 
 class EventProxy:
@@ -35,6 +34,7 @@ def get_handler(self):
 
     def invoke(self, event):
         with lambda_context.start(self.timeout) as context:
+            logger.info('Invoking "%s"', self.handler)
             return asyncio.run(self.invoke_async_with_timeout(event, context))
 
     async def invoke_async(self, event, context=None):
@@ -49,8 +49,8 @@ async def invoke_async(self, event, context=None):
 
         # Reject request if not starting at base path
         if not path.startswith(self.base_path):
-            err = f'Rejected {path} -- Base path is {self.base_path}'
-            sys.stderr.write(f'{err}\n')
+            err = f'Rejected {path} :: Base path is {self.base_path}'
+            logger.error(err)
             return self.jsonify(httpMethod, 403, message='Forbidden')
 
         # Get & invoke Lambda handler
@@ -59,7 +59,7 @@ async def invoke_async(self, event, context=None):
             loop = asyncio.get_running_loop()
             return await loop.run_in_executor(None, handler, event, context)
         except Exception as err:
-            sys.stderr.write(f'{err}\n')
+            logger.error(err)
             message = 'Internal server error'
             return self.jsonify(httpMethod, 502, message=message)