#COPYRIGHT 2019 THEWILDTHORN
class Connect4:

    def __init__(self, width, height, window=None): # Initializes the game by setting the parameters for the board and the data that will be stored in the board (or list of lists)
        self.width = width
        self.height = height
        self.data = []
        for row in range(self.height): # This for statement creates the list of lists represented by self.data
            boardRow = []
            for col in range(self.width):
                boardRow += [' ']
            self.data += [boardRow]

    def __repr__(self): # Sets the literal structure of the board so that it's easy to interpret for the players
        s = ''
        for row in range(self.height):
            s += '|'
            for col in range(self.width):
                s += self.data[row][col] + '|'
            s += '\n'
        s += '--'*self.width + '-\n'
        for col in range(self.width):
            s += ' ' + str(col%10)
        s += '\n'
        return s

    def clear(self): # Clears the current state of the board
        for row in range(self.height):
            for col in range(self.width): # For every value in the board
                self.data[row][col] = ' ' # Override whatever is there with a space (the original value each spot starts with)

    def addMove(self, col, ox): # Adds a "checker" to the column passed
        if self.allowsMove(col): # Uses allowsMove to check if the column is available
            for row in range(self.height):
                if self.data[row][col] != ' ': # If the spot is not empty...
                    self.data[row-1][col] = ox # Add the checker to the spot above it
                    return
            self.data[self.height-1][col] = ox # Otherwise but the checker there

    def allowsMove(self, col): # Tests to see if a column has room for another move
        if 0 <= col < self.width:
            return self.data[0][col] == ' '
        else:
            return False

    def delMove(self, col): # Deletes the last move played in column passed to the function
        for row in range(self.height): # For each row..
            if self.data[row][col] != ' ': # If the indexed spot is not empty
                self.data[row][col] = ' ' # Override whatever is there with a space
                return
        return

    def isFull(self): # Checks if the board is full or not
        for values in self.data: # For each value in the entire board, if there's a space anywhere then it's not full (False). If there isn't a space, then the board is full (True).
            if ' ' in values:
                return False
        return True

    def winsFor(self, ox): # Checks to see if a checker ('X' or 'O') won the Connect 4.
        for row in range(self.height): # Horizontal check
            for col in range(self.width - 3): # Uses an anchor point (subtracting 3 from the width and so forth) from where to start checking so it doesn't go out of range.
                if self.data[row][col] == ox and self.data[row][col+1] == ox and self.data[row][col+2] == ox and self.data[row][col+3] == ox:
                    return True
        for row in range(self.height - 3): # Vertical check
            for col in range(self.width):
                if self.data[row][col] == ox and self.data[row+1][col] == ox and self.data[row+2][col] == ox and self.data[row+3][col] == ox:
                    return True
        for row in range(self.height - 3): # SW>NE check
            for col in range(self.width - 3):
                if self.data[row][col] == ox and self.data[row+1][col+1] == ox and self.data[row+2][col+2] == ox and self.data[row+3][col+3] == ox:
                    return True
        for row in range(3, self.height): # SE>NW check
            for col in range(self.width - 3):
                if self.data[row][col] == ox and self.data[row-1][col+1] == ox and self.data[row-2][col+2] == ox and self.data[row-3][col+3] == ox:
                    return True
        return False

    def hostGame(self): # Makes the game more playable by asking the user their turn and checking for wins or ties automatically
        move = 0 # This is used later so it will rotate back and forth between asking X to move and asking O to move
        while self.isFull() == False and self.winsFor('OX') != True: # While the board is not full and nobody has won yet, continue the loop
            if move == 0: # When move is 0, it's X's turn (X always goes first)
                print(game) # Prints the board in its current state, this variable is located at the end of the file
                X = int(input('Player "X", what column would you like to move? ')) # Asks for the user's input for which column to move, then turns it into an integer and stores it in X
                if self.allowsMove(X): # If that position is allowed...
                    self.addMove(X, 'X') # Add that move for 'X'
                    move += 1 # Increment move by 1 so that move won't be 0, and thus it will be O's turn
                    if self.winsFor('X'): # Check to see if X won the game with that move
                        print('Congratulations Player "X", you win!')
                        return
                else:
                    while self.allowsMove(X) == False: # While the user keeps putting invalid columns...
                        X = int(input('Oops! Your response was invalid, try again: ')) # Keep asking until the user chooses a column that is allowed
                    self.addMove(X, 'X')
                    move += 1
                    if self.winsFor('X'):
                        print('Congratulations Player "X", you win!')
                        return
            else: # O's turn to play, same mechanics just for a different player
                print(game)
                O = int(input('Player "O", what column would you like to move? '))
                if self.allowsMove(O):
                    self.addMove(O, 'O')
                    move -= 1 # Decrements move so it will be 0 and thus will be X's turn again
                    if self.winsFor('O'):
                        print('Congratulations Player "O", you win!')
                        return
                else:
                    while self.allowsMove(O) == False:
                        O = int(input('Oops! Your response was invalid, try again: '))
                    move -= 1
                    self.addMove(O, 'O')
                    if self.winsFor('O'):
                        print('Congratulations Player "O", you win!')
                        return
        print('It\'s a tie!') # If it breaks out of the loop and someone hasn't already won, then that mean the board must be full and thus would be a tie
        return

def main():
    game = Connect4(7, 6)
    game.hostGame()
    print(game)

if __name__=='__main__':
    main()