Spaces:
Running
Running
File size: 1,422 Bytes
5eb71ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
__all__ = ["PLAYER1", "PLAYER2", "Connect4"]
PLAYER1, PLAYER2 = "red", "yellow"
class Connect4:
"""
A Connect Four game.
Play moves with :meth:`play`.
Get past moves with :attr:`moves`.
Check for a victory with :attr:`winner`.
"""
def __init__(self):
self.moves = []
self.top = [0 for _ in range(7)]
self.winner = None
@property
def last_player(self):
"""
Player who played the last move.
"""
return PLAYER1 if len(self.moves) % 2 else PLAYER2
@property
def last_player_won(self):
"""
Whether the last move is winning.
"""
b = sum(1 << (8 * column + row) for _, column, row in self.moves[::-2])
return any(b & b >> v & b >> 2 * v & b >> 3 * v for v in [1, 7, 8, 9])
def play(self, player, column):
"""
Play a move in a column.
Returns the row where the checker lands.
Raises :exc:`RuntimeError` if the move is illegal.
"""
if player == self.last_player:
raise RuntimeError("It isn't your turn.")
row = self.top[column]
if row == 6:
raise RuntimeError("This slot is full.")
self.moves.append((player, column, row))
self.top[column] += 1
if self.winner is None and self.last_player_won:
self.winner = self.last_player
return row |