feature: upload tags

Author: FelixAbrahamsson

next_cvat/client/job_annotations.py

@@ -51,6 +51,22 @@ def add_polyline_(
 
         return self
 
+    def add_tag_(
+        self,
+        tag: next_cvat.Tag,
+        image_name: str,
+        group: int = 0,
+    ) -> JobAnnotations:
+        label = self.job.task.project.label(name=tag.label)
+
+        frame = self.job.task.frame(image_name=image_name)
+
+        self.annotations["tags"].append(
+            tag.request(frame=frame.id, label_id=label.id, group=group)
+        )
+
+        return self
+
     def request(self) -> models.LabeledDataRequest:
         request = models.LabeledDataRequest()
         request.version = self.annotations["version"]

next_cvat/types/tag.py

@@ -2,6 +2,7 @@
 
 from typing import List
 
+from cvat_sdk.api_client import models
 from pydantic import BaseModel
 
 from .attribute import Attribute
@@ -11,3 +12,14 @@ class Tag(BaseModel):
     label: str
     source: str
     attributes: List[Attribute]
+
+    def request(
+        self, frame: int, label_id: int, group: int = 0
+    ) -> models.LabeledImageRequest:
+        return models.LabeledImageRequest(
+            frame=frame,
+            label_id=label_id,
+            group=group,
+            source=self.source,
+            attributes=[attr.model_dump() for attr in self.attributes],
+        )

tests/test_add_tag.py

@@ -0,0 +1,40 @@
+from pathlib import Path
+
+import pytest
+
+import next_cvat
+
+
+def test_add_tag():
+    if not Path(".env.cvat.secrets").exists():
+        pytest.skip("No secrets file found")
+
+    client = next_cvat.Client.from_env_file(".env.cvat.secrets")
+
+    project_id = 198488  # Using the same test project as in other tests
+    job_id = 1442235
+    task_id = 999670
+    deformation_attribute_id = 1389238
+
+    job = client.project(project_id).task(task_id).job(job_id)
+
+    annotations = job.annotations()
+
+    annotations.add_tag_(
+        next_cvat.Tag(
+            label="Deformation",
+            source="manual",
+            attributes=[
+                next_cvat.Attribute(
+                    name="Severity",
+                    value="High",
+                    spec_id=deformation_attribute_id,
+                )
+            ],
+        ),
+        image_name="20240916_000854_2011T_437.bmp",
+    )
+
+    job.update_annotations_(annotations)
+
+    print("Successfully added tag annotation")