56 lines
1 KiB
Python
56 lines
1 KiB
Python
from enum import Enum
|
|
|
|
|
|
class Piece:
|
|
def get_valid_jumps() -> list:
|
|
raise NotImplementedError
|
|
|
|
|
|
class Bee(Piece):
|
|
pass
|
|
|
|
|
|
class Beetle(Piece):
|
|
pass
|
|
|
|
|
|
class Spider(Piece):
|
|
pass
|
|
|
|
|
|
class Grasshopper(Piece):
|
|
pass
|
|
|
|
|
|
class Ant(Piece):
|
|
pass
|
|
|
|
|
|
class PieceTranslator(Enum):
|
|
q = Bee
|
|
b = Beetle
|
|
s = Spider
|
|
g = Grasshopper
|
|
a = Ant
|
|
|
|
|
|
|
|
class Bee(Piece):
|
|
def __init__(self, board, p, q, team):
|
|
self.board = board
|
|
self.p = p
|
|
self.q = q
|
|
self.team = team # 'team' should be 'upper' or 'lower' based on the piece case.
|
|
|
|
def get_valid_jumps(self):
|
|
# The bee can move to an adjacent empty space.
|
|
valid_moves = []
|
|
for neighbor in get_neighbors(self.p, self.q):
|
|
# Check if the neighbor is within the bounds of the board and is empty
|
|
if neighbor in self.board and self.board[neighbor] == '':
|
|
valid_moves.append(neighbor)
|
|
return valid_moves
|
|
|
|
|
|
get_neighbors = <function get_neighbors at 0x7eb8eda90940>
|
|
|