pieces now know if they are white or black

This commit is contained in:
Karma Riuk 2025-01-29 12:07:26 +01:00
parent 2363b39484
commit a2ebb314eb
2 changed files with 10 additions and 5 deletions

View File

@ -16,8 +16,8 @@ class Board:
pos_w_pawn = Position(x, 1)
pos_b_pawn = Position(x, 6)
self._white[pos_w_pawn] = Pawn(pos_w_pawn)
self._black[pos_b_pawn] = Pawn(pos_b_pawn)
self._white[pos_w_pawn] = Pawn(pos_w_pawn, Piece.WHITE)
self._black[pos_b_pawn] = Pawn(pos_b_pawn, Piece.BLACK)
pos_w_piece = Position(x, 0)
pos_b_piece = Position(x, 7)
@ -34,8 +34,8 @@ class Board:
elif x == 4:
piece = King
assert piece != None, f"Didn't know which piece to assign for {x = }"
self._white[pos_w_piece] = piece(pos_w_piece)
self._black[pos_b_piece] = piece(pos_b_piece)
self._white[pos_w_piece] = piece(pos_w_piece, Piece.WHITE)
self._black[pos_b_piece] = piece(pos_b_piece, Piece.BLACK)
def create_board():

View File

@ -2,8 +2,13 @@ from logic.position import Position
class Piece:
def __init__(self, pos) -> None:
WHITE = 0
BLACK = 1
def __init__(self, pos, colour) -> None:
self.pos = pos
assert colour == self.WHITE or colour == self.BLACK, "The colour of the piece must be either Piece.WHITE or Piece.BLACK"
self.colour = colour
def position(self) -> Position:
return self.pos