40 lines
992 B
Python
40 lines
992 B
Python
"""Definition of put function."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
from hashlib import md5
|
|
from io import BytesIO
|
|
|
|
from minio import Minio
|
|
|
|
|
|
def put(
|
|
client: Minio,
|
|
buffer: BytesIO,
|
|
) -> str:
|
|
"""Put buffer in bucket in MinIO and return MD5 checksum as object name."""
|
|
assert isinstance(client, Minio)
|
|
assert isinstance(buffer, BytesIO)
|
|
bucket_name = os.getenv('MINIO_BUCKET_NAME', default='')
|
|
assert len(bucket_name) > 0
|
|
# get md5 of image
|
|
checksum = md5(buffer.getbuffer()).hexdigest()
|
|
# prepare for saving
|
|
num_bytes = buffer.tell()
|
|
buffer.seek(0)
|
|
# send data to bucket
|
|
try:
|
|
client.put_object(
|
|
bucket_name=bucket_name,
|
|
object_name=checksum,
|
|
length=num_bytes,
|
|
data=buffer,
|
|
)
|
|
except Exception as exc:
|
|
logging.error('failed saving data to MinIO')
|
|
raise exc
|
|
logging.debug('saved data to %s', checksum)
|
|
return checksum
|