From 6b0a13423086a33624040587fc3d6cb2dda571ff Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Thu, 30 Jan 2025 16:56:44 +0100 Subject: [PATCH] fixed more colour issues --- src/logic/pieces/pawn.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/logic/pieces/pawn.py b/src/logic/pieces/pawn.py index 0c292ce..fe2339e 100644 --- a/src/logic/pieces/pawn.py +++ b/src/logic/pieces/pawn.py @@ -1,5 +1,5 @@ from logic.move import Move, PieceMove -from logic.pieces.piece import Piece +from logic.pieces.piece import Colour, Piece from logic.position import Position class Pawn(Piece): @@ -8,23 +8,23 @@ class Pawn(Piece): # can we capture to the left? if self.pos.x > 0 and ( - (self.colour == self.WHITE and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y + 1))) + (self.colour == Colour.WHITE and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y + 1))) or - (self.colour == self.BLACK and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y - 1))) + (self.colour == Colour.BLACK and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y - 1))) ): if capturable_piece.colour != self.colour: ret.append(PieceMove(self, capturable_piece.pos, is_capturing = True)) # can we capture to the right? if self.pos.x < 7 and ( - (self.colour == self.WHITE and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y + 1))) + (self.colour == Colour.WHITE and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y + 1))) or - (self.colour == self.BLACK and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y - 1))) + (self.colour == Colour.BLACK and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y - 1))) ): if capturable_piece.colour != self.colour: ret.append(PieceMove(self, capturable_piece.pos, is_capturing = True)) - if self.colour == Piece.WHITE: + if self.colour == Colour.WHITE: for dy in range(1, 3 if self.pos.y == 1 else 2): if self.pos.y + dy > 7 or board.piece_at(self.pos.x, self.pos.y + dy): break