label-studio-sdk 0.0.32__py3-none-any.whl → 0.0.34__py3-none-any.whl

label_studio_sdk/schema/label_config_schema.json

@@ -0,0 +1,226 @@
+{
+  "$schema": "http://json-schema.org/draft-06/schema#",
+  "title": "LabelConfig",
+  "definitions": {
+    "$": {
+      "type": "string"
+    },
+    "@name": {
+      "type": "string"
+    },
+    "@toName": {
+      "type": "string"
+    },
+    "@value": {
+      "anyOf": [
+        {"type": "string"},
+        {"type": "boolean"},
+        {"type": "number"}
+      ]},
+    "@valueList": {
+      "anyOf": [
+        {"type": "string"}
+      ]},
+    "tag_with_value": {
+      "type": "object",
+      "required": [
+        "@value"
+      ],
+      "properties": {
+        "@value": {
+          "$ref": "#/definitions/@value"
+        },
+        "$": {
+          "$ref": "#/definitions/$"
+        }
+      }
+    },
+    "tag_with_value_required_name": {
+      "type": "object",
+      "oneOf": [
+          {
+            "required": [
+                "@name",
+                "@value"
+            ]
+          },
+          {
+            "required": [
+                "@name",
+                "@valueList"
+            ]
+          }
+      ],
+      "properties": {
+        "@value": {
+          "$ref": "#/definitions/@value"
+        },
+        "$": {
+          "$ref": "#/definitions/$"
+        }
+      }
+    },
+    "tags_with_value": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/tag_with_value"}
+      }, {"$ref": "#/definitions/tag_with_value"}]
+    },
+    "tags_with_value_or_object": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/tag_with_value"}
+      }, {
+        "type": "object"
+      }, {"$ref": "#/definitions/tag_with_value"}]
+    },
+    "tags_with_value_required_name": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/tag_with_value_required_name"}
+      }, {"$ref": "#/definitions/tag_with_value_required_name"}]
+    },
+    "View": {
+      "type": "object",
+      "additionalProperties": true,
+      "properties": {
+        "Labels": {"$ref": "#/definitions/MaybeMultipleLabels"},
+        "Choices": {"$ref": "#/definitions/MaybeMultipleChoices"},
+        "Label": {"$ref": "#/definitions/MaybeMultipleLabel"},
+        "Choice": {"$ref": "#/definitions/MaybeMultipleChoice"},
+        "Image": {"$ref": "#/definitions/tags_with_value_required_name"},
+        "Text": {"$ref": "#/definitions/tags_with_value_required_name"},
+        "HyperText": {"$ref": "#/definitions/tags_with_value_required_name"},
+        "View": {"$ref": "#/definitions/MaybeMultipleView"},
+        "TextArea": {"$ref": "#/definitions/MaybeMultipleTextAreas"}
+      }
+    },
+    "MaybeMultipleView": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/View"}
+      }, {"$ref": "#/definitions/View"}]
+    },
+    "Choice": {
+      "$ref": "#/definitions/tag_with_value"
+    },
+    "MaybeMultipleChoice": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/Choice"}
+      }, {"$ref": "#/definitions/Choice"}]
+    },
+    "Label": {
+      "$ref": "#/definitions/tag_with_value"
+    },
+    "MaybeMultipleLabel": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/Label"}
+      }, {"$ref": "#/definitions/Label"}]
+    },
+    "Choices": {
+      "type": "object",
+      "anyOf": [
+        {"required": ["@name", "@toName", "Choice"]},
+        {"required": ["@name", "@toName", "@value"]},
+        {"required": ["@name", "@toName", "View"]}
+      ],
+      "properties": {
+        "@name": {
+          "$ref": "#/definitions/@name"
+        },
+        "@toName": {
+          "$ref": "#/definitions/@toName"
+        },
+        "@value": {
+          "$ref": "#/definitions/@value"
+        },
+        "$": {
+          "$ref": "#/definitions/$"
+        },
+        "Choice": {
+          "type": "array",
+          "items": {"$ref": "#/definitions/Choice"}
+        },
+        "View": {"$ref": "#/definitions/MaybeMultipleView"}
+      }
+    },
+    "MaybeMultipleChoices": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/Choices"}
+      }, {"$ref": "#/definitions/Choices"}]
+    },
+    "Labels": {
+      "type": "object",
+      "anyOf": [
+        {"required": ["@name", "@toName", "Label"]},
+        {"required": ["@name", "@toName", "@value"]},
+        {"required": ["@name", "@toName", "View"]}
+      ],
+      "properties": {
+        "@name": {
+          "$ref": "#/definitions/@name"
+        },
+        "@toName": {
+          "$ref": "#/definitions/@toName"
+        },
+        "@value": {
+          "$ref": "#/definitions/@value"
+        },
+        "$": {
+          "$ref": "#/definitions/$"
+        },
+        "Label": {"$ref": "#/definitions/MaybeMultipleLabel"},
+        "View": {"$ref": "#/definitions/MaybeMultipleView"}
+      }
+    },
+    "MaybeMultipleLabels": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/Labels"}
+      }, {"$ref": "#/definitions/Labels"}]
+    },
+    "TextArea": {
+      "type": "object",
+      "anyOf": [
+        {"required": ["@name", "@toName"]}
+      ],
+      "properties": {
+        "@name": {
+          "$ref": "#/definitions/@name"
+        },
+        "@toName": {
+          "$ref": "#/definitions/@toName"
+        },
+        "$": {
+          "$ref": "#/definitions/$"
+        }
+      }
+    },
+    "MaybeMultipleTextAreas": {
+      "anyOf": [{
+        "type": "array",
+        "items": {"$ref": "#/definitions/TextArea"}
+      }, {"$ref": "#/definitions/TextArea"}]
+    }
+    
+  },
+  "type": "object",
+  "properties": {
+    "View": {
+      "type": "object",
+      "additionalProperties": true,
+      "properties": {
+        "Choices": {"$ref": "#/definitions/MaybeMultipleChoices"},
+        "Labels": {"$ref": "#/definitions/MaybeMultipleLabels"},
+        "View": {"$ref": "#/definitions/MaybeMultipleView"},
+        "Image": {"$ref": "#/definitions/tags_with_value_required_name"},
+        "Text": {"$ref": "#/definitions/tags_with_value_required_name"},
+        "HyperText": {"$ref": "#/definitions/tags_with_value_required_name"},
+        "TextArea": {"$ref": "#/definitions/MaybeMultipleTextAreas"}
+      }
+    }
+  }
+}

