changes to state

zew15101 · Dec 4, 2019 · 5a38693 · 5a38693
1 parent 3a1fd22
commit 5a38693
Showing 1 changed file with 14 additions and 4 deletions.
diff --git 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
-
-