diff --git a/shared/data_store/connect.py b/shared/data_store/connect.py index ddf4c41..14c61ae 100644 --- a/shared/data_store/connect.py +++ b/shared/data_store/connect.py @@ -1,5 +1,6 @@ """Definition of connect function.""" +import logging import os from typing import Literal @@ -38,7 +39,14 @@ def connect( secret_key=minio_secret_key, secure=False, ) - # ensure bucket exists - if not client.bucket_exists(bucket_name=minio_bucket_name): - client.make_bucket(bucket_name=minio_bucket_name) + # ensure buckets exists + bucket_list = [ + 'images', + 'models', + ] + for bucket_name in bucket_list: + if not client.bucket_exists(bucket_name=bucket_name): + logging.info('creating bucket: %s', bucket_name) + client.make_bucket(bucket_name=minio_bucket_name) + logging.debug('finished') return client