label-studio-sdk 0.0.19__tar.gz → 0.0.21__tar.gz

{label-studio-sdk-0.0.19/label_studio_sdk.egg-info → label-studio-sdk-0.0.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: label-studio-sdk
-Version: 0.0.19
+Version: 0.0.21
 Summary: Label Studio annotation tool
 Home-page: https://github.com/heartexlabs/label-studio-sdk
 Author: Heartex

{label-studio-sdk-0.0.19 → label-studio-sdk-0.0.21}/label_studio_sdk/__init__.py

@@ -5,4 +5,4 @@ from .project import Project
 from .utils import parse_config
 
 
-__version__ = '0.0.19'
+__version__ = '0.0.21'

{label-studio-sdk-0.0.19 → label-studio-sdk-0.0.21}/label_studio_sdk/project.py

@@ -7,6 +7,7 @@ import logging
 from enum import Enum, auto
 from random import sample, shuffle
 from requests.exceptions import HTTPError, InvalidSchema, MissingSchema
+from requests import Response
 from pathlib import Path
 from typing import Optional, Union, List, Dict, Callable
 from .client import Client
@@ -1230,7 +1231,6 @@ class Project(Client):
             'presign_ttl': presign_ttl,
             'title': title,
             'description': description,
-            'project': self.id,
         }
         response = self.make_request('POST', '/api/storages/gcs', json=payload)
         return response.json()
@@ -1801,7 +1801,7 @@ class Project(Client):
             overlap=overlap,
         )
 
-    def export_snapshot_list(self):
+    def export_snapshot_list(self) -> list:
         """
         Get list of export snapshots for the current project
         -------
@@ -1835,7 +1835,7 @@ class Project(Client):
         annotation_filter_options_ground_truth: bool = True,
         annotation_filter_options_skipped: bool = True,
         interpolate_key_frames: bool = False,
-    ):
+    ) -> dict:
         """
         Create new export snapshot
         ----------
@@ -1904,7 +1904,7 @@ class Project(Client):
         )
         return response.json()
 
-    def export_snapshot_status(self, export_id: int):
+    def export_snapshot_status(self, export_id: int) -> ExportSnapshotStatus:
         """
         Get export snapshot status by Export ID
         ----------
@@ -1936,7 +1936,7 @@ class Project(Client):
 
     def export_snapshot_download(
         self, export_id: int, export_type: str = 'JSON', path: str = "."
-    ):
+    ) -> (int, str):
         """
         Download file with export snapshot in provided format
         ----------
@@ -1966,6 +1966,24 @@ class Project(Client):
                     f.write(chunk)
         return response.status_code, filename
 
+    def export_snapshot_delete(self, export_id: int) -> int:
+        """Delete an export file by specified export ID
+
+        Parameters
+        ----------
+        export_id: int
+            Existing Export ID from current project
+
+        Returns
+        ----------
+        Status code for operation
+        """
+        response = self.make_request(
+            'DELETE',
+            f'/api/projects/{self.id}/exports/{export_id}',
+        )
+        return response.status_code
+
     def get_files_from_tasks(self, tasks: Dict, get_tasks: bool = False):
         """Copy files from tasks to cache folder
 
@@ -1997,3 +2015,42 @@ class Project(Client):
                     except (FileNotFoundError, InvalidSchema, MissingSchema, IOError):
                         logger.debug(f"Couldn't copy file {task['data'][key]}.")
         return filenames
+
+    def delete_task(self, task_id: int) -> Response:
+        """Delete a task. To remove multiple tasks `use delete_tasks()`.
+        
+        Parameters
+        ----------
+        task_id: int
+            Task id.
+        """
+        assert isinstance(task_id, int), 'task_id should be int'
+        return self.make_request("DELETE", f"/api/tasks/{task_id}")
+
+    def delete_tasks(self, task_ids: list) -> Response:
+        """Delete multiple tasks by IDs.
+
+        Parameters
+        ----------
+        task_ids: list of int
+            Task ids.
+        """
+        assert isinstance(task_ids, list), 'task_ids should be list of int'
+        if not task_ids:  # avoid deletion of all tasks when task_ids = []
+            return Response()
+        payload = {"selectedItems": {"all": False, "included": task_ids}, "project": self.id}
+        return self.make_request("POST", f"/api/dm/actions?project={self.id}&id=delete_tasks", json=payload)
+
+    def delete_all_tasks(self, excluded_ids: list = None) -> Response:
+        """Delete all tasks from the project.
+
+        Parameters
+        ----------
+        excluded_ids: list of int
+            Task ids that should be excluded from the deletion.
+        """
+        assert isinstance(excluded_ids, list) or excluded_ids is None, 'excluded_ids should be list of int or None'
+        if excluded_ids is None:
+            excluded_ids = []
+        payload = {"selectedItems": {"all": True, "excluded": excluded_ids}, "project": self.id}
+        return self.make_request("POST", f"/api/dm/actions?project={self.id}&id=delete_tasks", json=payload)

{label-studio-sdk-0.0.19 → label-studio-sdk-0.0.21/label_studio_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: label-studio-sdk
-Version: 0.0.19
+Version: 0.0.21
 Summary: Label Studio annotation tool
 Home-page: https://github.com/heartexlabs/label-studio-sdk
 Author: Heartex