# Arup Guha
# 6/17/2024
# Day 11 Version of Connect 4 for SI@UCF Camp (written in class)

import pygame, sys
from pygame.locals import *
from connect4file import connect4

def main():

    # Create the object.
    mygame = connect4()

    # Game loop.
    while True: 

        for event in pygame.event.get():

            if event.type == QUIT:
                pygame.quit()
                sys.exit()

            # This is for teams moving in one of the 7 columns.
            if event.type == KEYDOWN:

                if event.key == K_1:
                    mygame.executeTurn(0)
                elif event.key == K_2:
                    mygame.executeTurn(1)
                elif event.key == K_3:
                    mygame.executeTurn(2)
                elif event.key == K_4:
                    mygame.executeTurn(3)
                elif event.key == K_5:
                    mygame.executeTurn(4)
                elif event.key == K_6:
                    mygame.executeTurn(5)
                elif event.key == K_7:
                    mygame.executeTurn(6)
                    
            # Draw and update the display.
            mygame.draw()
            pygame.display.update()

# Run it.
main()