Python_snake_classless/snake.py at master · cafe-barrita/Python_snake_classless · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
import pygame
import random
import sys
from pygame.locals import *

DIR = {"up": (0, -1), "down": (0, 1), "right": (1, 0), "left": (-1, 0)}
RED = (255, 0, 0)
GREEN = (0, 255, 0)

N_CUADROS = 40
RAD = 6


def move_snake(snake):
    global key_pressed
    if snake["dir"] != "right" and K_LEFT == key_pressed:
        snake["dir"] = "left"
    elif snake["dir"] != "left" and K_RIGHT == key_pressed:
        snake["dir"] = "right"
    elif snake["dir"] != "down" and K_UP == key_pressed:
        snake["dir"] = "up"
    elif snake["dir"] != "up" and K_DOWN == key_pressed:
        snake["dir"] = "down"
    snake["rings"].pop()
    x = snake["rings"][0][0] + 2 * RAD * DIR[snake["dir"]][0]
    y = snake["rings"][0][1] + 2 * RAD * DIR[snake["dir"]][1]
    snake["rings"].insert(0, (x, y))


def enlarge_snake(snake):
    x = snake["rings"][0][0] + 2 * RAD * DIR[snake["dir"]][0]
    y = snake["rings"][0][1] + 2 * RAD * DIR[snake["dir"]][1]
    snake["rings"].insert(0, (x, y))


def generate_food():
    x = random.randrange(1, N_CUADROS - 1)
    y = random.randrange(1, N_CUADROS - 1)
    return 2 * RAD * x, 2 * RAD * y


def check_food(snake):
    global food
    if snake["rings"][0] == food:
        enlarge_snake(snake)
        food = generate_food()


def draw(screen, snake, food):
    screen.blit(pygame.Surface(screen.get_size()), (0, 0))
    pygame.draw.rect(screen, RED, Rect(food, (2 * RAD, 2 * RAD)))
    for ring in snake["rings"]:
        pygame.draw.circle(screen, GREEN, (ring[0] + RAD, ring[1] + RAD), RAD)


def check_collisions(snake):
    head = snake["rings"][0]
    if head[0] < 0 or head[0] > N_CUADROS * 2 * RAD:
        return True
    if head[1] < 0 or head[1] > N_CUADROS * 2 * RAD:
        return True
    for ring in snake["rings"][1:]:
        if head == ring:
            return True

    return False


def game_over(screen):
    font = pygame.font.SysFont('Arial', RAD * 8, bold=True)
    text = font.render('Game Over', 0, RED)
    pos = (7 * 2 * RAD, (N_CUADROS - 4) * RAD)
    screen.blit(text, pos)


pygame.init()
clock = pygame.time.Clock()
screen = pygame.display.set_mode((N_CUADROS * 2 * RAD, N_CUADROS * 2 * RAD))

snake = {"dir": "right", "rings":
    [(12 * 2 * RAD, 13 * 2 * RAD), (12 * 2 * RAD, 12 * 2 * RAD), (12 * 2 * RAD, 11 * 2 * RAD),
     (12 * 2 * RAD, 10 * 2 * RAD)]}
food = generate_food()

draw(screen, snake, food)
pygame.display.flip()

key_pressed = None

while True:

    clock.tick(10)

    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()
            sys.exit(0)
        if event.type == KEYDOWN:
            key_pressed = event.key
        if event.type == KEYUP:
            key_pressed = None

    if check_collisions(snake):
        game_over(screen)
    else:
        check_food(snake)
        move_snake(snake)
        draw(screen, snake, food)

    pygame.display.flip()