Merging changes to enable local modules

jmeekhof · jmeekhof · commit 089761923e1b · 2017-06-05T23:11:52.000-04:00
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -3,6 +3,7 @@
 import json
 import logging
 import os
+import sys
 import time
 from imp import load_source
 from shutil import copy, copyfile
@@ -110,6 +111,12 @@ def invoke(src, alt_event=None, verbose=False):
         path_to_event_file = os.path.join(src, 'event.json')
     event = read(path_to_event_file, loader=json.loads)
 
+    #Tweak to allow module to import local modules
+    try:
+        sys.path.index(src)
+    except:
+        sys.path.append(src)
+
     handler = cfg.get('handler')
     # Inspect the handler string (<module>.<function name>) and translate it
     # into a function we can execute.
