renamed module

misc/generate_random_prediction.py

@@ -5,8 +5,8 @@ from pathlib import Path
 from dotenv import load_dotenv
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb
-from shared.mongodb.src.classes import VisualCommunication
+from shared.docstore import connect_mongodb
+from shared.docstore.src.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 

misc/get_visual_communication.py

@@ -5,7 +5,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb, get_visual_communication
+from shared.docstore import connect_mongodb, get_visual_communication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 

misc/image_download.py

@@ -5,8 +5,8 @@ from pathlib import Path
 
 from dotenv import load_dotenv
 
-from shared.mongodb import connect_mongodb
-from shared.mongodb.classes import VisualCommunication
+from shared.docstore import connect_mongodb
+from shared.docstore.classes import VisualCommunication
 
 if __name__ == '__main__':
     # prepare env vars

misc/image_upload.py

@@ -6,8 +6,8 @@ from dotenv import load_dotenv
 from pymongo.errors import DuplicateKeyError
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb
-from shared.mongodb.src.classes import VisualCommunication
+from shared.docstore import connect_mongodb
+from shared.docstore.src.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 

misc/image_upload_to_server.py

@@ -6,8 +6,8 @@ from dotenv import load_dotenv
 from pymongo.errors import DuplicateKeyError
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb
-from shared.mongodb.src.classes import VisualCommunication
+from shared.docstore import connect_mongodb
+from shared.docstore.src.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 

misc/model_outputs_from_annotation.py

@@ -1,4 +1,4 @@
-from shared.mongodb.classes import ModelData
+from shared.docstore.classes import ModelData
 
 if __name__ == '__main__':
     # instantiate data object

misc/prediction_upload.py

@@ -5,8 +5,8 @@ from pathlib import Path
 from dotenv import load_dotenv
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb, upsert_prediction
-from shared.mongodb.src.classes import VisualCommunication
+from shared.docstore import connect_mongodb, upsert_prediction
+from shared.docstore.src.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 

misc/total_annotated.py

@@ -5,7 +5,7 @@ from pathlib import Path
 
 from dotenv import load_dotenv
 
-from shared.mongodb import connect_mongodb, count_documents
+from shared.docstore import connect_mongodb, count_documents
 
 if __name__ == '__main__':
     # prepare env vars

misc/total_documents.py

@@ -5,7 +5,7 @@ from pathlib import Path
 
 from dotenv import load_dotenv
 
-from shared.mongodb import connect_mongodb, count_documents
+from shared.docstore import connect_mongodb, count_documents
 
 if __name__ == '__main__':
     # prepare env vars

model/src/main.py

@@ -10,7 +10,7 @@ from tqdm import tqdm
 from utils import DEVICE, VCDADataset, load_model
 
 from shared.datastore import Datastore
-from shared.mongodb.classes import ModelData
+from shared.docstore.classes import ModelData
 from shared.utils import setup_logging
 
 if __name__ == '__main__':

model/src/models/visual_communication.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from torch import nn
 
-from shared.mongodb.src.classes import ModelData
+from shared.docstore.src.classes import ModelData
 
 from .angle import AngleTail
 from .contact import ContactTail

other/transfer_images_mongo_minio.py

@@ -10,8 +10,8 @@ from dotenv import load_dotenv
 from pymongo.collection import Collection
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb
-from shared.mongodb.src.classes import VisualCommunication
+from shared.docstore import connect_mongodb
+from shared.docstore.src.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 

pyproject.toml

@@ -119,7 +119,7 @@ module = "models.*"
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]
-module = "shared.mongodb.*"
+module = "shared.docstore.*"
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]

shared/docstore/src/classes/visual_communication.py

@@ -13,7 +13,7 @@ from pydantic import BaseModel, ConfigDict
 from pymongo.collection import Collection
 
 from shared.datastore import Datastore
-from shared.mongodb.src.classes import ModelData
+from shared.docstore.src.classes import ModelData
 
 
 class VisualCommunication(BaseModel):

shared/docstore/src/get_visual_communication.py

@@ -6,8 +6,8 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.mongodb.src.classes import VisualCommunication
-from shared.mongodb.src.exceptions import NoDocumentFoundException
+from shared.docstore.src.classes import VisualCommunication
+from shared.docstore.src.exceptions import NoDocumentFoundException
 
 
 def get_visual_communication(

shared/docstore/src/upsert_annotation.py

@@ -4,7 +4,7 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.mongodb.src.classes import ModelData
+from shared.docstore.src.classes import ModelData
 
 
 def upsert_annotation(

shared/docstore/src/upsert_prediction.py

@@ -4,7 +4,7 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.mongodb.src.classes import ModelData
+from shared.docstore.src.classes import ModelData
 
 
 def upsert_prediction(

shared/docstore/src/upsert_visual_communication.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from pymongo.collection import Collection
 
-from shared.mongodb.src.classes import VisualCommunication
+from shared.docstore.src.classes import VisualCommunication
 
 
 def upsert_visual_communication(

web_ui/src/app/init_app.py

@@ -12,9 +12,9 @@ from pydantic import ValidationError
 from pymongo.collection import Collection
 
 from shared.datastore import Datastore
-from shared.mongodb import count_documents, get_visual_communication, upsert_annotation
-from shared.mongodb.src.classes import ModelData, VisualCommunication
-from shared.mongodb.src.exceptions import NoDocumentFoundException
+from shared.docstore import count_documents, get_visual_communication, upsert_annotation
+from shared.docstore.src.classes import ModelData, VisualCommunication
+from shared.docstore.src.exceptions import NoDocumentFoundException
 
 from .layout import app_layout
 

web_ui/src/app/layout/labels.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import dash_mantine_components as dmc
 from dash import dcc, html
 
-from shared.mongodb.classes import (
+from shared.docstore.classes import (
     AngleData,
     ContactData,
     DistanceData,

web_ui/src/main.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import os
 
 from shared.datastore import Datastore
-from shared.mongodb import connect_mongodb
+from shared.docstore import connect_mongodb
 from shared.utils import check_env, setup_logging
 
 from .app import init_app