from random import choice
from time import sleep
class Game:
def __init__(self):
self.map = []
self.p1 = 0
self.p2 = 0
self.game = False
def set_up(self):
a = 0
self.game = True
self.map = [[],[],[],[],[],[]]
while a < len(self.map):
while len(self.map[a]) < 6:
self.map[a].append(' ')
a +=1
user = input('What piece do you want to be represented by: R or Y ? ').lower()
if user=='r':
self.p1 ='R'
self.p2 = 'Y'
elif user=='y':
self.p1 = 'Y'
self.p2 = 'R'
else:
self.p1 = 'R'
self.p2 = 'Y'
def display(self,n):
a = str(n).replace("[",'')
b = a.replace(',','|')
c = b.replace("'","")
d = c.replace(']','')
print('|',d,'|')
def display2(self):
print(' 0 1 2 3 4 5')
print('-'*20)
for i in self.map:
self.display(i)
print('-'*20)
def play1(self):
self.display2()
col = 0
try:
col = eval(input('Pick a column: '))
except:
NameError
TypeError
ValueError
SyntaxError
print('Try Again')
self.play1()
if col not in range(0,6):
print('Out of range')
self.play1()
else:
if self.game==True:
self.insert(col,self.p1)
def check_full(self):
c = 0
a = 0
r = 0
while a < 6:
while r < 6:
if self.map[a][r] in ['R','Y']:
c +=1
r +=1
r = 0
a+=1
if c >=36:
print('FULL')
self.game = False
self.set_up()
def check_map(self,n):
a = 0 #row
b = 0 #space
c = 0 #counting
while a < 6:
while b < 6:
if self.map[a][b]==n:
c +=1
else:
c = 0
if c >=4:
self.game = False
b +=1
b = 0
a +=1
a = 0 #row
b = 0 #space
while b < 6:
while a < 6:
if self.map[a][b]==n:
c +=1
else:
c = 0
if c>=4:
self.game = False
a +=1
b +=1
a = 0
b = 0
def insert(self,n,u):
a = 5
p = False
while a !=-1:
if self.map[a][n] not in ['R','Y']:
if p==False:
self.map[a][n] = u
p = True
a -=1
self.check_full()
if self.game==True:
self.check_map(u)
def play1_com(self):
n = choice([0,1,2,3,4,5])
print(n)
sleep(1)
if self.game==True:
self.display2()
self.insert(n,self.p1)
def play2(self):
n = choice([0,1,2,3,4,5])
print(n)
sleep(1)
if self.game==True:
self.display2()
self.insert(n,self.p2)
def cycle(self):
self.set_up()
mode = input('1.YOU vs COM\n2.COM vs COM: ')
if mode=='2':
while self.game==True:
self.play1_com()
self.play2()
else:
while self.game==True:
self.play1()
self.play2()
print('GAME OVER')
game = Game()
game.cycle()
