add retry logic to evaluate_move helper #13

Author: Phillyclause89

chmengine/engines/cmhmey2_pool_executor.py

@@ -1,6 +1,7 @@
 """Cmhmey Jr.'s Mad Scientist Uncle Who Likes to make clones of Cmhmey Jr."""
 
 from concurrent.futures import Future, ProcessPoolExecutor, as_completed
+from sqlite3 import OperationalError
 from typing import Dict, Optional, Union
 
 from chess import Board, Move
@@ -22,16 +23,34 @@ def evaluate_move(board: Board, depth: int = 1, debug: bool = False, cache_dir:
     Parameters
     ----------
     board : Board
+        The chess board state to evaluate.
     depth : int
+        The search depth for the evaluation.
     debug : bool
+        Whether to enable debug output.
     cache_dir : str
+        The cache directory for the engine.
 
     Returns
     -------
     Pick
+        The best move and its associated evaluation score.
+
+    Raises
+    ------
+    OperationalError
+        If the database remains locked after exhausting retries or for unexpected operational errors.
     """
     CMHMEngine2.cache_dir = cache_dir
-    return CMHMEngine2(board=board, depth=depth).pick_move(debug=debug)
+    try:
+        return CMHMEngine2(board=board, depth=depth).pick_move(debug=debug)
+    except OperationalError as error_o:
+        if "database is locked" in str(error_o):
+            try:
+                return evaluate_move(board=board, depth=depth, debug=debug, cache_dir=cache_dir)
+            except RecursionError:
+                pass
+        raise OperationalError('Unexpected Operational Error: {str(error_o)}') from error_o
 
 
 class CMHMEngine2PoolExecutor:

tests/test_cmhmey2_pool_executor.py

@@ -96,7 +96,7 @@ def test_debug_output(self):
     def test_pick_move(self) -> None:
         """Tests pick_move method."""
         start = perf_counter()
-        pick = self.executor.pick_move(debug=True)
+        pick = self.executor.pick_move()
         duration_first = (perf_counter() - start) / self.executor.engine.board.legal_moves.count()
         print(
             f"{self.executor.engine.fen()} pick_move call: ({pick[0].uci()},"
@@ -111,7 +111,7 @@ def test_pick_move(self) -> None:
         for i, move in enumerate(init_w_moves, 2):
             self.executor.engine.board.push(move)
             start = perf_counter()
-            response_pick = self.executor.pick_move(debug=True)
+            response_pick = self.executor.pick_move()
             duration_rep_pick = (perf_counter() - start) / self.executor.engine.board.legal_moves.count()
             first_time_pick_times.append(duration_rep_pick)
             print(
@@ -120,7 +120,7 @@ def test_pick_move(self) -> None:
             )
             self.executor.engine.board.pop()
             start = perf_counter()
-            new_pick = self.executor.pick_move(debug=True)
+            new_pick = self.executor.pick_move()
             new_duration = (perf_counter() - start) / self.executor.engine.board.legal_moves.count()
             init_board_pick_times.append(new_duration)
             revisit_pick_times.append(new_duration)