import pygame

pygame.init()
largura_tela = 800
altura_tela = 600
tela = pygame.display.set_mode((largura_tela, altura_tela))
terminou = False

# Cria relógio
clock = pygame.time.Clock()

cor = (255, 255, 0) # cor amarela
area = (100, 100, 30, 30)
pygame.draw.rect(tela, cor, area)

cor = (255, 0, 0) # cor vermelha
area = (200, 200, 30, 30)
pygame.draw.rect(tela, cor, area)

while not terminou:

    # Checar os eventos do mouse aqui:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            terminou = True

            # Finaliza a janela do jogo
            pygame.display.quit()

    
    if not terminou:
        # Atualiza o desenho na tela
        pygame.display.update()

        # Atraso de 50Hz
        clock.tick(50)

# Finaliza o pygame
pygame.quit()
