label-studio-sdk 0.0.18__tar.gz → 0.0.20__tar.gz

{label-studio-sdk-0.0.18/label_studio_sdk.egg-info → label-studio-sdk-0.0.20}/PKG-INFO

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

{label-studio-sdk-0.0.18 → label-studio-sdk-0.0.20}/label_studio_sdk/__init__.py

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

{label-studio-sdk-0.0.18 → label-studio-sdk-0.0.20}/label_studio_sdk/project.py

@@ -6,12 +6,17 @@ import logging
 
 from enum import Enum, auto
 from random import sample, shuffle
-from requests.exceptions import HTTPError
+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
 from .utils import parse_config, chunk
 
+from label_studio_tools.core.utils.io import get_local_path
+from label_studio_tools.core.label_config import parse_config
+
+
 logger = logging.getLogger(__name__)
 
 
@@ -1226,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()
@@ -1797,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
         -------
@@ -1831,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
         ----------
@@ -1900,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
         ----------
@@ -1932,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
         ----------
@@ -1961,3 +1965,90 @@ class Project(Client):
                 for chunk in response:
                     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
+
+        Parameters
+        ----------
+        tasks: Dict
+        Tasks to download to local storage
+        get_tasks: bool
+        Get all tasks from current project
+
+        Returns
+        -------
+        list
+            List of filenames
+        """
+        if get_tasks:
+            tasks = self.get_tasks()
+        filenames = []
+        if tasks:
+            for task in tasks:
+                for key in task['data']:
+                    try:
+                        filename = get_local_path(
+                            task['data'][key],
+                            access_token=self.api_key,
+                            hostname=self.url,
+                        )
+                        filenames.append(filename)
+                    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'
+        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.18 → label-studio-sdk-0.0.20/label_studio_sdk.egg-info}/PKG-INFO

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

label-studio-sdk-0.0.20/label_studio_sdk.egg-info/requires.txt

@@ -0,0 +1,4 @@
+lxml>=4.2.5
+requests<3,>=2.22.0
+pydantic<=1.11,>1.7
+label-studio-tools>=0.0.1

label-studio-sdk-0.0.20/requirements.txt

@@ -0,0 +1,4 @@
+lxml>=4.2.5
+requests>=2.22.0,<3
+pydantic<=1.11,>1.7
+label-studio-tools>=0.0.1

label-studio-sdk-0.0.18/label_studio_sdk.egg-info/requires.txt

@@ -1,3 +0,0 @@
-lxml>=4.2.5
-requests<3,>=2.22.0
-pydantic~=1.10

label-studio-sdk-0.0.18/requirements.txt

@@ -1,3 +0,0 @@
-lxml>=4.2.5
-requests>=2.22.0,<3
-pydantic~=1.10