Add required flag on certain arguments to command-line tools

andrewsg · andrewsg · commit 3f92d71b8325 · 2017-09-14T16:21:48.000-07:00
diff --git a/appengine/flexible/tasks/create_app_engine_queue_task.py b/appengine/flexible/tasks/create_app_engine_queue_task.py
@@ -80,17 +80,20 @@ def create_task(project, queue, location, payload=None, in_seconds=None):
 
     parser.add_argument(
         '--project',
-        help='Project of the queue to add the task to.'
+        help='Project of the queue to add the task to.',
+        required=True,
     )
 
     parser.add_argument(
         '--queue',
-        help='ID (short name) of the queue to add the task to.'
+        help='ID (short name) of the queue to add the task to.',
+        required=True,
     )
 
     parser.add_argument(
         '--location',
-        help='Location of the queue to add the task to.'
+        help='Location of the queue to add the task to.',
+        required=True,
     )
 
     parser.add_argument(
diff --git a/tasks/pull_queue_snippets.py b/tasks/pull_queue_snippets.py
@@ -105,31 +105,37 @@ def acknowledge_task(task):
         help=create_task.__doc__)
     create_task_parser.add_argument(
         '--project',
-        help='Project of the queue to add the task to.'
+        help='Project of the queue to add the task to.',
+        required=True,
     )
     create_task_parser.add_argument(
         '--queue',
-        help='ID (short name) of the queue to add the task to.'
+        help='ID (short name) of the queue to add the task to.',
+        required=True,
     )
     create_task_parser.add_argument(
         '--location',
-        help='Location of the queue to add the task to.'
+        help='Location of the queue to add the task to.',
+        required=True,
     )
 
     pull_and_ack_parser = subparsers.add_parser(
         'pull-and-ack-task',
         help=create_task.__doc__)
     pull_and_ack_parser.add_argument(
         '--project',
-        help='Project of the queue to pull the task from.'
+        help='Project of the queue to pull the task from.',
+        required=True,
     )
     pull_and_ack_parser.add_argument(
         '--queue',
-        help='ID (short name) of the queue to pull the task from.'
+        help='ID (short name) of the queue to pull the task from.',
+        required=True,
     )
     pull_and_ack_parser.add_argument(
         '--location',
-        help='Location of the queue to pull the task from.'
+        help='Location of the queue to pull the task from.',
+        required=True,
     )
 
     args = parser.parse_args()
