From a2ebb314eb2dd718668128e58228c6d3ba259aa4 Mon Sep 17 00:00:00 2001
From: Karma Riuk <riukkarma@gmail.com>
Date: Wed, 29 Jan 2025 12:07:26 +0100
Subject: [PATCH] pieces now know if they are white or black

---
 src/logic/board.py        | 8 ++++----
 src/logic/pieces/piece.py | 7 ++++++-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/logic/board.py b/src/logic/board.py
index f2f5d36..451fbe0 100644
--- a/src/logic/board.py
+++ b/src/logic/board.py
@@ -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():
diff --git a/src/logic/pieces/piece.py b/src/logic/pieces/piece.py
index 6b67e46..c0ae3e8 100644
--- a/src/logic/pieces/piece.py
+++ b/src/logic/pieces/piece.py
@@ -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