From 44c36cdeb53eeb577dc7af6c2d64b75824492e1d Mon Sep 17 00:00:00 2001 From: Ruben Fiszel Date: Mon, 3 Jun 2024 17:20:29 +0200 Subject: [PATCH] fix: fix python client for S3Object --- python-client/wmill/wmill/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python-client/wmill/wmill/client.py b/python-client/wmill/wmill/client.py index 44c6575128..1cf289f3db 100644 --- a/python-client/wmill/wmill/client.py +++ b/python-client/wmill/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, s3object["storage"]) + reader = S3BufferedReader(f"{self.workspace}", self.client, s3object["s3"], s3_resource_path, s3object["storage"] if "storage" in s3object else None) return reader def write_s3_file( @@ -479,7 +479,7 @@ 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: + if s3object is not None and "storage" in s3object and s3object["storage"] is not None: query_params["storage"] = s3object["storage"] try: