Merge pull request nficano#18 from cyriac/master

nficano · web-flow · commit 76ee21c2af21 · 2016-12-18T02:09:13.000-05:00
lambda init accepts optional folder name
diff --git a/scripts/lambda b/scripts/lambda
@@ -16,8 +16,14 @@ def cli():
 
 
 @click.command(help="Create a new function for Lambda.")
-def init():
-    aws_lambda.init(CURRENT_DIR)
+@click.argument('folder', nargs=-1, type=click.Path(file_okay=False, writable=True))
+def init(folder):
+    path = CURRENT_DIR
+    if len(folder) > 0:
+        path = "{}/{}".format(CURRENT_DIR, folder[0])
+        if not os.path.exists(path):
+            os.makedirs(path)
+    aws_lambda.init(path)
 
 
 @click.command(help="Bundles package for deployment.")
