snake_game/snake.py at master · tessied/snake_game · 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
# A module to position, move, and extend the snake

from turtle import Turtle
MOVE_STEPS = 20


class Snake:

    def __init__(self):
        self.snake_body = []
        self.create()
        self.head = self.snake_body[0]

    def create(self):
        for index in range(3):
            self.add_length((0 - index * 20, 0))

    def reset(self):
        for segment in self.snake_body:
            segment.goto(5000, 5000)
        self.snake_body.clear()
        self.create()
        self.head = self.snake_body[0]

    def move(self):
        for index in range(len(self.snake_body) - 1, 0, -1):
            self.snake_body[index].goto(self.snake_body[index - 1].pos())
        self.head.forward(MOVE_STEPS)

    def add_length(self, position):
        new_turtle = Turtle(shape="square")
        new_turtle.pu()
        new_turtle.color("white")
        new_turtle.goto(position)
        self.snake_body.append(new_turtle)

    def extend(self):
        self.add_length(self.snake_body[-1].position())

    def up(self):
        if self.head.heading() != 270:
            self.head.setheading(90)

    def down(self):
        if self.head.heading() != 90:
            self.head.setheading(270)

    def left(self):
        if self.head.heading() != 0:
            self.head.setheading(180)

    def right(self):
        if self.head.heading != 180:
            self.head.setheading(0)