cleaning up

Author: zyd14

README.md

@@ -3,9 +3,12 @@ A tool for distributing processes across AWS Lambda functions with built-in resu
 
 ## Usage  
 ```python
-# Example lambda function
+# Example lambda function (should be deployed to AWS)
 # filename: example_handler.py
 # Lambda function name: example_lambda
+from time import sleep
+from random import randint
+
 from parallelagram.remote_handler import remote_handler
 
 
@@ -25,6 +28,8 @@ def execute(max_time: int):
 # Calling code to asynchronously invoke the lambda defined above and retrieve its results
 
 from parallelagram.launchables import Lambdable, ResponseCollector
+
+
 def main():
     lambda_list = [Lambdable(func_path='example_handler.execute',
                              remote_aws_lambda_func_name='example_lambda',

app.py

@@ -2,7 +2,7 @@
 
 from aws_cdk import core
 
-from parallelagram.parallelagram_stack import LambdaStack
+from stack.stack import LambdaStack
 
 
 app = core.App()

example_lambda/example_handler.py

@@ -1,9 +1,15 @@
 from random import randint
 from time import sleep
 
+from parallelagram.remote_handler import remote_handler
 
+
+@remote_handler
 def handler(event, context):
-    sleep_time = randint(1, 10)
+    pass
+
+def execute(max_time: int):
+    sleep_time = randint(1, max_time)
     print(f'sleeping for {sleep_time} seconds')
     sleep(sleep_time)
     return {'message': f'slept for {sleep_time} seconds'}

parallelagram/remote_handler.py

@@ -91,18 +91,6 @@ def remote_handler(func):
 
             If a message key is not present in the event then the decorated function handler will be invoked with the raw
             event and context arguments as a handler usually would be.
-
-            Because zappa dynamically generates its handler function on deployment, it is currently not possible to use this
-            wrapper on zappa-deployed functions.  Users who deploy their lambdas via zappa should use the @task (original
-            zappa wrapper) or @remote_runner function wrappers instead, which should both still play nice with the manager
-            functionalities provided in remote_manager.
-
-            Example for a non-Zappa deployed function:
-
-    get_request_from_s3: bool = False,
-                     request_s3_bucket: str = '',
-                     request_s3_key: str = '',
-                     response_to_s3: str = ''
     """
 
     @wraps(func)

stack/parallelagram_stack.py stack/stack.pystack/parallelagram_stack.py renamed to stack/stack.py

@@ -3,7 +3,7 @@
 import pytest
 
 from aws_cdk import core
-from stack.parallelagram_stack import LambdaStack
+from stack.stack import LambdaStack
 
 
 def get_template():