From eca7a6ae0ca15dc0ed61cfe7bf07e15b583ba9b9 Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Wed, 29 Jan 2025 16:59:13 +0100 Subject: [PATCH] now black pawns can also go forward --- src/logic/pieces/pawn.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/logic/pieces/pawn.py b/src/logic/pieces/pawn.py index 1be50b0..76e7171 100644 --- a/src/logic/pieces/pawn.py +++ b/src/logic/pieces/pawn.py @@ -23,9 +23,17 @@ class Pawn(Piece): if capturable_piece.colour != self.colour: ret.append(capturable_piece.pos) - 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 - ret.append(Position(self.pos.x, self.pos.y + dy)) + if self.colour == Piece.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 + ret.append(Position(self.pos.x, self.pos.y + dy)) + else: + for dy in range(1, 3 if self.pos.y == 6 else 2): + if self.pos.y - dy < 0 or board.piece_at(self.pos.x, self.pos.y - dy): + break + ret.append(Position(self.pos.x, self.pos.y - dy)) + + print(ret) return ret