implemented base functions

shared/datastore/src/get_image.py

@@ -2,13 +2,14 @@
 
 import logging
 import os
-from io import BytesIO
 
 from minio import Minio
 from PIL import Image
 
 from shared.utils import check_env
 
+from .get import get
+
 
 def get_image(
     client: Minio,
@@ -23,19 +24,11 @@ def get_image(
     bucket_name = str(os.getenv('MINIO_BUCKET_NAME'))
     object_name = f'images/{object_name}'
     # get object from bucket
-    try:
-        response = client.get_object(
-            bucket_name=bucket_name,
-            object_name=object_name,
-        )
-        buffer = BytesIO(response.data)
-    except Exception as exc:
-        logging.error('failed getting data from MinIO')
-        raise exc
-    finally:
-        response.close()
-        response.release_conn()
-    buffer.seek(0)
+    buffer = get(
+        client=client,
+        bucket_name=bucket_name,
+        object_name=object_name,
+    )
     # convert data to image
     image = Image.open(buffer)
     logging.debug('got data from %s', object_name)