45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
"""Definition of put_image function."""
|
|
|
|
import logging
|
|
import os
|
|
from hashlib import md5
|
|
from io import BytesIO
|
|
|
|
from minio import Minio
|
|
from PIL import Image
|
|
|
|
|
|
def put_image(
|
|
client: Minio,
|
|
image: Image.Image,
|
|
) -> str:
|
|
"""Put image in image subfolder in bucket in Minio and return MD5 checksum
|
|
used as object name."""
|
|
assert isinstance(client, Minio)
|
|
assert isinstance(image, Image.Image)
|
|
bucket_name = os.getenv('MINIO_BUCKET_NAME', default='')
|
|
assert len(bucket_name) > 0
|
|
subfolder = 'images'
|
|
# save image to buffer
|
|
buffer = BytesIO()
|
|
image.save(buffer, 'png')
|
|
# get md5 of image
|
|
checksum = md5(buffer.getbuffer()).hexdigest()
|
|
# prepare for saving
|
|
num_bytes = buffer.tell()
|
|
buffer.seek(0)
|
|
# send data to bucket
|
|
object_name = f'{subfolder}/{checksum}'
|
|
try:
|
|
client.put_object(
|
|
bucket_name=bucket_name,
|
|
object_name=object_name,
|
|
length=num_bytes,
|
|
data=buffer,
|
|
)
|
|
except Exception as exc:
|
|
logging.error('failed saving data to MinIO')
|
|
raise exc
|
|
logging.debug('saved data to %s', object_name)
|
|
return checksum
|