Merge pull request #34 from SamuelM333/master

Allow kwargs in lambda handlers

Author: tianchu

datadog_lambda/wrapper.py

@@ -59,7 +59,6 @@ def __init__(self, func):
 
     def _before(self, event, context):
         set_cold_start()
-
         try:
             submit_invocations_metric(context)
             # Extract Datadog trace context from incoming requests
@@ -77,10 +76,10 @@ def _after(self, event, context):
         except Exception:
             traceback.print_exc()
 
-    def __call__(self, event, context):
+    def __call__(self, event, context, **kwargs):
         self._before(event, context)
         try:
-            return self.func(event, context)
+            return self.func(event, context, **kwargs)
         except Exception:
             submit_errors_metric(context)
             raise

scripts/run_tests.sh

@@ -1,4 +1,4 @@
-#!/bin/sh
+#!/bin/bash
 
 # Unless explicitly stated otherwise all files in this repository are licensed
 # under the Apache License Version 2.0.