pygame 贪吃蛇

我不是女神ヾ 2022-05-29 04:55 263阅读 0赞

import pygame, random
    
    
    BLACK = [0, 0, 0]
    WHITE = [255, 255, 255]
    RED = [255, 0, 0]
    GREEN = [0, 255, 0]
    YELLOW = [247, 148, 29]
    UP = 0
    RIGHT = 1
    DOWN = 2
    LEFT = 3
    class GameObject(object):
        def __init__(self, x, y, color = RED):
            self._x = x
            self._y = y
            self._color = color
        @property
        def x(self):
            return self._x
        @property
        def y(self):
            return self._y
        @property
        def color(self):
            return self._color
    
    
    class SnakeNode(GameObject):
        def __init__(self, x, y, size):
            super().__init__(x, y)
            self._size = size
            self._color = GREEN
        @property
        def size(self):
            return self._size
        def draw(self, screen):
            pygame.draw.rect(screen, self._color, [self._x, self._y, self._size, self._size], 0)
            pygame.draw.rect(screen, BLACK, [self._x, self._y, self._size, self._size], 1)
    
    class Snake(GameObject):
        def __init__(self, x, y, size):
            super().__init__(x, y)
            self._size = size
            self._Nodes = []
            self._sdir = LEFT
            self.grow = False
            for index in range(5):
                node = SnakeNode(self._x + 20 * index, self._y, self._size)
                self._Nodes.append(node)
        @property
        def head(self):
            return self._Nodes[0]
        @property
        def sdir(self):
            return self._sdir
    
        @sdir.setter
        def sdir(self, new_dir):
            if (self._sdir + new_dir) % 2 != 0:
                self._sdir = new_dir
    
        def draw(self, screen):
            for node in self._Nodes:
                node.draw(screen)
    
        def move(self):
            snake_dir = self._sdir
            x, y, size = self.head.x, self.head.y, self.head.size
            if snake_dir == UP:         # 0
                y -= 20
            elif snake_dir == RIGHT:    # 1
                x += 20
            elif snake_dir == DOWN:     # 2
                y += 20
            elif snake_dir == LEFT:     # 3
                x -= 20
            new_head = SnakeNode(x, y, size)
            self._Nodes.insert(0, new_head)
            if not self.grow:
                self._Nodes.pop()
    
    
    
    
        def collision(self, wall):
            return self.head.x < wall.x or self.head.x + self._size> wall.x + wall.width or self.head.y < wall.y or self.head.y + self._size > wall.y + wall.height
    
        def eat(self, food):
            print(food.x, food.y, '食物')
            print(self.head.x, self.head.y, '蛇')
            return self.head._x == food.x and self.head._y ==food.y
    
    
    
    
    
    
    
    class Food(GameObject):
        def __init__(self, x, y, size):
            super().__init__(x,y)
            self._size = size
            self._is_red = True
    
        def draw(self, screen):
            if self._is_red:
                pygame.draw.circle(screen, RED, [self._x, self._y], self._size, 0)
                self._is_red = False
            else:
                pygame.draw.circle(screen, WHITE, [self._x, self._y], self._size, 0)
                self._is_red = True
    
    
    
    
    
    class Wall(GameObject):
        def __init__(self, x, y, width, height):
            super().__init__(x, y)
            self._width = width
            self._height = height
        @property
        def width(self):
            return self._width
        @property
        def height(self):
            return self._height
    
        def draw(self, screen):
            pygame.draw.rect(screen, YELLOW, [self._x, self._y, self._width, self._height], 5)
    
    
    
    
    def main():
        def refresh():
            screen.fill(WHITE)
            food.draw(screen)
            wall.draw(screen)
            snake.draw(screen)
            snake.move()
            snake.grow = False
            pygame.display.flip()
            clock.tick(10)
        pygame.display.init()
        screen = pygame.display.set_mode([640, 800])
        pygame.display.set_caption('贪吃蛇')
        screen.fill(WHITE)
        wall = Wall(10, 10, 620, 620)
        food = Food(170, 170, 10)
        snake = Snake(310, 310, 20)
    
        clock = pygame.time.Clock()
        running = True
        while running:
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    running = False
                elif event.type == pygame.KEYDOWN:
                    key = event.key
                    if key == pygame.K_w:
                        snake.sdir = UP
                    elif key == pygame.K_d:
                        snake.sdir = RIGHT
                    elif key == pygame.K_s:
                        snake.sdir = DOWN
                    elif key == pygame.K_a:
                        snake.sdir = LEFT
                    elif key == pygame.K_F2:
                        del snake
                        snake = Snake(310, 310, 20)
                    else:
                        pass
            if not snake.collision(wall):
                refresh()
                if snake.eat(food):
                    del food
                    snake.grow = True
                    x1, y1, size1 = 20 * random.randint(1, 31) + 10, 20 * random.randint(1, 31) + 10, 10
                    food = Food(x1, y1, size1)
    
    
    
        pygame.display.quit()
    
    
    if __name__ == '__main__':
        main()