label_studio_sdk/users.py

@@ -1,18 +1,20 @@
 from datetime import datetime
-from pydantic import BaseModel
 from enum import Enum
 from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
 from .client import Client
 
 
 class UserRole(Enum):
-    ANNOTATOR = 'AN'
-    REVIEWER = 'RE'
-    MANAGER = 'MA'
-    ADMINISTRATOR = 'AD'
-    OWNER = 'OW'
-    NOT_ACTIVATED = 'NO'
-    DISABLED = 'DI'
+    ANNOTATOR = "AN"
+    REVIEWER = "RE"
+    MANAGER = "MA"
+    ADMINISTRATOR = "AD"
+    OWNER = "OW"
+    NOT_ACTIVATED = "NO"
+    DISABLED = "DI"
 
 
 class OrgMembership(BaseModel):
@@ -30,8 +32,8 @@ class User(BaseModel):
     last_activity: datetime
     initials: str
     phone: str
-    active_organization: Optional[int]
-    org_membership: Optional[List[OrgMembership]]
+    active_organization: Optional[int] = None
+    org_membership: Optional[List[OrgMembership]] = Field(default_factory=list)
     client: Client
 
     class Config:
@@ -46,9 +48,9 @@ class User(BaseModel):
             User role
         """
         response = self.client.make_request(
-            'PATCH',
-            f'/api/organizations/{self.active_organization}/memberships',
-            json={'user_id': self.id, 'role': role.value},
+            "PATCH",
+            f"/api/organizations/{self.active_organization}/memberships",
+            json={"user_id": self.id, "role": role.value},
         )
         for membership in self.org_membership:
             if membership.organization_id == self.active_organization:

label_studio_sdk/utils.py

@@ -1,5 +1,6 @@
 """ .. include::../docs/utils.md
 """
+
 import logging
 
 from lxml import etree
@@ -7,9 +8,9 @@ from collections import defaultdict
 
 logger = logging.getLogger(__name__)
 
-_LABEL_TAGS = {'Label', 'Choice'}
+_LABEL_TAGS = {"Label", "Choice"}
 _NOT_CONTROL_TAGS = {
-    'Filter',
+    "Filter",
 }
 
 
@@ -49,12 +50,12 @@ def parse_config(config_string):
         return {}
 
     def _is_input_tag(tag):
-        return tag.attrib.get('name') and tag.attrib.get('value')
+        return tag.attrib.get("name") and tag.attrib.get("value")
 
     def _is_output_tag(tag):
         return (
-            tag.attrib.get('name')
-            and tag.attrib.get('toName')
+            tag.attrib.get("name")
+            and tag.attrib.get("toName")
             and tag.tag not in _NOT_CONTROL_TAGS
         )
 
@@ -65,7 +66,7 @@ def parse_config(config_string):
             parent = parent.getparent()
             if parent is None:
                 return
-            name = parent.attrib.get('name')
+            name = parent.attrib.get("name")
             if name in outputs:
                 return name
 
@@ -74,55 +75,55 @@ def parse_config(config_string):
     inputs, outputs, labels = {}, {}, defaultdict(dict)
     for tag in xml_tree.iter():
         if _is_output_tag(tag):
-            tag_info = {'type': tag.tag, 'to_name': tag.attrib['toName'].split(',')}
+            tag_info = {"type": tag.tag, "to_name": tag.attrib["toName"].split(",")}
             # Grab conditionals if any
             conditionals = {}
-            if tag.attrib.get('perRegion') == 'true':
-                if tag.attrib.get('whenTagName'):
-                    conditionals = {'type': 'tag', 'name': tag.attrib['whenTagName']}
-                elif tag.attrib.get('whenLabelValue'):
+            if tag.attrib.get("perRegion") == "true":
+                if tag.attrib.get("whenTagName"):
+                    conditionals = {"type": "tag", "name": tag.attrib["whenTagName"]}
+                elif tag.attrib.get("whenLabelValue"):
                     conditionals = {
-                        'type': 'label',
-                        'name': tag.attrib['whenLabelValue'],
+                        "type": "label",
+                        "name": tag.attrib["whenLabelValue"],
                     }
-                elif tag.attrib.get('whenChoiceValue'):
+                elif tag.attrib.get("whenChoiceValue"):
                     conditionals = {
-                        'type': 'choice',
-                        'name': tag.attrib['whenChoiceValue'],
+                        "type": "choice",
+                        "name": tag.attrib["whenChoiceValue"],
                     }
             if conditionals:
-                tag_info['conditionals'] = conditionals
-            outputs[tag.attrib['name']] = tag_info
+                tag_info["conditionals"] = conditionals
+            outputs[tag.attrib["name"]] = tag_info
         elif _is_input_tag(tag):
-            inputs[tag.attrib['name']] = {
-                'type': tag.tag,
-                'value': tag.attrib['value'].lstrip('$'),
+            inputs[tag.attrib["name"]] = {
+                "type": tag.tag,
+                "value": tag.attrib["value"].lstrip("$"),
             }
         if tag.tag not in _LABEL_TAGS:
             continue
         parent_name = _get_parent_output_tag_name(tag, outputs)
         if parent_name is not None:
-            actual_value = tag.attrib.get('alias') or tag.attrib.get('value')
+            actual_value = tag.attrib.get("alias") or tag.attrib.get("value")
             if not actual_value:
                 logger.debug(
                     'Inspecting tag {tag_name}... found no "value" or "alias" attributes.'.format(
-                        tag_name=etree.tostring(tag, encoding='unicode').strip()[:50]
+                        tag_name=etree.tostring(tag, encoding="unicode").strip()[:50]
                     )
                 )
             else:
                 labels[parent_name][actual_value] = dict(tag.attrib)
     for output_tag, tag_info in outputs.items():
-        tag_info['inputs'] = []
-        for input_tag_name in tag_info['to_name']:
+        tag_info["inputs"] = []
+        for input_tag_name in tag_info["to_name"]:
             if input_tag_name not in inputs:
                 logger.warning(
-                    f'to_name={input_tag_name} is specified for output tag name={output_tag}, '
-                    'but we can\'t find it among input tags'
+                    f"to_name={input_tag_name} is specified for output tag name={output_tag}, "
+                    "but we can't find it among input tags"
                 )
                 continue
-            tag_info['inputs'].append(inputs[input_tag_name])
-        tag_info['labels'] = list(labels[output_tag])
-        tag_info['labels_attrs'] = labels[output_tag]
+            tag_info["inputs"].append(inputs[input_tag_name])
+        tag_info["labels"] = list(labels[output_tag])
+        tag_info["labels_attrs"] = labels[output_tag]
     return outputs
 
 

label_studio_sdk/workspaces.py

@@ -1,13 +1,15 @@
+from typing import Optional
+
 from pydantic import BaseModel
-from typing import List, Optional
-from .users import User
+
 from .client import Client
+from .users import User
 
 
 class Workspace(BaseModel):
     id: int
     title: str
-    description: Optional[str]
+    description: Optional[str] = ''
     color: str
     is_personal: bool
     created_by: int
@@ -25,9 +27,9 @@ class Workspace(BaseModel):
             User
         """
         response = self.client.make_request(
-            'POST',
-            f'/api/workspaces/{self.id}/memberships',
-            json={'workspace': self.id, 'user': user.id},
+            "POST",
+            f"/api/workspaces/{self.id}/memberships",
+            json={"workspace": self.id, "user": user.id},
         )
         return response.json()
 
