fix net counter

official-stockfish · Feb 28, 2024 · e241033 · e241033
1 parent 1bc8b1a
commit e241033
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/server/fishtest/api.py b/server/fishtest/api.py
@@ -608,6 +608,9 @@ def download_nn(self):
         nn = self.request.rundb.get_nn(self.request.matchdict["id"])
         if nn is None:
             raise exception_response(404)
+        else:
+            self.request.rundb.increment_nn_downloads(self.request.matchdict["id"])
+
         return HTTPFound(
             "https://data.stockfishchess.org/nn/" + self.request.matchdict["id"]
         )

diff --git a/server/fishtest/rundb.py b/server/fishtest/rundb.py
@@ -477,11 +477,10 @@ def update_nn(self, net):
         self.nndb.update_one({"name": net["name"]}, {"$set": net})
 
     def get_nn(self, name):
-        nn = self.nndb.find_one({"name": name}, {"nn": 0})
-        if nn:
-            self.nndb.update_one({"name": name}, {"$inc": {"downloads": 1}})
-            return nn
-        return None
+        return self.nndb.find_one({"name": name}, {"nn": 0})
+
+    def increment_nn_downloads(self, name):
+        self.nndb.update_one({"name": name}, {"$inc": {"downloads": 1}})
 
     def get_nns(
         self, user_id, user="", network_name="", master_only=False, limit=0, skip=0