mongo.fix: return new department document in case not found

hacettepeoyt · Dec 13, 2023 · c8132c9 · c8132c9
1 parent 77fce7f
commit c8132c9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/mongo.py b/src/mongo.py
@@ -17,17 +17,18 @@ def __fetch_collection(self) -> AsyncIOMotorClient:
         collection = db['departments']
         return collection
 
-    async def find(self, department_id: str) -> list[dict]:
+    async def find(self, department_id: str) -> dict:
         collection = self.__fetch_collection()
         document = await collection.find_one({'department_id': department_id})
 
         if not document:
-            await collection.insert_one({
+            document = {
                 'department_id': department_id,
                 'announcement_list': [],
                 'is_active': True
-            })
-            return []
+            }
+            await collection.insert_one(document)
+            return document
 
         return document