added verbose switch

nficano · nficano · commit b3ed0360ad26 · 2016-04-04T20:05:04.000-04:00
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -51,14 +51,16 @@ def deploy(src):
         create_function(cfg, path_to_zip_file)
 
 
-def invoke(src, alt_event=None):
+def invoke(src, alt_event=None, verbose=False):
     """Simulates a call to your function.
 
     :param str src:
         The path to your Lambda ready project (folder must contain a valid
         config.yaml and handler module (e.g.: service.py).
     :param str alt_event:
         An optional argument to override which event file to use.
+    :param bool verbose:
+        Whether to print out verbose details.
     """
     # Load and parse the config file.
     path_to_config_file = os.path.join(src, 'config.yaml')
@@ -78,15 +80,15 @@ def invoke(src, alt_event=None):
 
     # TODO: look into mocking the ``context`` variable, currently being passed
     # as None.
+
     start = time.time()
     results = fn(event, None)
     end = time.time()
-    elapsed = end - start
-    timeout = cfg.get('timeout', 15)
 
     print("{0}".format(results))
-    print("\nexecution time: {:.8f}s\nfunction execution "
-          "timeout: {:2}s".format(elapsed, timeout))
+    if verbose:
+        print("\nexecution time: {:.8f}s\nfunction execution "
+              "timeout: {:2}s".format(end - start, cfg.get('timeout', 15)))
 
 
 def init(src, minimal=False):
diff --git a/scripts/lambda b/scripts/lambda
@@ -21,9 +21,10 @@ def init():
 
 
 @click.command(help="Run a local test of your function.")
-@click.option('--event-file', default=None, help='Alternate event file.',)
-def invoke(event_file):
-    aws_lambda.invoke(CURRENT_DIR, event_file)
+@click.option('--event-file', default=None, help='Alternate event file.')
+@click.option('--verbose', '-v', is_flag=True)
+def invoke(event_file, verbose):
+    aws_lambda.invoke(CURRENT_DIR, event_file, verbose)
 
 
 @click.command(help="Register and deploy your code to lambda.")
