console_only.py

#Spencer Wittrock 84023369 & Emma Lisowski 26903080

from connectfour import *

def game_board_gui(GameState) ->str:
    '''prints a readable gameboard where 0s are represented by dots, 1 is represented by R and 2 is represented by Y'''
    zero_list = []
    for x in range(BOARD_COLUMNS):
        print(x+1,end=' ')
    print()
    for column in GameState.board:
        column_list = []
        for row in column:
            if row == 0:
                column_list.append('.')
            elif row == 1:
                column_list.append('R')
            elif row == 2:
                column_list.append('Y')
        zero_list.append(column_list)
    for column in zip(*zero_list):
        print(*column)

def determine_winner(GameState:GameState):
    '''checks the gamestate to see if a player has won'''
    if winner(GameState) == RED:
        return 'Red player wins'
    elif winner(GameState) == YELLOW:
        return 'Yellow player wins'


def command_choice(choice:str, column_choice:int, GameState: 'GameState'):
    '''handles the drio ir pop input logic'''
    if choice == 'drop' or choice == 'DROP':
        game_board_gui(drop(GameState, column_choice))
        return drop(GameState, column_choice)
                
    elif choice == 'pop' or choice == 'POP':
        game_board_gui(pop(GameState, column_choice))
        return pop(GameState, column_choice)


def run_game():
    '''initializes the gameboard and runs the drop or pop logic'''
    new_game()
    game_board_gui(new_game())
    GameState = new_game()
    while True:
        try:
            print('Player' , GameState.turn)
            choice = input('drop, or pop? ')
            if choice == 'drop' or choice == 'pop':
                column_choice = int(input('Which column? (1-{})'.format(BOARD_COLUMNS)))-1
            else:
                raise "error"

            GameState = command_choice(choice, column_choice, GameState)

            if determine_winner(GameState) != None:
                print(determine_winner(GameState))
                break
            
        except:
            print('ERROR')

if __name__ == '__main__':
    run_game()