From d55adf87acc20b66eeb64500de411e902f00b29b Mon Sep 17 00:00:00 2001 From: Brian Bjarke Jensen Date: Sat, 13 Jul 2024 23:42:42 +0200 Subject: [PATCH] added buckets exist check --- shared/data_store/connect.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) 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