wmill 1.339.2__tar.gz → 1.340.2__tar.gz

{wmill-1.339.2 → wmill-1.340.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wmill
-Version: 1.339.2
+Version: 1.340.2
 Summary: A client library for accessing Windmill server wrapping the Windmill client API
 Home-page: https://windmill.dev
 License: Apache-2.0

{wmill-1.339.2 → wmill-1.340.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wmill"
-version = "1.339.2"
+version = "1.340.2"
 description = "A client library for accessing Windmill server wrapping the Windmill client API"
 license = "Apache-2.0"
 homepage = "https://windmill.dev"

{wmill-1.339.2 → wmill-1.340.2}/wmill/client.py

@@ -440,7 +440,7 @@ class Windmill:
             print(file_reader.read())
         '''
         """
-        reader = S3BufferedReader(f"{self.workspace}", self.client, s3object["s3"], s3_resource_path)
+        reader = S3BufferedReader(f"{self.workspace}", self.client, s3object["s3"], s3_resource_path, s3object["storage"])
         return reader
 
     def write_s3_file(
@@ -479,6 +479,8 @@ class Windmill:
             query_params["file_key"] = s3object["s3"]
         if s3_resource_path is not None and s3_resource_path != "":
             query_params["s3_resource_path"] = s3_resource_path
+        if s3object is not None and s3object["storage"] is not None:
+            query_params["storage"] = s3object["storage"]
 
         try:
             # need a vanilla client b/c content-type is not application/json here

{wmill-1.339.2 → wmill-1.340.2}/wmill/s3_reader.py

@@ -5,12 +5,14 @@ import httpx
 
 
 class S3BufferedReader(BufferedReader):
-    def __init__(self, workspace: str, windmill_client: httpx.Client, file_key: str, s3_resource_path: str | None):
+    def __init__(self, workspace: str, windmill_client: httpx.Client, file_key: str, s3_resource_path: str | None, storage: str | None):
         params = {
             "file_key": file_key,
         }
         if s3_resource_path is not None:
             params["s3_resource_path"] = s3_resource_path
+        if storage is not None:
+            params["storage"] = storage
         self._context_manager = windmill_client.stream(
             "GET",
             f"/w/{workspace}/job_helpers/download_s3_file",

{wmill-1.339.2 → wmill-1.340.2}/wmill/s3_types.py

@@ -1,5 +1,6 @@
 class S3Object(dict):
     s3: str
+    storage: str | None
 
     def __getattr__(self, attr):
         return self[attr]