From f983145dfbb48e846a4a569cdec18388e84acafd Mon Sep 17 00:00:00 2001
From: Brian Bjarke Jensen <bbj@skov.dk>
Date: Wed, 3 Apr 2024 20:58:55 +0200
Subject: [PATCH] added database util functions

---
 core/database/__init__.py                          | 7 +++++++
 core/database/utils/upsert_annotation.py           | 2 +-
 core/database/utils/upsert_prediction.py           | 2 +-
 core/database/utils/upsert_visual_communication.py | 0
 4 files changed, 9 insertions(+), 2 deletions(-)
 mode change 100644 => 100755 core/database/utils/upsert_annotation.py
 mode change 100644 => 100755 core/database/utils/upsert_prediction.py
 mode change 100644 => 100755 core/database/utils/upsert_visual_communication.py

diff --git a/core/database/__init__.py b/core/database/__init__.py
index f044a35..ef9599c 100644
--- a/core/database/__init__.py
+++ b/core/database/__init__.py
@@ -4,3 +4,10 @@ from __future__ import annotations
 from .classes import Dataset
 from .classes import NoDocumentFoundException
 from .classes import VisualCommunication
+from .utils import connect
+from .utils.count_documents import count_documents
+from .utils.get_visual_communication import get_visual_communication
+from .utils.list_names import list_names
+from .utils.upsert_annotation import upsert_annotation
+from .utils.upsert_prediction import upsert_prediction
+from .utils.upsert_visual_communication import upsert_visual_communication
diff --git a/core/database/utils/upsert_annotation.py b/core/database/utils/upsert_annotation.py
old mode 100644
new mode 100755
index 54be9eb..2124d2f
--- a/core/database/utils/upsert_annotation.py
+++ b/core/database/utils/upsert_annotation.py
@@ -7,7 +7,7 @@ from pymongo.collection import Collection
 from core.dto import ModelData
 
 
-def upsert_annotations(
+def upsert_annotation(
     collection: Collection,
     vis_com_name: str,
     annotations: ModelData,
diff --git a/core/database/utils/upsert_prediction.py b/core/database/utils/upsert_prediction.py
old mode 100644
new mode 100755
index 2b1cd1a..f47d75b
--- a/core/database/utils/upsert_prediction.py
+++ b/core/database/utils/upsert_prediction.py
@@ -7,7 +7,7 @@ from pymongo.collection import Collection
 from core.dto import ModelData
 
 
-def upsert_predictions(
+def upsert_prediction(
     collection: Collection,
     vis_com_name: str,
     predictions: ModelData,
diff --git a/core/database/utils/upsert_visual_communication.py b/core/database/utils/upsert_visual_communication.py
old mode 100644
new mode 100755