databutton 0.37.8__tar.gz → 0.38.0__tar.gz

{databutton-0.37.8 → databutton-0.38.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databutton
-Version: 0.37.8
+Version: 0.38.0
 Summary: The CLI for databutton.com
 License: Databutton
 Author: Databutton
@@ -16,7 +16,7 @@ Requires-Dist: certifi (>=2022.12.7,<2023.0.0)
 Requires-Dist: chromadb (>=0.4.22,<0.5.0)
 Requires-Dist: cryptography (>=40.0.1,<41.0.0)
 Requires-Dist: httpx (>=0.20.0,<1)
-Requires-Dist: pandas (>=0.25,<3)
+Requires-Dist: pandas[feather] (>=0.25,<3)
 Requires-Dist: pydantic (>=1,<3)
 Requires-Dist: tenacity (>=8.0.0,<9.0.0)
 Description-Content-Type: text/markdown

databutton-0.38.0/databutton/experimental/__init__.py

@@ -0,0 +1,3 @@
+from . import auth, folder
+
+__all__ = ["auth", "folder"]

databutton-0.38.0/databutton/experimental/folder/__init__.py

@@ -0,0 +1,3 @@
+from .folder import download, upload
+
+__all__ = ["upload", "download"]

databutton-0.38.0/databutton/experimental/folder/folder.py

@@ -0,0 +1,44 @@
+import io
+import os
+import zipfile
+
+import databutton as db
+
+
+def zip_folder_to_bytes(folder_path: str) -> bytes:
+    """Zip the contents of a folder and return as bytes."""
+    zip_bytes = io.BytesIO()  # Use BytesIO as a buffer
+    with zipfile.ZipFile(zip_bytes, "w", zipfile.ZIP_DEFLATED) as zipf:
+        for root, _, files in os.walk(folder_path):
+            for file in files:
+                file_path = os.path.join(root, file)
+                zipf.write(
+                    file_path,
+                    os.path.relpath(file_path, os.path.join(folder_path, "..")),
+                )
+    zip_bytes.seek(0)  # Go to the start of the BytesIO buffer
+    return zip_bytes.getvalue()
+
+
+def unzip_bytes_to_folder(bytes_data: bytes, extract_path: str):
+    """Unzip bytes into a folder."""
+    # Ensure the extraction path exists
+    os.makedirs(extract_path, exist_ok=True)
+    zip_bytes = io.BytesIO(bytes_data)
+    with zipfile.ZipFile(zip_bytes, "r") as zip_ref:
+        zip_ref.extractall(extract_path)
+
+
+def upload(key: str, folder_path: str):
+    """Zip and upload a folder to blob storage."""
+    zipped_bytes = zip_folder_to_bytes(folder_path)
+    db.storage.binary.put(key, zipped_bytes)
+
+
+def download(key: str, folder_path: str):
+    """Download and unzip a folder from blob storage."""
+    zipped_bytes = db.storage.binary.get(key)
+    if zipped_bytes is not None:
+        # Ensure the target folder exists before extraction
+        os.makedirs(folder_path, exist_ok=True)
+        unzip_bytes_to_folder(zipped_bytes, folder_path)

{databutton-0.37.8 → databutton-0.38.0}/databutton/version.py

@@ -5,4 +5,4 @@ This module contains project version information.
 .. moduleauthor:: Databutton <support@databutton.com>
 """
 
-__version__ = "0.37.8"
+__version__ = "0.38.0"

{databutton-0.37.8 → databutton-0.38.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "databutton"
-version = "0.37.8"
+version = "0.38.0"
 description = "The CLI for databutton.com"
 authors = ["Databutton <hi@databutton.io>"]
 license = "Databutton"
@@ -10,7 +10,7 @@ packages = [{ include = "databutton" }]
 [tool.poetry.dependencies]
 python = ">=3.10,<3.12"
 httpx = ">=0.20.0,<1"
-pandas = "<3,>=0.25"
+pandas = { extras = ["feather"], version = "<3,>=0.25" }
 pydantic = ">=1,<3"
 PyJWT = "^2.2.0"
 PyYAML = "^6.0"
@@ -30,7 +30,7 @@ isort = "^5.12.0"
 python-semantic-release = "7.28.1"
 pytest-mock = "^3.7.0"
 pandas-stubs = "^1.5.2.221124"
-ruff = "^0.2.0"
+ruff = "^0.3.0"
 types-pyjwt = "^1.7.1"
 
 [build-system]

databutton-0.37.8/databutton/experimental/__init__.py

@@ -1,3 +0,0 @@
-from . import auth
-
-__all__ = ["auth"]