import pygame
import sys

WIDTH = 800
HEIGHT = 600

FPS = 60

COLOR_BLACK = (0, 0, 0)
COLOR_RED = (255, 0, 0)

def main():
    # Initialize imported pygame modules
    pygame.init()

    # Set the window's caption
    pygame.display.set_caption("Pong")

    clock = pygame.time.Clock()

    screen = pygame.display.set_mode((WIDTH, HEIGHT))

    background = pygame.Surface((WIDTH, HEIGHT))
    background = background.convert()
    background.fill(COLOR_BLACK)

    # Blit everything to screen
    screen.blit(background, (0, 0))

    # Update the screen
    pygame.display.flip()

    # Main loop
    while True:
        clock.tick(FPS)

        # Erase everything drawn at last step by filling the background
        # with color black
        background.fill(COLOR_BLACK)

        # Check for Quit event
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()

        # Check for key presses and update paddles accordingly
        keys_pressed = pygame.key.get_pressed()
        if keys_pressed[pygame.K_w]:
            # Do something
            pass
        if keys_pressed[pygame.K_s]:
            # Do something
            pass
        if keys_pressed[pygame.K_UP]:
            # Do something
            pass
        if keys_pressed[pygame.K_DOWN]:
            # Do something
            pass

        # Update game state

        # Render current game state
        pygame.draw.rect(background,
                     COLOR_RED,
                     pygame.Rect(100, 120, 200, 400))
        
        pygame.display.flip()
        screen.blit(background, (0, 0))


if __name__ == '__main__':
    main()