snake_v2.py

import pygame
from random import randint

FOOD_SIZE = 10
DISPLAY_WIDTH = 640
DISPLAY_HEIGHT = 480
BG_COLOUR = (44, 62, 80)
SNAKE_COLOUR = (236, 240, 241)
FOOD_COLOUR = (241, 196, 15)


def draw_food(display, coords):
    new_food_list = []
    for pair in coords:
        new_food_list.append(pygame.draw.rect(display,
                                              FOOD_COLOUR,
                                              (pair[0],
                                               pair[1],
                                               FOOD_SIZE, FOOD_SIZE)))
    return new_food_list


def main():
    pygame.init()
    clock = pygame.time.Clock()
    snake_display = pygame.display.set_mode((DISPLAY_WIDTH, DISPLAY_HEIGHT))

    pygame.display.update()

    # generate food
    food_coordinates = []
    for i in range(4):
        new_x = randint(0, DISPLAY_WIDTH - 1) // FOOD_SIZE * FOOD_SIZE
        new_y = randint(0, DISPLAY_HEIGHT - 1) // FOOD_SIZE * FOOD_SIZE
        food_coordinates.append([new_x, new_y])
    print(food_coordinates)

    snake_x = DISPLAY_WIDTH / 2
    snake_y = DISPLAY_HEIGHT / 2
    snake_rectangle = pygame.Rect(snake_x, snake_y, FOOD_SIZE, FOOD_SIZE)
    snake_x_change = FOOD_SIZE
    snake_y_change = 0
    game_over = False
    while not game_over:
        clock.tick(25)
        snake_display.fill(BG_COLOUR)
        snake = pygame.draw.rect(snake_display, SNAKE_COLOUR, snake_rectangle)
        food_list = draw_food(snake_display, food_coordinates)
        pygame.display.update()
        for event in pygame.event.get():
            # print(event)
            if event.type == pygame.QUIT:
                game_over = True
            elif event.type == pygame.KEYDOWN:
                if event.key == pygame.K_UP:
                    snake_x_change = 0
                    snake_y_change = -FOOD_SIZE
                elif event.key == pygame.K_DOWN:
                    snake_x_change = 0
                    snake_y_change = FOOD_SIZE
                elif event.key == pygame.K_LEFT:
                    snake_x_change = -FOOD_SIZE
                    snake_y_change = 0
                elif event.key == pygame.K_RIGHT:
                    snake_x_change = FOOD_SIZE
                    snake_y_change = 0
                print("move by {}, {}".format(snake_x_change, snake_y_change))
        snake_rectangle = snake_rectangle.move(snake_x_change, snake_y_change)
        if snake.x < 0:
            snake_rectangle.update(DISPLAY_WIDTH, snake.y, FOOD_SIZE, FOOD_SIZE)
        elif snake.x > DISPLAY_WIDTH:
            snake_rectangle.update(0, snake.y, FOOD_SIZE, FOOD_SIZE)
        elif snake.y < 0:
            snake_rectangle.update(snake.x, DISPLAY_HEIGHT, FOOD_SIZE, FOOD_SIZE)
        elif snake.y > DISPLAY_WIDTH:
            snake_rectangle.update(snake.x, 0, FOOD_SIZE, FOOD_SIZE)

        if snake.collidelist(food_list) >= 0:
            food_coordinates.pop(snake.collidelist(food_list))


if __name__ == "__main__":
    main()