42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
"""Definition of get_image function."""
|
|
|
|
import logging
|
|
import os
|
|
from io import BytesIO
|
|
|
|
from minio import Minio
|
|
from PIL import Image
|
|
|
|
|
|
def get_image(
|
|
client: Minio,
|
|
object_name: str,
|
|
) -> Image.Image:
|
|
"""Get image from image subfolder in bucket in Minio."""
|
|
assert isinstance(client, Minio)
|
|
assert isinstance(object_name, str)
|
|
assert len(object_name) > 0
|
|
# prepare arguments
|
|
assert 'MINIO_BUCKET_NAME' in os.environ
|
|
bucket_name = os.getenv('MINIO_BUCKET_NAME', default='')
|
|
subfolder = 'images'
|
|
# get object from bucket
|
|
object_name = f'{subfolder}/{object_name}'
|
|
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)
|
|
# convert data to image
|
|
image = Image.open(buffer)
|
|
logging.debug('got data from %s', object_name)
|
|
return image
|