Merge pull request #39 from HDE/master

Release v0.1.7

Author: yxd-hde

.gitignore

@@ -58,3 +58,6 @@ target/
 
 # Python environment
 .python-version
+
+# Vitual Environments
+venv/

LICENSE

@@ -1,6 +1,6 @@
 The MIT License (MIT)
 
-Copyright (c) 2015-2017 HDE, Inc.
+Copyright (c) 2015-2018 HDE, Inc.
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

README.md

@@ -8,7 +8,7 @@ Run lambda function on local machine
 
 ## Prepare development environment
 
-Please use a newly created virtualenv of Python 2.7 or Python 3.6 .
+Please use a newly created virtualenv of Python 2.7 or Python 3.6.
 
 ## Installation
 

README.rst

@@ -9,7 +9,7 @@ Run lambda function on local machine
 Prepare development environment
 -------------------------------
 
-Please use a newly created virtualenv of Python 2.7 or Python 3.6 .
+Please use a newly created virtualenv of Python 2.7 or Python 3.6.
 
 Installation
 ------------

lambda_local/__init__.py

@@ -1,7 +1,7 @@
 '''
 python-lambda-local: Main module
 
-Copyright 2015-2017 HDE, Inc.
+Copyright 2015-2018 HDE, Inc.
 Licensed under MIT.
 '''
 

lambda_local/context.py

@@ -1,5 +1,5 @@
 '''
-Copyright 2015-2017 HDE, Inc.
+Copyright 2015-2018 HDE, Inc.
 Licensed under MIT.
 '''
 

lambda_local/environment_variables.py

@@ -2,6 +2,11 @@
 import os
 
 
+def export_variables(environment_variables):
+    for env_name, env_value in environment_variables.items():
+        os.environ[str(env_name)] = str(env_value)
+
+
 def set_environment_variables(json_file_path):
     """
     Read and set environment variables from a flat json file.
@@ -25,5 +30,4 @@ def set_environment_variables(json_file_path):
         with open(json_file_path) as json_file:
             env_vars = json.loads(json_file.read())
 
-            for env_name, env_value in env_vars.items():
-                os.environ[str(env_name)] = str(env_value)
+            export_variables(env_vars)

lambda_local/event.py

@@ -1,5 +1,5 @@
 '''
-Copyright 2015-2017 HDE, Inc.
+Copyright 2015-2018 HDE, Inc.
 Licensed under MIT.
 '''
 

lambda_local/main.py

@@ -1,5 +1,5 @@
 '''
-Copyright 2015-2017 HDE, Inc.
+Copyright 2015-2018 HDE, Inc.
 Licensed under MIT.
 '''
 
@@ -15,7 +15,7 @@
 
 from . import event
 from . import context
-from .environment_variables import set_environment_variables
+from .environment_variables import set_environment_variables, export_variables
 from .timeout import time_limit
 from .timeout import TimeoutException
 
@@ -31,6 +31,17 @@
 EXITCODE_ERR = 1
 
 
+def call(func, event, timeout, environment_variables={}, arn_string="", version_name="", library=None):
+    export_variables(environment_variables)
+    e = json.loads(event)
+    c = context.Context(timeout, arn_string, version_name)
+    if library is not None:
+        load_lib(library)
+    request_id = uuid.uuid4()
+
+    return _runner(request_id, e, c, func)
+
+
 def run(args):
     # set env vars if path to json file was given
     set_environment_variables(args.environment_variables)
@@ -41,16 +52,23 @@ def run(args):
         load_lib(args.library)
     request_id = uuid.uuid4()
     func = load(request_id, args.file, args.function)
+    
+    (result, err_type) = _runner(request_id, e, c, func)
+
+    if err_type is not None:
+        sys.exit(EXITCODE_ERR)
+
 
+def _runner(request_id, event, context, func):
     logger = logging.getLogger()
     result = None
 
-    logger.info("Event: {}".format(e))
+    logger.info("Event: {}".format(event))
 
     logger.info("START RequestId: {}".format(request_id))
 
     start_time = timeit.default_timer()
-    result, err_type = execute(func, e, c)
+    result, err_type = execute(func, event, context)
     end_time = timeit.default_timer()
 
     logger.info("END RequestId: {}".format(request_id))
@@ -64,8 +82,7 @@ def run(args):
     logger.info("REPORT RequestId: {}\tDuration: {}".format(
         request_id, duration))
 
-    if err_type is not None:
-        sys.exit(EXITCODE_ERR)
+    return (result, err_type)
 
 
 def load_lib(path):

lambda_local/timeout.py

@@ -1,5 +1,5 @@
 '''
-Copyright 2015-2017 HDE, Inc.
+Copyright 2015-2018 HDE, Inc.
 Licensed under MIT.
 '''