databutton 0.37.8__py3-none-any.whl → 0.38.0__py3-none-any.whl

databutton/experimental/__init__.py

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

databutton/experimental/folder/__init__.py

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

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/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.dist-info → databutton-0.38.0.dist-info}/METADATA

@@ -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.37.8.dist-info → databutton-0.38.0.dist-info}/RECORD

@@ -1,8 +1,10 @@
 databutton/__init__.py,sha256=h5GdA1J_7UCaGxMFbobohHAiweUvqhOnm7TQ7cPaeGU,418
 databutton/cachetools.py,sha256=soq0BDkE_YNaybBx-fI4pJ1x8k9KuotZ0f_OuZCCI9k,3554
-databutton/experimental/__init__.py,sha256=e2vI2X5xQxTPZtsiYXnO91y52BQ4tTx1zbbvDnFK8tk,39
+databutton/experimental/__init__.py,sha256=S5AFlCNuvWf5HR2BUs_uzNd-GmjbpVm6eA__DLMUpPU,57
 databutton/experimental/auth/__init__.py,sha256=hqMbUs9GIVW-bJrayhWzdmSLkmU1wvFHDp6jMKwQrWc,73
 databutton/experimental/auth/get_user.py,sha256=qi6nkG35PRgeljV4J-cHdEE4nhfwJHOYv93tBXR5A9I,3314
+databutton/experimental/folder/__init__.py,sha256=27dUyREJ-08xP4OmQUtAyWWz7_hWEat6ajqiJ2xgyeo,71
+databutton/experimental/folder/folder.py,sha256=LjzUZiE4nbvk2164s0dKY-hssCuJz5rWdUSsiw-rmRA,1548
 databutton/internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databutton/internal/auth.py,sha256=MY4S7Gvb6xufmZ0SssZka0eY_18lJbxHW61cHrisnro,3182
 databutton/internal/byteutils.py,sha256=2SYQxhbK1Kj__u_ap01kPpeb1b_bcZAXSz6aqWwetUA,740
@@ -22,8 +24,8 @@ databutton/secrets/__init__.py,sha256=sI0okrfRBs2l5tcLTrKc9uUfPVVRMM9D7WED8p226t
 databutton/secrets/secrets.py,sha256=pehXsD09iKTxHaNQJ6awvIHXgRXBJrf4Xhrc11ZUeOk,3427
 databutton/storage/__init__.py,sha256=7ZNd4eQQ7lYYIe1MqCuFbFF1aAVCuj2c_zmJfaPVlDY,252
 databutton/storage/storage.py,sha256=x9LTyoAfLTWFCoy7XhA0tVUTkjldx8gYax09RkhMR1Q,20369
-databutton/version.py,sha256=NZcgP3ieiynaLngkEJPA1A_OBcVcVEj1nmqyqoKWVFs,175
-databutton-0.37.8.dist-info/LICENSE,sha256=c7h4pcVZapFEmqWoRTRv_S1P_aibrtDobqmlR-QtMUk,45
-databutton-0.37.8.dist-info/METADATA,sha256=b_URw0kOu65AEt4GIMIWLUmmgt21bTfXCP34qZyP2rc,2744
-databutton-0.37.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-databutton-0.37.8.dist-info/RECORD,,
+databutton/version.py,sha256=nxbdNEKRX8YnlQvq3y9BsGGdJQZVOcowtLGPv4Q6zrw,175
+databutton-0.38.0.dist-info/LICENSE,sha256=c7h4pcVZapFEmqWoRTRv_S1P_aibrtDobqmlR-QtMUk,45
+databutton-0.38.0.dist-info/METADATA,sha256=30xJ0gxIGDrkcAq-m4lHtEE5YWrSc6Hch6DaDeO_nwU,2753
+databutton-0.38.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+databutton-0.38.0.dist-info/RECORD,,