Ensure 'stage' argument sanity (aka avoid misleading 'divio project pull db WRONGNAMEHERE')

Author: filwaitman

divio_cli/cli.py

@@ -208,7 +208,7 @@ def sort_projects(items):
 @click.option(
     "--backup/--no-backup", default=None, help="Take a backup on deployment."
 )
-@click.argument("stage", default="test")
+@click.argument("stage", default="test", type=click.Choice(['test', 'live']))
 @allow_remote_id_override
 @click.pass_obj
 def project_deploy(obj, remote_id, stage, backup):
@@ -217,7 +217,7 @@ def project_deploy(obj, remote_id, stage, backup):
 
 
 @project.command(name="deploy-log")
-@click.argument("stage", default="test")
+@click.argument("stage", default="test", type=click.Choice(['test', 'live']))
 @allow_remote_id_override
 @click.pass_obj
 def project_deploy_log(obj, remote_id, stage):
@@ -418,7 +418,7 @@ def project_pull():
 
 
 @project_pull.command(name="db")
-@click.argument("stage", default="test")
+@click.argument("stage", default="test", type=click.Choice(['test', 'live']))
 @allow_remote_id_override
 @click.pass_obj
 def pull_db(obj, remote_id, stage):
@@ -432,7 +432,7 @@ def pull_db(obj, remote_id, stage):
 
 
 @project_pull.command(name="media")
-@click.argument("stage", default="test")
+@click.argument("stage", default="test", type=click.Choice(['test', 'live']))
 @allow_remote_id_override
 @click.pass_obj
 def pull_media(obj, remote_id, stage):
@@ -449,7 +449,7 @@ def project_push():
 
 
 @project_push.command(name="db")
-@click.argument("stage", default="test")
+@click.argument("stage", default="test", type=click.Choice(['test', 'live']))
 @click.option(
     "-d",
     "--dumpfile",
@@ -487,7 +487,7 @@ def push_db(obj, remote_id, stage, dumpfile, noinput):
 
 
 @project_push.command(name="media")
-@click.argument("stage", default="test")
+@click.argument("stage", default="test", type=click.Choice(['test', 'live']))
 @click.option(
     "--noinput", is_flag=True, default=False, help="Don't ask for confirmation"
 )