pySnake/game.py at master · aghinsa/pySnake · 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
111
112
113
114
import pygame
import time

from pygame.locals import *
from dataclasses import dataclass
from typing import Any,Tuple
from random import randint

@dataclass
class GameConfig:
    height : int
    width : int
    player : Any
    food : Any
    player_size : Tuple[int]
    food_size : Tuple[int]

class Game:
    def __init__(self,config:GameConfig)->None:
        self.config = config
        self.window_size = (self.config.width,self.config.height)
        self.player = self.config.player(5,self.window_size)
        self._running = True
        self.food = self.config.food(5,5) # setting init position

    def on_init(self):
        pygame.init()
        self.display = pygame.display.set_mode(
            self.window_size,
            pygame.HWSURFACE
        )
        pygame.display.set_caption('Snake')
        self._running=True
        self.snake_body = pygame.Surface( self.config.player_size )

        self.food_img = pygame.Surface( self.config.food_size )

    def on_event(self,event):
        if event.type == QUIT:
            self._running=False

    def spawn_food(self):
        step = self.food.step
        nx=randint(2,10)*step
        ny=randint(2,10)*step
        while((nx,ny) in zip(self.player.x,self.player.y)):
            nx=randint(2,10)*step
            ny=randint(2,10)*step
        self.food.position=(nx,ny)


    def on_loop(self):
        self.player.update()

        # check collison with food
        head = self.snake_body.get_rect(topleft=self.player.position)
        food_pos = self.food_img.get_rect(topleft=self.food.position)


        if( head.colliderect(food_pos) ):
            self.player.length = self.player.length+1
            self.player.eat(food_pos)
            step = self.food.step
            self.spawn_food()

        #check collision with self
        for _pos in zip(self.player.x[1:],self.player.y[2:]) :
            head = pygame.Rect(self.player.position,(1,1))
            _pos = pygame.Rect(_pos,(1,1))


            if( head.colliderect(_pos) ):
                print(f"Game Over!!\nScore : {self.player.length}")
                exit(0)

    def on_render(self):
        self.display.fill((255,255,255))
        self.player.draw(self.display,self.config.player_size)
        self.food.draw(self.display,self.config.food_size)
        pygame.display.flip()

    def on_cleanup(self):
        pygame.quit()

    def on_execute(self):
        if (self.on_init() == False):
            self._running = False

        while(self._running):
            pygame.event.pump()
            keys = pygame.key.get_pressed()

            if(keys[K_RIGHT]):
                self.player.moveRight()

            if(keys[K_LEFT]):
                self.player.moveLeft()

            if(keys[K_UP]):
                self.player.moveUp()

            if(keys[K_DOWN]):
                self.player.moveDown()

            if(keys[K_ESCAPE]):
                self._running=False

            self.on_loop()
            self.on_render()
            time.sleep(50/1000.0)

        print("Exiting")
        self.on_cleanup()