Vire com as setas

Lição 21 de 40 do curso Crie jogos com Pygame da Coddy.

A cobra vira uma vez por pressionamento de tecla, então leia as setas dos eventos KEYDOWN. Um dicionário associa cada tecla de seta à sua direção:

TURNS = {
    pygame.K_LEFT: (-1, 0),
    pygame.K_RIGHT: (1, 0),
    pygame.K_UP: (0, -1),
    pygame.K_DOWN: (0, 1),
}

event.key in TURNS é verdadeiro para as quatro setas, e TURNS[event.key] é a nova direção. Um teste substitui quatro instruções if:

elif event.type == pygame.KEYDOWN and event.key in TURNS:
    direction = TURNS[event.key]

Para cima é (0, -1): as linhas são contadas de cima para baixo na janela, como os pixels.

Desafio

Fácil

Acima de pygame.init(), adicione o dicionário TURNS para as quatro setas.
No loop de eventos, quando a tecla de um evento KEYDOWN estiver em TURNS, defina direction como o valor correspondente.

Experimente você mesmo

import pygame

CELL = 20
COLS, ROWS = 24, 18

pygame.init()
screen = pygame.display.set_mode((COLS * CELL, ROWS * CELL))
pygame.display.set_caption("Snake")
clock = pygame.time.Clock()


def draw_cell(cell, color):
    col, row = cell
    pygame.draw.rect(screen, color, (col * CELL + 1, row * CELL + 1, CELL - 2, CELL - 2))


head = (5, 9)
direction = (1, 0)

running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    head = (head[0] + direction[0], head[1] + direction[1])

    screen.fill((30, 30, 46))
    draw_cell(head, (180, 255, 160))
    pygame.display.flip()
    clock.tick(10)

pygame.quit()