fixed deprecated import

other/transfer_images_minio_subfolder.py

@@ -5,7 +5,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 from PIL import Image
 
-from shared.data_store import connect, get, put_image
+from shared.data_store import connect_minio, get, put_image
 from shared.utils import setup_logging
 
 if __name__ == '__main__':
@@ -16,7 +16,7 @@ if __name__ == '__main__':
     # setup logging
     setup_logging()
     # connect to minio
-    minio_client = connect()
+    minio_client = connect_minio()
     # list images in bucket
     BUCKET_NAME = 'visual-critical-discourse-analysis'
     obj_list = minio_client.list_objects(

other/transfer_images_mongo_minio.py

@@ -1,4 +1,5 @@
 """Script to move all images from MongoDB to MinIO."""
+
 from __future__ import annotations
 
 import logging
@@ -9,12 +10,9 @@ from bson import ObjectId
 from dotenv import load_dotenv
 from pymongo.collection import Collection
 
-from shared.data_store import connect as connect_minio
-from shared.data_store import put
-from shared.database import connect
-from shared.database import VisualCommunication
-from shared.utils import check_env
-from shared.utils import setup_logging
+from shared.data_store import connect_minio, put
+from shared.database import VisualCommunication, connect
+from shared.utils import check_env, setup_logging
 
 
 def list_mongo_document_ids(
@@ -98,7 +96,7 @@ if __name__ == '__main__':
         try:
             # save image to buffer
             buffer = BytesIO()
-            vis_com.image.save(buffer, 'png')   # type: ignore
+            vis_com.image.save(buffer, 'png')  # type: ignore
             # put buffer in minio
             object_name = put(
                 client=minio_client,
@@ -110,13 +108,14 @@ if __name__ == '__main__':
             continue
         # update visual communication in mongodb
         try:
-            vis_com.image = None    # type: ignore
+            vis_com.image = None  # type: ignore
             vis_com.object_name = object_name
             update_visual_communication(collection, vis_com)
         except Exception as exc:
             logging.debug(exc)
             logging.error(
-                'failed updating visual communication %s', vis_com.name,
+                'failed updating visual communication %s',
+                vis_com.name,
             )
             continue
         logging.debug('updated visual communication %s', vis_com.name)