/
game.py
executable file
·43 lines (35 loc) · 1.39 KB
/
game.py
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
#!/usr/bin/env python
from board import Board
import sys
from players import HumanPlayer, ComputerPlayer
class ConnectFour(object):
def __init__(self, cell_list=None, columns=7, column_size=6, human=False, level='easy'):
self.adversaries = [HumanPlayer(), HumanPlayer(False, 'O')]
if not human:
self.adversaries[1] = ComputerPlayer(level=level)
self.board = Board()
def read_player_move(self, ):
return self.current_player.move(self.board)
def play(self, ):
self.current_player = [player for player in self.adversaries
if player.turn is True][0]
while True:
print self.board
move = self.read_player_move()
if move in ('q', 'Q'):
sys.exit(0)
self.board.set_cell(move, self.current_player.name)
over = self.board.game_over()
if over:
break
self.current_player = self.get_next_player()[0]
print self.board
print "Cat's Game!" if over == 'tie' else \
self.current_player.name + " won the game!"
def get_next_player(self, ):
for player in self.adversaries:
player.turn = not player.turn
return [player for player in self.adversaries if player.turn is True]
if __name__ == '__main__':
game = ConnectFour()
game.play()