diff options
| -rw-r--r-- | console/main.py | 50 |
1 files changed, 35 insertions, 15 deletions
diff --git a/console/main.py b/console/main.py index 5953d4a..e3454ae 100644 --- a/console/main.py +++ b/console/main.py @@ -18,27 +18,47 @@ class bord: def outside_board(self, coords): return coords[0] < 0 or \ coords[1] < 0 or \ - coords[0] > self.width - 1 or \ - coords[1] > self.height - 1 + coords[0] > self.height - 1 or \ + coords[1] > self.width - 1 + + def recursive_solve(self, coords, check_for, direction, current_length): + new_position = ( + coords[0] + direction[0], + coords[1] + direction[1] + ) + if self.outside_board(new_position) or self.board[new_position[1]][new_position[0]] != check_for: + return current_length + else: + return self.recursive_solve(new_position, check_for, direction, current_length + 1) + + def check_win(self, coords): + directions = [ + ( 0, 1), + ( 1, 1), + ( 1, 0), + ( 1, -1), + ( 0, -1), + (-1, -1), + (-1, 0), + (-1, 1) + ] + values = list() + for direction in directions: + values.append(self.recursive_solve(coords, self.board[coords[0]][coords[1]], direction, 0)) def drop_fisje(self, column, disc): - row = -1 - for r in range(len(self.board)): - if self.board[r][column] != EMPTY: - row = r - 1 - break - self.board[row][column] = disc + for row in self.board: + print(row) + for row, value in enumerate(self.board): + if self.board[row][column] == EMPTY: + self.board[row][column] = disc + self.check_win((row, column)) + return def main(): disc_a = True gert = bord(7, 6) - gert.board[0][0] = "x" - gert.board[1][0] = DISC_A - gert.board[2][0] = DISC_A - gert.board[3][0] = DISC_A - gert.board[0][1] = DISC_B - gert.board[0][2] = DISC_B - gert.board[0][3] = DISC_B + gert.board[0][2] = DISC_A while True: gert.print() column = int(input("column?: ")) - 1 |