From d9f1798684f441b5de93c456898768985786aad5 Mon Sep 17 00:00:00 2001
From: Adam Harvey <adam@ahprojects.com>
Date: Tue, 20 Apr 2021 13:54:44 +0200
Subject: [PATCH] add --project_id argument for create task (#3090)

---
 utils/cli/core/core.py       | 4 ++++
 utils/cli/core/definition.py | 7 +++++++
 2 files changed, 11 insertions(+)

diff --git a/utils/cli/core/core.py b/utils/cli/core/core.py
index 57b39d1f..facd17d4 100644
--- a/utils/cli/core/core.py
+++ b/utils/cli/core/core.py
@@ -66,16 +66,20 @@ class CLI():
                      completion_verification_period=20,
                      git_completion_verification_period=2,
                      dataset_repository_url='',
+                     project_id=None,
                      lfs=False, **kwargs):
         """ Create a new task with the given name and labels JSON and
         add the files to it. """
         url = self.api.tasks
+        labels = [] if project_id is not None else labels
         data = {'name': name,
                 'labels': labels,
                 'overlap': overlap,
                 'segment_size': segment_size,
                 'bug_tracker': bug,
         }
+        if project_id:
+            data.update({'project_id': project_id})
         response = self.session.post(url, json=data)
         response.raise_for_status()
         response_json = response.json()
diff --git a/utils/cli/core/definition.py b/utils/cli/core/definition.py
index c15dbafe..48e11456 100644
--- a/utils/cli/core/definition.py
+++ b/utils/cli/core/definition.py
@@ -112,6 +112,12 @@ task_create_parser.add_argument(
     type=parse_label_arg,
     help='string or file containing JSON labels specification'
 )
+task_create_parser.add_argument(
+    '--project',
+    default=None,
+    type=int,
+    help='project ID if project exists'
+)
 task_create_parser.add_argument(
     '--overlap',
     default=0,
@@ -175,6 +181,7 @@ task_create_parser.add_argument(
     action='store_true',
     help='using lfs for dataset repository (default: %(default)s)'
 )
+
 #######################################################################
 # Delete
 #######################################################################