use click features to check if argument is a file

cyriac · cyriac · commit 1b0ee0ec56d9 · 2016-10-24T22:50:04.000+05:30
diff --git a/scripts/lambda b/scripts/lambda
@@ -16,13 +16,11 @@ def cli():
 
 
 @click.command(help="Create a new function for Lambda.")
-@click.argument('folder', default=False)
+@click.argument('folder', nargs=-1, type=click.Path(file_okay=False, writable=True))
 def init(folder):
 path = CURRENT_DIR
- if folder:
- path = "{}/{}".format(CURRENT_DIR, folder)
- if os.path.isfile(path):
- raise click.ClickException("{} is a file".format(path))
+ if len(folder) > 0:
+ path = "{}/{}".format(CURRENT_DIR, folder[0])
 if not os.path.exists(path):
 os.makedirs(path)
 aws_lambda.init(path)

-Original file line number
+Diff line change
 @click.command(help="Create a new function for Lambda.")
 -@click.argument('folder', default=False)
 +@click.argument('folder', nargs=-1, type=click.Path(file_okay=False, writable=True))
 definit(folder):
 path=CURRENT_DIR
 -iffolder:
 -path="{}/{}".format(CURRENT_DIR, folder)
 -ifos.path.isfile(path):
 -raiseclick.ClickException("{} is a file".format(path))
 +iflen(folder) >0:
 +path="{}/{}".format(CURRENT_DIR, folder[0])
 ifnotos.path.exists(path):
 os.makedirs(path)
 aws_lambda.init(path)