Updated the file in response to the PR comment

Devasy23 · Mar 16, 2024 · 80b3f41 · 80b3f41
1 parent 10739a1
commit 80b3f41
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/API/database.py b/API/database.py
@@ -2,7 +2,6 @@
 from datetime import datetime
 from pymongo import MongoClient
 
-
 class Database:
     def __init__(self, uri="mongodb://localhost:27017/", db_name="ImageDB"):
         self.client = MongoClient(uri)
@@ -20,5 +19,5 @@ def find_one(self, collection, filter, projection=None):
     def find_one_and_delete(self, collection, query):
         return self.db[collection].find_one_and_delete(query)
 
-    def find_one_and_update(self, collection, query, update):
-        return self.db[collection].find_one_and_update(filter=query, update=update)
+    def update_one(self, collection, query, update):
+        return self.db[collection].update_one(query, {"$set": update})