python_studies/snake.py at main · ODenteAzul/python_studies · 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
from turtle import Turtle

STARTING_POSITIONS = [(0,0),(-20,0),(-40,0)]
MOVE_PACE = 20
UP = 90
DOWN = 270
LEFT = 180
RIGHT = 0

class Snake:

    def __init__(self):
        self.segments = []
        self.create_snake()


    def create_snake(self):
        for position in STARTING_POSITIONS:
            new_seg = Turtle()
            new_seg.shape('square')
            new_seg.color('white')
            new_seg.penup()
            new_seg.goto(position)
            self.segments.append(new_seg)

        self.head = self.segments[0]


    def create_segment(self, pos):
        new_seg = Turtle()
        new_seg.shape('square')
        new_seg.color('white')
        new_seg.penup()
        new_seg.goto(pos)
        self.segments.append(new_seg)


    def move_snake(self, new):
        position = []
        prior_position = []
        first = True
        for seg in self.segments:
            if first:
                position = seg.pos()
                seg.forward(MOVE_PACE)
                if seg.pos()[0] < -320:
                    self.segments[0].goto(300, seg.pos()[1])
                elif seg.pos()[0] > 320:
                    self.segments[0].goto(-300, seg.pos()[1])
                elif  seg.pos()[1] < -320:
                    self.segments[0].goto(seg.pos()[0], 300)
                elif seg.pos()[1] > 320:
                    self.segments[0].goto(seg.pos()[0], -300)
                first = False
            else:
                prior_position = seg.pos()
                seg.goto(position)
                position = prior_position
        if new:
            self.create_segment(position)


    def change_up(self):
        if self.segments[0].heading() != DOWN :
            self.segments[0].setheading(UP)


    def change_down(self):
        if self.segments[0].heading() != UP :
            self.segments[0].setheading(DOWN)


    def change_left(self):
        if self.segments[0].heading() != RIGHT :
            self.segments[0].setheading(LEFT)


    def change_right(self):
        if self.segments[0].heading() != LEFT :
            self.segments[0].setheading(RIGHT)