added random_stockfish_strategy.py

chesspp/random_stockfish_strategy.py

@@ -0,0 +1,36 @@
+import random
+
+import chess
+import chess.engine
+
+from chesspp.i_strategy import IStrategy
+from chesspp.eval import score_stockfish
+
+
+class RandomStockfishStrategy(IStrategy):
+    def __init__(self, rollout_depth: int, path="../stockfish/stockfish-windows-x86-64-avx2",
+                 random_seed: random.Random = random.Random()) -> None:
+        super().__init__(rollout_depth)
+        self._stockfish = None
+        self.path = path
+        self.random_seed = random_seed
+
+    def __del__(self):
+        if self._stockfish is not None:
+            self._stockfish.quit()
+
+    @property
+    def stockfish(self) -> chess.engine.SimpleEngine:
+        if self._stockfish is None:
+            self._stockfish = self.stockfish = chess.engine.SimpleEngine.popen_uci(self.path)
+        return self._stockfish
+
+    @stockfish.setter
+    def stockfish(self, stockfish):
+        self._stockfish = stockfish
+
+    def pick_next_move(self, board: chess.Board) -> chess.Move:
+        return self.random_seed.choice(list(board.legal_moves))
+
+    def analyze_board(self, board: chess.Board) -> int:
+        return score_stockfish(board, self.stockfish)