containerio 2.3.1__tar.gz → 2.3.2__tar.gz

{containerio-2.3.1 → containerio-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: containerio
-Version: 2.3.1
+Version: 2.3.2
 Summary: Azure Blob Storage handler with unified local/cloud interface
 Author: Alliance SwissPass
 Requires-Dist: polars>=1.30.0

{containerio-2.3.1 → containerio-2.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "containerio"
-version = "2.3.1"
+version = "2.3.2"
 description = "Azure Blob Storage handler with unified local/cloud interface"
 readme = "README.md"
 requires-python = ">=3.11"

{containerio-2.3.1 → containerio-2.3.2}/src/containerio/handler.py

@@ -684,12 +684,8 @@ class StorageHandler:
             pl.DataFrame: The content of the JSON file as a Polars
             DataFrame.
         """
-        if self.local:
-            file_path = self.base_path / blob_name
-            return pl.read_json(source=file_path, **kwargs)
-        else:
-            data = self._read_blob_as_bytes(blob_name)
-            return pl.read_json(source=BytesIO(data), **kwargs)
+        data = self.read_blob_as_bytes(blob_name=blob_name)
+        return pl.read_json(source=BytesIO(data), **kwargs)
 
     def read_csv_blob(self, blob_name: str, **kwargs: Dict) -> pl.DataFrame:
         """Reads a CSV file from storage into a Polars DataFrame.
@@ -900,17 +896,9 @@ class StorageHandler:
         Returns:
             None
         """
-        if self.local:
-            file_path = self.base_path / blob_name
-            file_path.parent.mkdir(parents=True, exist_ok=True)
-            df.write_json(file=file_path, **kwargs)
-        else:
-            # Write the DataFrame to an in-memory stream
-            buffer = BytesIO()
-            df.write_json(file=buffer, **kwargs)
-            buffer.seek(0)
-            # Upload the bytes to Azure Blob Storage
-            self._write_blob_as_bytes(blob_name, buffer.getvalue())
+        buffer = BytesIO()
+        df.write_json(file=buffer, **kwargs)
+        self.write_blob_as_bytes(blob_name=blob_name, data=buffer.getvalue())
 
     def write_csv_blob(self, blob_name: str, df: pl.DataFrame, **kwargs: Dict) -> None:
         """Writes a Polars DataFrame to a CSV file in storage.