kalbio 0.2.2__tar.gz → 0.2.3__tar.gz

{kalbio-0.2.2 → kalbio-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kalbio
-Version: 0.2.2
+Version: 0.2.3
 Summary: Python client for kaleidoscope.bio
 License: MIT
 Author: Ahmed Elnaiem

{kalbio-0.2.2 → kalbio-0.2.3}/kalbio/activities.py

@@ -65,7 +65,7 @@ from kalbio.entity_fields import DataFieldTypeEnum
 from kalbio.programs import Program
 from kalbio.labels import Label
 from kalbio.workspace import WorkspaceUser, WorkspaceGroup
-from typing import Any, BinaryIO, List, Literal, Optional, Union
+from typing import Any, BinaryIO, List, Literal, Optional, TypeAlias, Union
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -153,7 +153,7 @@ class ActivityStatusEnum(str, Enum):
     DONE = "done"
 
 
-type ActivityType = Union[
+ActivityType: TypeAlias = Union[
     Literal["task"],
     Literal["experiment"],
     Literal["project"],
@@ -671,7 +671,7 @@ class Activity(_KaleidoscopeBaseModel):
             setattr(self, k, v)
 
 
-type ActivityIdentifier = Union[Activity, str]
+ActivityIdentifier: TypeAlias = Union[Activity, str]
 """Identifier class for Activity
 
 Activities are able to be identified by:
@@ -681,7 +681,7 @@ Activities are able to be identified by:
 * UUID
 """
 
-type DefinitionIdentifier = Union[ActivityDefinition, str]
+DefinitionIdentifier: TypeAlias = Union[ActivityDefinition, str]
 """Identifier class for ActivityDefinition
 
 ActivityDefinitions are able to be identified by:

{kalbio-0.2.2 → kalbio-0.2.3}/kalbio/client.py

@@ -119,6 +119,7 @@ class KaleidoscopeClient:
 
     _client_id: str
     _client_secret: str
+    _additional_headers: dict
     _iap_client_id: Optional[str]
 
     _refresh_token: str = None
@@ -133,6 +134,7 @@ class KaleidoscopeClient:
         client_id: Optional[str] = _env_client_id,
         client_secret: Optional[str] = _env_client_secret,
         url: str = PROD_API_URL,
+        additional_headers: dict = {},
         iap_client_id: Optional[str] = None,
         verify_ssl: bool = True,
     ):
@@ -206,6 +208,7 @@ class KaleidoscopeClient:
 
         self._client_id = client_id
         self._client_secret = client_secret
+        self.additional_headers = additional_headers
         self._iap_client_id = iap_client_id
         self._verify_ssl = verify_ssl
 
@@ -258,7 +261,7 @@ class KaleidoscopeClient:
                 "client_id": self._client_id,
                 "client_secret": self._client_secret,
             },
-            headers=self._get_iap_headers(),
+            headers={**self._get_iap_headers(), **self.additional_headers},
             timeout=TIMEOUT_MAXIMUM,
             verify=self._verify_ssl,
         )
@@ -284,7 +287,7 @@ class KaleidoscopeClient:
                 "grant_type": "refresh_token",
                 "refresh_token": self._refresh_token,
             },
-            headers=self._get_iap_headers(),
+            headers={**self._get_iap_headers(), **self.additional_headers},
             timeout=TIMEOUT_MAXIMUM,
             verify=self._verify_ssl,
         )
@@ -327,6 +330,7 @@ class KaleidoscopeClient:
             "Content-Type": "application/json",
             "X-Kal-Authorization": f"Bearer {self._access_token}",
             **self._get_iap_headers(),
+            **self.additional_headers,
         }
 
         return headers

{kalbio-0.2.2 → kalbio-0.2.3}/kalbio/dashboards.py

@@ -21,11 +21,11 @@ from functools import lru_cache
 from kalbio._kaleidoscope_model import _KaleidoscopeBaseModel
 from kalbio.client import KaleidoscopeClient
 from pydantic import TypeAdapter
-from typing import List, Literal, Union
+from typing import List, Literal, TypeAlias, Union
 
 _logger = logging.getLogger(__name__)
 
-type DashboardType = Union[
+DashboardType: TypeAlias = Union[
     Literal["decision"],
     Literal["data"],
     Literal["chart"],

{kalbio-0.2.2 → kalbio-0.2.3}/kalbio/entity_fields.py

@@ -41,7 +41,7 @@ from functools import lru_cache
 from kalbio._kaleidoscope_model import _KaleidoscopeBaseModel
 from kalbio.client import KaleidoscopeClient
 from pydantic import TypeAdapter
-from typing import List, Optional, Union
+from typing import List, Optional, TypeAlias, Union
 
 _logger = logging.getLogger(__name__)
 
@@ -151,7 +151,7 @@ class EntityField(_KaleidoscopeBaseModel):
         return f"{self.field_name}"
 
 
-type EntityFieldIdentifier = Union[EntityField, str]
+EntityFieldIdentifier: TypeAlias = Union[EntityField, str]
 """An Identifier Type for Entity Fields.
 
 An EntityField should be able to be identified by:

{kalbio-0.2.2 → kalbio-0.2.3}/kalbio/exports.py

@@ -119,7 +119,7 @@ class ExportsService:
         url = "/records/export/csv"
         file = self._client._get_file(
             url,
-            f"{download_path if download_path else "/tmp"}/{filename}",
+            f"{download_path if download_path else '/tmp'}/{filename}",
             params,
         )
 

{kalbio-0.2.2 → kalbio-0.2.3}/kalbio/records.py

@@ -62,6 +62,7 @@ from typing import (
     Dict,
     List,
     Optional,
+    TypeAlias,
     TypedDict,
     Union,
     Unpack,
@@ -631,7 +632,7 @@ class Record(_KaleidoscopeBaseModel):
             setattr(self, k, v)
 
 
-type RecordIdentifier = Union[Record, str, dict[EntityFieldIdentifier, str]]
+RecordIdentifier: TypeAlias = Union[Record, str, dict[EntityFieldIdentifier, str]]
 """Identifier type for Record
 
 Record can be identified by:
@@ -1154,7 +1155,7 @@ class RecordsService:
         ]
         # fmt: on
 
-        resp = self._client._get(f"/records?record_ids={",".join(to_fetch)}") or []
+        resp = self._client._get(f"/records?record_ids={','.join(to_fetch)}") or []
         self._create_record_list(resp)
 
         ordered = [

{kalbio-0.2.2 → kalbio-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kalbio"
-version = "0.2.2"
+version = "0.2.3"
 description = "Python client for kaleidoscope.bio"
 authors = ["Ahmed Elnaiem <ahmed@kaleidoscope.bio>"]
 readme = "README.md"