Fixed backward compatibility issue with pymongo

NeonGeckoCom · Oct 9, 2023 · 2995cde · 2995cde
1 parent aadc3b3
commit 2995cde
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 12 deletions.
diff --git a/chat_server/server_utils/db_utils.py b/chat_server/server_utils/db_utils.py
@@ -493,7 +493,7 @@ def save_translations(cls, translation_mapping: dict) -> Dict[str, List[str]]:
                     filter_expression = {"_id": shout_id}
                     cls.db_controller.exec_query(
                         query=MongoQuery(
-                            command=MongoCommands.UPDATE,
+                            command=MongoCommands.UPDATE_MANY,
                             document=MongoDocuments.SHOUTS,
                             filters=filter_expression,
                             data={"translations": {}},
@@ -546,7 +546,7 @@ def get_user_preferences(cls, user_id):
             if user and not user.get("preferences"):
                 cls.db_controller.exec_query(
                     MongoQuery(
-                        command=MongoCommands.UPDATE,
+                        command=MongoCommands.UPDATE_MANY,
                         document=MongoDocuments.USERS,
                         filters=MongoFilter(key="_id", value=user_id),
                         data={"preferences": prefs},
@@ -570,7 +570,7 @@ def set_user_preferences(cls, user_id, preferences_mapping: dict):
                 }
                 cls.db_controller.exec_query(
                     MongoQuery(
-                        command=MongoCommands.UPDATE,
+                        command=MongoCommands.UPDATE_MANY,
                         document=MongoDocuments.USERS,
                         filters=MongoFilter("_id", user_id),
                         data=update_mapping,
@@ -603,7 +603,7 @@ def save_tts_response(
             )
             cls.db_controller.exec_query(
                 query=MongoQuery(
-                    command=MongoCommands.UPDATE,
+                    command=MongoCommands.UPDATE_MANY,
                     document=MongoDocuments.SHOUTS,
                     filters=MongoFilter("_id", shout_id),
                     data={f"audio.{lang}.{gender}": audio_file_name},
@@ -628,7 +628,7 @@ def save_stt_response(cls, shout_id, message_text: str, lang: str = "en"):
         try:
             cls.db_controller.exec_query(
                 query=MongoQuery(
-                    command=MongoCommands.UPDATE,
+                    command=MongoCommands.UPDATE_MANY,
                     document=MongoDocuments.SHOUTS,
                     filters=MongoFilter("_id", shout_id),
                     data={f"transcripts.{lang}": message_text},

diff --git a/chat_server/server_utils/prompt_utils.py b/chat_server/server_utils/prompt_utils.py
@@ -81,7 +81,7 @@ def handle_prompt_message(message: dict) -> bool:
                     }
                     db_controller.exec_query(
                         MongoQuery(
-                            command=MongoCommands.UPDATE,
+                            command=MongoCommands.UPDATE_MANY,
                             document=MongoDocuments.PROMPTS,
                             filters=MongoFilter(key="_id", value=prompt_id),
                             **data_kwargs,
@@ -128,7 +128,7 @@ def handle_prompt_message(message: dict) -> bool:
                         }
                         db_controller.exec_query(
                             MongoQuery(
-                                command=MongoCommands.UPDATE,
+                                command=MongoCommands.UPDATE_MANY,
                                 document=MongoDocuments.PROMPTS,
                                 filters=MongoFilter(key="_id", value=prompt_id),
                                 **data_kwargs,

diff --git a/chat_server/server_utils/user_utils.py b/chat_server/server_utils/user_utils.py
@@ -135,7 +135,7 @@ def get_bot_data(
     elif not bot_data.get("is_bot") == "1":
         db_controller.exec_query(
             MongoQuery(
-                command=MongoCommands.UPDATE,
+                command=MongoCommands.UPDATE_MANY,
                 document=MongoDocuments.USERS,
                 filters=MongoFilter("_id", bot_data["_id"]),
                 data={"is_bot": "1"},

diff --git a/chat_server/sio.py b/chat_server/sio.py
@@ -261,7 +261,7 @@ async def user_message(sid, data):
         )
         db_controller.exec_query(
             query=MongoQuery(
-                command=MongoCommands.UPDATE,
+                command=MongoCommands.UPDATE_MANY,
                 document=MongoDocuments.CHATS,
                 filters=filter_expression,
                 data={"chat_flow": new_shout_data["_id"]},
@@ -363,7 +363,7 @@ async def prompt_completed(sid, data):
     try:
         db_controller.exec_query(
             MongoQuery(
-                command=MongoCommands.UPDATE,
+                command=MongoCommands.UPDATE_MANY,
                 document=MongoDocuments.PROMPTS,
                 filters=MongoFilter(key="_id", value=prompt_id),
                 data=prompt_summary_agg,

diff --git a/utils/database_utils/mongo_utils/structures.py b/utils/database_utils/mongo_utils/structures.py
@@ -52,6 +52,8 @@ class MongoCommands(Enum):
     DELETE_MANY = "delete_many"
     # Update operation
     UPDATE = "update"
+    UPDATE_MANY = "update_many"
+    UPDATE_ONE = "update_one"
 
 
 class MongoDocuments(Enum):
@@ -135,7 +137,10 @@ def build_filters(self):
     def build_setter(self) -> dict:
         """Builds setter for Mongo Query"""
         res = None
-        if self.command.value == MongoCommands.UPDATE.value:
+        if self.command.value in (
+            MongoCommands.UPDATE_MANY.value,
+            MongoCommands.UPDATE_ONE.value,
+        ):
             res = {f"${self.data_action.lower()}": self.data}
         elif self.command.value in (
             MongoCommands.INSERT_ONE.value,

diff --git a/utils/database_utils/mongodb_connector.py b/utils/database_utils/mongodb_connector.py
@@ -43,7 +43,8 @@ class MongoDBConnector(DatabaseConnector):
         "delete_many",
         "find",
         "find_one",
-        "update",
+        "update_one",
+        "update_many",
     )
 
     @property