bitbully.py

"""BitBully: A Connect Four AI agent with opening book support."""
 
import importlib
import importlib.resources
import pathlib
from importlib.resources.abc import Traversable
from typing import Literal
 
from bitbully import bitbully_core
 
from .board import Board
 
 
class BitBully:
    """A Connect Four AI agent with opening book support."""
 
    def __init__(self, opening_book: Literal["default", "8-ply", "12-ply", "12-ply-dist"] | None = "default") -> None:
        """Initializes the BitBully agent with the specified opening book.
 
        Args:
            opening_book (Literal["default", "8-ply", "12-ply", "12-ply-dist"] | None):
                The type of opening book to use. Options are:
                - "default": Uses the default 12-ply distance-based opening book.
                - "8-ply": Uses an 8-ply opening book.
                - "12-ply": Uses a 12-ply opening book.
                - "12-ply-dist": Uses a 12-ply distance-based opening book.
                - None: No opening book will be used.
        """
        self.opening_book_type = opening_book
 
        db_path: Traversable | str | None = None
        if opening_book == "default":
            db_path = importlib.resources.files("bitbully").joinpath("assets/book_12ply_distances.dat")
        elif opening_book == "8-ply":
            db_path = importlib.resources.files("bitbully").joinpath("assets/book_8ply.dat")
        elif opening_book == "12-ply":
            db_path = importlib.resources.files("bitbully").joinpath("assets/book_12ply.dat")
        elif opening_book == "12-ply-dist":
            db_path = importlib.resources.files("bitbully").joinpath("assets/book_12ply_distances.dat")
 
        if db_path:
            self.bitbully_agent = bitbully_core.BitBullyCore(pathlib.Path(str(db_path)))
        else:
            self.bitbully_agent = bitbully_core.BitBullyCore()
 
    def score_next_moves(self, board: Board) -> list[int]:
        """Scores all possible moves for the given board state.
 
        Args:
            board (Board):
                The current board state.
 
        Returns:
            list[int]: A list of scores for each column (0-6).
        """
        return self.bitbully_agent.scoreMoves(board._board)