From 5a3869368773877ad75c53ce3f91948b2edf7d55 Mon Sep 17 00:00:00 2001 From: "john.wohl@uconn.edu" Date: Wed, 4 Dec 2019 18:45:11 -0500 Subject: [PATCH] changes to state --- state.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/state.py b/state.py index 93601e2..993211d 100644 --- a/state.py +++ b/state.py @@ -1,3 +1,6 @@ +WHITE = "W" +BLACK = "B" + class State: def __init__(self): self.nrows = 18 @@ -10,14 +13,21 @@ def setup(self): self.board.append([]) for col in range(self.ncols): if((row + col) % 2 == 0): - self.board[row].append("B") + self.board[row].append(BLACK) else: - self.board[row].append("W") + self.board[row].append(WHITE) + + def removePiece(self, row, col): + self.board[row][col] = None + + def placePiece(self, row, col, color): + if(not self.board[row][col]): + self.board[row][col] = color + else: + print("Error: attempted to place piece on occupied spot") def __str__(self): string = "" for row in range(self.nrows): string += str(self.board[row]) + "\n" return string - -