pygame贪吃蛇源码

使用 python pygame 模块实现的初代贪吃蛇源码；可更容易学习
更多示例欢迎访问网站：https://python-abc.xyz/

1. import sys
   
2. import random
   
3. import pygame
   
4. from pygame.math import Vector2
   
5. 
   
6. 
   
7. class Fruit:
   
8.     """水果类"""
   
9.     def __init__(self):
   
10.         self.randomize()
    
11. 
    
12.     def draw(self):
    
13.         """绘制水果"""
    
14.         rect = pygame.Rect(self.pos.x * cell_size, self.pos.y * cell_size, cell_size, cell_size)
    
15.         pygame.draw.rect(screen, fruit_color, rect)
    
16. 
    
17.     def randomize(self):
    
18.         """随机位置"""
    
19.         self.x = random.randint(0, cell_number - 1)
    
20.         self.y = random.randint(0, cell_number - 1)
    
21.         self.pos = Vector2(self.x, self.y)
    
22. 
    
23. 
    
24. class Snake:
    
25.     """贪吃蛇"""
    
26.     def __init__(self):
    
27.         self.body = [Vector2(5, 10), Vector2(4, 10), Vector2(3, 10)]
    
28.         self.direction = Vector2(1, 0)
    
29. 
    
30.     def draw(self):
    
31.         """绘制贪吃蛇"""
    
32.         for block in self.body:
    
33.             x_pos = int(block.x * cell_size)
    
34.             y_pos = int(block.y * cell_size)
    
35.             rect = pygame.Rect(x_pos, y_pos, cell_size, cell_size)
    
36.             pygame.draw.rect(screen, snake_color, rect)
    
37. 
    
38.     def move(self):
    
39.         """移动"""
    
40.         self.eat()
    
41.         self.body.pop()
    
42. 
    
43.     def eat(self):
    
44.         """吃到水果"""
    
45.         self.body.insert(0, self.body[0] + self.direction)
    
46. 
    
47. 
    
48. class Main:
    
49.     def __init__(self):
    
50.         self.snake = Snake()
    
51.         self.fruit = Fruit()
    
52. 
    
53.     def update(self):
    
54.         self.snake.move()
    
55.         self.check_eat()
    
56.         self.check_fail()
    
57. 
    
58.     def draw_elements(self):
    
59.         self.fruit.draw()
    
60.         self.snake.draw()
    
61. 
    
62.     def check_eat(self):
    
63.         """判断是否吃到了水果"""
    
64.         if self.fruit.pos == self.snake.body[0]:
    
65.             self.fruit.randomize()
    
66.             self.snake.eat()
    
67. 
    
68.     def check_fail(self):
    
69.         """失败检测
    
70.         1. 蛇头撞墙失败
    
71.         2. 蛇头吃到自己失败
    
72.         """
    
73.         if not 0 <= self.snake.body[0].x < cell_number or not 0 <= self.snake.body[0].y < cell_number:
    
74.             print('撞墙了')
    
75.             self.game_over()
    
76. 
    
77.         for block in self.snake.body[1:]:
    
78.             if block == self.snake.body[0]:
    
79.                 print('吃到自己了')
    
80.                 self.game_over()
    
81. 
    
82.     @staticmethod
    
83.     def game_over():
    
84.         pygame.quit()
    
85.         sys.exit()
    
86. 
    
87. 
    
88. screen_color = (175, 215, 70)
    
89. fruit_color = (126, 166, 114)
    
90. snake_color = (183, 111, 122)
    
91. cell_size = 40
    
92. cell_number = 20
    
93. 
    
94. pygame.init()
    
95. screen = pygame.display.set_mode((cell_number * cell_size, cell_number * cell_size))
    
96. clock = pygame.time.Clock()
    
97. 
    
98. SCREEN_UPDATE = pygame.USEREVENT
    
99. pygame.time.set_timer(SCREEN_UPDATE, 200)
    
100. 
     
101. main = Main()
     
102. 
     
103. while True:
     
104.     for event in pygame.event.get():
     
105.         if event.type == pygame.QUIT:
     
106.             main.game_over()
     
107.         if event.type == SCREEN_UPDATE:
     
108.             main.update()
     
109.         if event.type == pygame.KEYDOWN:
     
110.             if event.key == pygame.K_UP:
     
111.                 if main.snake.direction.y != 1:
     
112.                     main.snake.direction = Vector2(0, -1)
     
113.             if event.key == pygame.K_RIGHT:
     
114.                 if main.snake.direction.x != -1:
     
115.                     main.snake.direction = Vector2(1, 0)
     
116.             if event.key == pygame.K_DOWN:
     
117.                 if main.snake.direction.y != -1:
     
118.                     main.snake.direction = Vector2(0, 1)
     
119.             if event.key == pygame.K_LEFT:
     
120.                 if main.snake.direction.x != 1:
     
121.                     main.snake.direction = Vector2(-1, 0)
     
122. 
     
123.     screen.fill(screen_color)
     
124.     main.draw_elements()
     
125.     pygame.display.update()
     
126.     clock.tick(60)
     

复制代码