Update state.py

state.py

@@ -20,4 +20,67 @@ def __str__(self):
             string += str(self.board[row]) + "\n"
         return string
 
-
+    def first_moves_set(self):
+        firstMoves = []
+        firstMoves.append((1, 1))
+        firstMoves.append((1, 18))
+        firstMoves.append((18, 1))
+        firstMoves.append((18, 18))
+        firstMoves.append((9, 9))
+        firstMoves.append((9, 10))
+        firstMoves.append((10, 9))
+        firstMoves.append((10, 10))
+        return firstMoves
+
+    def second_moves_set(self):
+        secondMoves = []
+        #If bottom left corner of the board is removed
+        if (self.board[1][1] == ' '):
+            secondMoves.append((1, 2))
+            secondMoves.append((2, 1))
+            return secondMoves
+        #Top left corner is removed
+        elif (self.board[1][18] == ' '):
+            secondMoves.append((1, 17))
+            secondMoves.append((2, 18))
+            return secondMoves
+        #Top right
+        elif (self.board[18][18] == ' '):
+            secondMoves.append((17, 18))
+            secondMoves.append((18, 17))
+            return secondMoves
+        #Bottom right
+        elif (self.board[18][1] == ' '):
+            secondMoves.append((17, 1))
+            secondMoves.append((18, 2))
+            return secondMoves
+
+        #Middle game states, bottom left
+        elif (self.board[9][9] == ' '):
+            secondMoves.append((9, 10))
+            secondMoves.append((9, 8))
+            secondMoves.append((8, 9))
+            secondMoves.append((10, 9))
+            return secondMoves
+        #Mid top left
+        elif (self.board[9][10] == ' '):
+            secondMoves.append((9, 11))
+            secondMoves.append((9, 9))
+            secondMoves.append((10, 10))
+            secondMoves.append((8, 10))
+            return secondMoves
+        #Mid top right
+        elif (self.board[10][10] == ' '):
+            secondMoves.append((9, 10))
+            secondMoves.append((11, 10))
+            secondMoves.append((10, 9))
+            secondMoves.append((10, 11))
+            return secondMoves
+        #Mid bot right
+        elif (self.board[10][9] == ' '):
+            secondMoves.append((11, 9))
+            secondMoves.append((9, 9))
+            secondMoves.append((10, 10))
+            secondMoves.append((10, 8))
+            return secondMoves
+