datalab-api 0.2.9__tar.gz → 0.2.10__tar.gz

{datalab_api-0.2.9/src/datalab_api.egg-info → datalab_api-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: datalab-api
-Version: 0.2.9
+Version: 0.2.10
 Summary: A Python API for instances of the datalab data management platform (https://github.com/datalab-org/datalab).
 Author-email: Matthew Evans <matthew@ml-evs.science>
 License: MIT
@@ -43,6 +43,7 @@ Requires-Dist: pandas~=2.2; extra == "cheminventory-helper"
 Requires-Dist: openpyxl~=3.0; extra == "cheminventory-helper"
 Provides-Extra: all
 Requires-Dist: datalab-api[cli,dev,docs]; extra == "all"
+Dynamic: license-file
 
 # <div align="center"><i>datalab</i> Python API</div>
 

{datalab_api-0.2.9 → datalab_api-0.2.10}/src/datalab_api/__init__.py

@@ -282,8 +282,9 @@ class DatalabClient(BaseDatalabClient):
                 warnings.warn(f"Will not overwrite existing file {f['name']}")
                 continue
             with open(f["name"], "wb") as file:
-                response = self.session.get(url, follow_redirects=True)
-                file.write(response.content)
+                with self.session.stream("GET", url, follow_redirects=True) as response:
+                    for chunk in response.iter_bytes(chunk_size=1024):
+                        file.write(chunk)
 
     def get_block(self, item_id: str, block_id: str, block_data: dict[str, Any]) -> dict[str, Any]:
         """Get a block with a given ID and block data.

{datalab_api-0.2.9 → datalab_api-0.2.10}/src/datalab_api/_base.py

@@ -270,7 +270,7 @@ class BaseDatalabClient(metaclass=AutoPrettyPrint):
 
         return self.api_key
 
-    def __enter__(self) -> "BaseDatalabClient":
+    def __enter__(self):
         return self
 
     def __exit__(self, *_):

{datalab_api-0.2.9 → datalab_api-0.2.10/src/datalab_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: datalab-api
-Version: 0.2.9
+Version: 0.2.10
 Summary: A Python API for instances of the datalab data management platform (https://github.com/datalab-org/datalab).
 Author-email: Matthew Evans <matthew@ml-evs.science>
 License: MIT
@@ -43,6 +43,7 @@ Requires-Dist: pandas~=2.2; extra == "cheminventory-helper"
 Requires-Dist: openpyxl~=3.0; extra == "cheminventory-helper"
 Provides-Extra: all
 Requires-Dist: datalab-api[cli,dev,docs]; extra == "all"
+Dynamic: license-file
 
 # <div align="center"><i>datalab</i> Python API</div>
 

{datalab_api-0.2.9 → datalab_api-0.2.10}/tests/test_client.py

@@ -28,7 +28,7 @@ def test_redirect_url(fake_ui_html, fake_info_json, fake_block_info_json):
 
 
 @respx.mock
-def test_sample_list(fake_samples_json, fake_info_json, fake_block_info_json):
+def test_sample_list(fake_samples_json, fake_info_json, fake_sample_json, fake_block_info_json):
     fake_api = respx.get("https://api.datalab.industries/").mock(
         return_value=Response(200, content="<!doctype html></html>")
     )
@@ -41,6 +41,9 @@ def test_sample_list(fake_samples_json, fake_info_json, fake_block_info_json):
     fake_block_info_api = respx.get("https://api.datalab.industries/info/blocks").mock(
         return_value=Response(200, json=fake_block_info_json)
     )
+    fake_item_api = respx.get("https://api.datalab.industries/get-item-data/KUVEKJ").mock(
+        return_value=Response(200, json=fake_sample_json)
+    )
     with DatalabClient("https://api.datalab.industries") as client:
         samples = client.get_items(display=True)
         assert fake_samples_api.called
@@ -49,3 +52,6 @@ def test_sample_list(fake_samples_json, fake_info_json, fake_block_info_json):
         assert fake_block_info_api.called
         assert len(samples)
         assert samples[0]["item_id"] == "test"
+        sample = client.get_item("KUVEKJ", display=True)
+        assert fake_item_api.called
+        assert sample["item_id"] == "KUVEKJ"