@@ -40,9 +42,9 @@ class Workspace(BaseModel):
             User
         """
         response = self.client.make_request(
-            'DELETE',
-            f'/api/workspaces/{self.id}/memberships',
-            json={'workspace': self.id, 'user': user.id},
+            "DELETE",
+            f"/api/workspaces/{self.id}/memberships",
+            json={"workspace": self.id, "user": user.id},
         )
         if response.status_code != 204:
             raise ValueError(str(response.content))
@@ -59,11 +61,11 @@ class Workspace(BaseModel):
 
         final_results = []
         response = self.client.make_request(
-            'GET', f'/api/workspaces/{self.id}/projects'
+            "GET", f"/api/workspaces/{self.id}/projects"
         )
         projects = response.json()
         for project_data in projects:
-            project_id = project_data['id']
+            project_id = project_data["id"]
             final_results.append(
                 Project.get_from_id(
                     client=self.client,

{label_studio_sdk-0.0.32.dist-info → label_studio_sdk-0.0.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: label-studio-sdk
-Version: 0.0.32
+Version: 0.0.34
 Summary: Label Studio annotation tool
 Home-page: https://github.com/heartexlabs/label-studio-sdk
 Author: Heartex
@@ -16,6 +16,8 @@ License-File: LICENSE
 Requires-Dist: lxml >=4.2.5
 Requires-Dist: requests <3,>=2.22.0
 Requires-Dist: pydantic <3,>1.7
+Requires-Dist: jsonschema >=3.2.0
+Requires-Dist: xmljson >=0.2.0
 Requires-Dist: label-studio-tools >=0.0.1
 
 Label Studio Python SDK

label_studio_sdk-0.0.34.dist-info/RECORD

@@ -0,0 +1,37 @@
+label_studio_sdk/__init__.py,sha256=N8tl9SDuUXJTqwCfMX6ytuanIMMO7hPAAWOv_8LDPcc,190
+label_studio_sdk/client.py,sha256=7lZeuQNw39C_Iqb6O-LSY8dOYbrE7hhA8qrc0SH3Qw0,14052
+label_studio_sdk/data_manager.py,sha256=IzmX9qokkVE51hrtX9IuXkLr80pVWSXq8Hp51-xDuHI,8309
+label_studio_sdk/exceptions.py,sha256=Xpx6phdRD_dTWlA-nq0v0Feta0RVDltPUjRXSiXJ0Yk,196
+label_studio_sdk/objects.py,sha256=fn0w3i7V-Y4SEP6v5bpz4vztCgzEW0kCInZBVP5-rCY,879
+label_studio_sdk/project.py,sha256=dlVnwciSkXZ_0JAW3PK3gDmbcqLKemy6qDx6fufGA6w,93509
+label_studio_sdk/users.py,sha256=G5wxBf7sGpZlN_0sR2NDovESIw1v2RfXWveCLHTgvN4,1402
+label_studio_sdk/utils.py,sha256=ZeBEwUmHyK_GdUjAEOdIqkbKRT7SviO4Db0Uj6tMNfI,4518
+label_studio_sdk/workspaces.py,sha256=9SrmZyWqj_0etCyTWzeC6v30jAIDnjerx4cNBlP3MGg,1913
+label_studio_sdk/label_interface/__init__.py,sha256=Eg6y3mAaYdKzJ5ZPhU_BTX2qoNPafthdxLD2I_rmXoU,38
+label_studio_sdk/label_interface/base.py,sha256=bI8pifudG4vI9MfvP-HIdHNJYDqg1pQimLQJGPUwnZk,2501
+label_studio_sdk/label_interface/control_tags.py,sha256=Cq9MTgtd23AZBx9zT_iXeV6905tzGL5_jOGZaBw7lE0,20247
+label_studio_sdk/label_interface/interface.py,sha256=dHpmHJrODkQ1q6g3Z8EAr_12cQcmdQtF5SsB7X1uglY,33133
+label_studio_sdk/label_interface/label_tags.py,sha256=nWEo21Gd8IPzIO72UqraLrChIbvrSMCA_eEhzYGnGCc,2282
+label_studio_sdk/label_interface/object_tags.py,sha256=CKNLYemwlvQ98NWUp4Nes0jklO28kE853KF-SML2f8Q,8181
+label_studio_sdk/label_interface/region.py,sha256=rlJDOVAoFtk4gcg1y2L_Pd8Rjl2cwNGuUUGsqf2YDbg,823
+label_studio_sdk/schema/label_config_schema.json,sha256=TIUgx4ft9lv4Yhj0ANjLy2YoChhPI7NyT3rH8rYFLA8,6101
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/test_client.py,sha256=5s_g5AO6rLk9mPynfO2awjA_rtOxUDEUBGlPW_q4VCQ,1107
+tests/test_export.py,sha256=I98dOIov7TOByiQFE6EEE9O2MVxsbbcct3BMlYfnE9s,3196
+tests/test_interface/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+tests/test_interface/configs.py,sha256=xDreW0adFRg_qAxto1VEwhVrH-k6ptfmpQVHM90GV8s,3275
+tests/test_interface/mockups.py,sha256=3IWXc9t5c9scDlqgeGmGqykfigm8w5b0Fa61wI8y2pU,1395
+tests/test_interface/test_compat.py,sha256=-lw-I2sJhSFT5OwuB2jkp-xxkyQ1GDOijgx6qQHCglo,1924
+tests/test_interface/test_control_tags.py,sha256=V4rVD6TDMeylo6NCmB7EywFS82zNJhqiXp59rJ5Sac0,1518
+tests/test_interface/test_data_generation.py,sha256=PdnQgP5Q0iYB7CiGQXlWxCjclEYud2Uiyre0fZdqkfM,1213
+tests/test_interface/test_lpi.py,sha256=8U95E6MBbPTOPqI7BJ0pWbWWIFMoFrDkE_6dcnXky30,552
+tests/test_interface/test_main.py,sha256=Hswmu7qI01PBvFVAiweq4DmmE1wJcQJ32-lQZoykArg,4650
+tests/test_interface/test_object_tags.py,sha256=K4fAZ1rNF_O1oeFOLEkNMn0MmTSG1GExT7yS3DKlx9Q,1008
+tests/test_interface/test_region.py,sha256=6FP2SFhXBUC1DbjSGwfUfoH2NVpfytibaC3qAw7nYAU,833
+tests/test_interface/test_validate_summary.py,sha256=mp0sPuJ389qV1TyKrWXbpTY9wxuo9Ye9mbrawxN5sVk,1059
+tests/test_interface/test_validation.py,sha256=dK3TqAIhrSCzLIeThs5aN9aXkrxN74BMLv8VE1SAi8M,1544
+label_studio_sdk-0.0.34.dist-info/LICENSE,sha256=lyszEV775P2PLmRYrU2TOQZ79PS2IPIiTyQEXU18IvA,11342
+label_studio_sdk-0.0.34.dist-info/METADATA,sha256=RkSqDolATfhDW-8EgYq-RncwgwFeReNvRD7eZ0UFJ8I,712
+label_studio_sdk-0.0.34.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+label_studio_sdk-0.0.34.dist-info/top_level.txt,sha256=0q87AjeX7fjpCV6nwmyUVFTNqO0mRqC5BqfwxzdPeUs,23
+label_studio_sdk-0.0.34.dist-info/RECORD,,

{label_studio_sdk-0.0.32.dist-info → label_studio_sdk-0.0.34.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.41.2)
+Generator: bdist_wheel (0.42.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{label_studio_sdk-0.0.32.dist-info → label_studio_sdk-0.0.34.dist-info}/top_level.txt

@@ -1,3 +1,2 @@
-docs
 label_studio_sdk
 tests

tests/test_client.py

@@ -1,26 +1,37 @@
-from unittest.mock import patch, Mock
+from unittest.mock import Mock
 
 from label_studio_sdk.client import Client
 
 
 def test_client_headers():
-    mock_session = Mock(spec=['request'])
+    mock_session = Mock(spec=["request"])
     mock_session.request.return_value = Mock(status_code=200)
-    client = Client(url='http://fake.url', api_key='fake_key',
-                    session=mock_session, versions={'label-studio': '1.0.0'},
-                    extra_headers={'Proxy-Authorization': 'Bearer fake_bearer'})
+    client = Client(
+        url="http://fake.url",
+        api_key="fake_key",
+        session=mock_session,
+        versions={"label-studio": "1.0.0"},
+        extra_headers={"Proxy-Authorization": "Bearer fake_bearer"},
+    )
 
     client.check_connection()
     args, kwargs = mock_session.request.call_args
-    assert kwargs['headers'] == {'Authorization': f'Token fake_key', 'Proxy-Authorization': 'Bearer fake_bearer'}
+    assert kwargs["headers"] == {
+        "Authorization": f"Token fake_key",
+        "Proxy-Authorization": "Bearer fake_bearer",
+    }
 
 
 def test_client_no_extra_headers():
-    mock_session = Mock(spec=['request'])
+    mock_session = Mock(spec=["request"])
     mock_session.request.return_value = Mock(status_code=200)
-    client = Client(url='http://fake.url', api_key='fake_key',
-                    session=mock_session, versions={'label-studio': '1.0.0'})
+    client = Client(
+        url="http://fake.url",
+        api_key="fake_key",
+        session=mock_session,
+        versions={"label-studio": "1.0.0"},
+    )
 
     client.check_connection()
     args, kwargs = mock_session.request.call_args
-    assert kwargs['headers'] == {'Authorization': f'Token fake_key'}
+    assert kwargs["headers"] == {"Authorization": f"Token fake_key"}