Fixing ObjectId issues

Grexiem · Dec 10, 2024 · 044e569 · 044e569
1 parent 783f91a
commit 044e569
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/api/api.py b/api/api.py
@@ -94,7 +94,7 @@ def score(player, id):
     if request.method == "GET":
         score = get_score_player(player, blindtest_db["players"])
         for bt in score:
-            if bt["id"] == id:
+            if str(bt["_id"]) == id:
                 return jsonify({"score": bt["score"]})
         return jsonify({"score": 0})
     if request.method == "POST":

diff --git a/api/bt.py b/api/bt.py
@@ -36,19 +36,19 @@ def blindtest(id, round, bt_db):
 
 
 def get_blindtest(id_blindtest, bt_db):
-    query = {"_id": id_blindtest}
+    query = {"_id": ObjectId(id_blindtest)}
     result = bt_db.find_one(query)
     return result
 
 
 def get_score_bt(id_blindtest, bt_db):
-    query = {"_id": id_blindtest}
+    query = {"_id": ObjectId(id_blindtest)}
     result = bt_db.find_one(query)
     return result["score"]
 
 
 def change_score_bt(player, score, id_blindtest, bt_db):
-    query = {"_id": id_blindtest}
+    query = {"_id": ObjectId(id_blindtest)}
     result = bt_db.find_one(query)
     update_query = {"$set": {player: score}}
     if result == None:

diff --git a/api/player.py b/api/player.py
@@ -1,3 +1,4 @@
+from bson import ObjectId
 import pymongo
 
 folder_player = "../players/"