diff --git a/shared/datastore/src/get.py b/shared/datastore/src/get.py index b41ef63..dfd9206 100644 --- a/shared/datastore/src/get.py +++ b/shared/datastore/src/get.py @@ -1,7 +1,6 @@ """Definition of get function.""" import logging -import os from io import BytesIO from traceback import print_exc @@ -10,19 +9,22 @@ from minio import Minio def get( client: Minio, + bucket_name: str, object_name: str, ) -> BytesIO: """Get buffer from bucket in MinIO.""" assert isinstance(client, Minio) assert isinstance(object_name, str) - bucket_name = os.getenv('MINIO_BUCKET_NAME', default='') - # get buffer + assert isinstance(bucket_name, str) + assert len(bucket_name) > 0 try: + # make request response = client.get_object( bucket_name=bucket_name, object_name=object_name, ) assert response.status == 200 + # get buffer buffer = BytesIO() chunk_size = 2**14 while chunk := response.read(chunk_size):