Add resourceTag GET and projectResourceTag CREATE for project tagging

Author: manibox

labelbox/__init__.py

@@ -23,3 +23,5 @@
 from labelbox.schema.model_run import ModelRun
 from labelbox.schema.benchmark import Benchmark
 from labelbox.schema.iam_integration import IAMIntegration
+from labelbox.schema.resource_tag import ResourceTag
+from labelbox.schema.project_resource_tag import ProjectResourceTag

labelbox/schema/organization.py

@@ -1,3 +1,4 @@
+import json
 from typing import TYPE_CHECKING, List, Optional
 
 from labelbox.exceptions import LabelboxError
@@ -126,6 +127,24 @@ def remove_user(self, user: "User") -> None:
             updateUser(where: {id: $%s}, data: {deleted: true}) { id deleted }
         }""" % (user_id_param, user_id_param), {user_id_param: user.uid})
 
+    def get_resource_tags(self) -> List["ResourceTag"]:
+        """
+        Returns all resource tags for an organization
+        """
+        res = self.client.execute(
+            """query {
+                organization {
+                    resourceTag {
+                        id,
+                        text,
+                        color
+                    }
+                }
+            }""")
+
+        # print(json.dumps(res['organization']['resourceTag'], indent=2, sort_keys=True))
+        return res['organization']['resourceTag']
+
     def get_iam_integrations(self) -> List["IAMIntegration"]:
         """
         Returns all IAM Integrations for an organization

labelbox/schema/project.py

@@ -121,6 +121,33 @@ def members(self) -> PaginatedCollection:
                                    {id_param: str(self.uid)},
                                    ["project", "members"], ProjectMember)
 
+    def create_project_resource_tags(self, tagIdsParam) -> List["ProjectResourceTag"]:
+        """ Creates a project resource tag
+
+        TODO
+        """
+        project_id_param = "projectId"
+        tag_ids_param = "resourceTagIds"
+
+        query_str = """mutation AttatchProjectResourceTags($%s: ID!, $%s: [String!]) {
+            project(where:{id:$%s}){updateProjectResourceTags(input:{%s:$%s}){id}}
+        }""" % (
+            project_id_param,
+            tag_ids_param,
+            project_id_param,
+            tag_ids_param,
+            tag_ids_param
+        )
+
+        print('tagIdsParam', tagIdsParam)
+        res = self.client.execute(
+            query_str, {
+                project_id_param: self.uid,
+                tag_ids_param: tagIdsParam
+            })
+
+        return res["project"]["updateProjectResourceTags"]
+
     def labels(self, datasets=None, order_by=None) -> PaginatedCollection:
         """ Custom relationship expansion method to support limited filtering.
 

labelbox/schema/project_resource_tag.py

@@ -0,0 +1,15 @@
+from labelbox.orm.db_object import DbObject, Updateable
+from labelbox.orm.model import Field, Relationship
+
+class ProjectResourceTag(DbObject, Updateable):
+    """ Project resource tag to associate ProjectResourceTag to Project.
+
+    Attributes:
+        resourceTagId (str)
+        projectId (str)
+
+        resource_tag (Relationship): `ToOne` relationship to ResourceTag
+    """
+
+    resourceTagId = Field.ID("resourceTagId")
+    projectId = Field.ID("projectId")

labelbox/schema/resource_tag.py

@@ -0,0 +1,15 @@
+from labelbox.orm.db_object import DbObject
+from labelbox.orm.model import Field, Relationship
+
+class ResourceTag(DbObject):
+    """ Resource tag to label and identify your labelbox resources easier.
+
+    Attributes:
+        text (str)
+        color (str)
+
+        project_resource_tag (Relationship): `ToMany` relationship to ProjectResourceTag
+    """
+
+    text = Field.String("text")
+    color = Field.String("color")