43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""Definition of connect function."""
|
|
|
|
import logging
|
|
import os
|
|
|
|
from minio import Minio
|
|
|
|
|
|
def connect() -> Minio:
|
|
"""Connect to MinIO server."""
|
|
# ensure necessary env vars available
|
|
env_var_list = [
|
|
'MINIO_ENDPOINT',
|
|
'MINIO_ACCESS_KEY',
|
|
'MINIO_SECRET_KEY',
|
|
'MINIO_BUCKET_NAME',
|
|
]
|
|
for env_var in env_var_list:
|
|
# ensure env var set
|
|
assert (
|
|
env_var in os.environ
|
|
), f"environment variable not set: {
|
|
env_var
|
|
}"
|
|
# prepare arguments
|
|
minio_endpoint = os.getenv('MINIO_ENDPOINT')
|
|
minio_access_key = os.getenv('MINIO_ACCESS_KEY')
|
|
minio_secret_key = os.getenv('MINIO_SECRET_KEY')
|
|
minio_bucket_name = os.getenv('MINIO_BUCKET_NAME')
|
|
# connect client
|
|
client = Minio(
|
|
endpoint=minio_endpoint,
|
|
access_key=minio_access_key,
|
|
secret_key=minio_secret_key,
|
|
secure=False,
|
|
)
|
|
# ensure bucket exists
|
|
if not client.bucket_exists(bucket_name=minio_bucket_name):
|
|
logging.info('creating bucket: %s', minio_bucket_name)
|
|
client.make_bucket(bucket_name=minio_bucket_name)
|
|
logging.debug('finished')
|
|
return client
|