使用pygame制作一个贪吃蛇的小游戏

之前我们已经学习了如果使用pygame创建一个窗口，现在我们来学习使用pygame来制作一个经典的小游戏–贪吃蛇。

• 首先我们需要导入待使用的模块：

import pygame, sys, random
from pygame.locals import *

pygame.locals 模块包含pygame使用的各种常量，它的内容会被自动放入到 Pygame 模块的名字空间中。

• 模块导入之后我们可以先定义好会用到的颜色，根据大家的喜好定义：

# 定义颜色
pinkColor = pygame.Color(255, 182, 193)
blackColor = pygame.Color(0, 0, 0)
whiteColor = pygame.Color(255, 255, 255)

pygame.Color()是用于描述颜色的对象，

Color(name) -> Color
Color(r, g, b , a) -> Color
Color(rgbvalue) –>Color

# Color对象的方法&属性
pygame.Color.r  ：获取或者设置Color对象的红色值
pygame.Color.g ：获取或者设置Color对象的绿色值
pygame.Color.b ：获取或者设置Color对象的蓝色值
pygame.Color.a ：获取或者设置Color对象的alpha值
pygame.Color.cmy ：获取或者设置Color对象的cmy值
pygame.Color.hsva ：获取或者设置Color对象的hsav值
pygame.Color.hsla ：获取或者设置Color对象的hsla值
pygame.Color.i 1i2i3 ：获取或者设置Color对象的I1I2I3描述
pygame.Color.normalize ： 返回一个Color对象的RGBA（显示通道）值
pygame.Color.correct gamma ：Color对象请求一个确定gamma值
pygame.Color.set length  ：设置在Color对象中元素的数值为1，2，3，或4

• 当游戏结束时，我们需要退出这个游戏，所我们需要给游戏定义一个用于退出的函数，很简单，就是先退出pygame窗口，再退出程序：

# 定义游戏结束的函数
def gameover():
    # 退出pygame窗口
    pygame.quit()
    # 退出程序
    sys.exit()

• 定义好结束函数之后，我们需要定义一个入口函数，用于进入游戏，游戏的主要代码都写在这里面：

def main():
    # 初始化
    pygame.init()
    # 定义一个变量来控制速度
    time_clock = pygame.time.Clock()

    # 创建窗口，定义标题
    screen = pygame.display.set_mode((640, 480))
    pygame.display.set_caption("贪吃蛇")

首先我们需要初始化pygame，创建好游戏窗口，顺便定义一个用来控制速度的变量，这个变量用于贪吃蛇的移动。

• 接着初始化一些贪吃蛇和食物用到的变量，将整个界面看成许多20x20的小方块，每个方块代表一个单位

    # 定义蛇的初始化变量
    snakePosition = [100, 100]  # 蛇头位置
    # 定义一个贪吃蛇的长度列表，其中有几个元素就代表有几段身体，这里我们定义5段身体
    snakeSegments = [[100, 100], [80, 100], [60, 100], [40, 100], [20, 100]]

    # 初始化食物位置
    foodPostion = [300, 300]
   
    # 食物数量，0表示被吃了，1表示没被吃
    foodTotal = 1
    
    # 初始方向，向右
    direction = 'right'
    # 定义一个改变方向的变量，按键
    changeDirection = direction

• 初始化好数据之后，使用while循环来侦听事件，通过不断地循环使蛇不断地前进

    while True:
        # 从队列中获取事件
        for event in pygame.event.get():
            # 判断是否为退出事件
            if event.type == QUIT:
                pygame.quit()
                sys.exit()
            # 按键事件
            elif event.type == KEYDOWN:
                # 如果是右键头或者是d，蛇向右移动
                if event.key == K_RIGHT or event.key == K_d:
                    changeDirection = 'right'
                # 如果是左键头或者是a，蛇向左移动
                if event.key == K_LEFT or event.key == K_a:
                    changeDirection = 'left'
                if event.key == K_UP or event.key == K_w:
                    changeDirection = 'up'
                if event.key == K_DOWN or event.key == K_s:
                    changeDirection = 'down'
                # 对应键盘上的Esc键，表示退出
                if event.key == K_ESCAPE:
                    pygame.event.post(pygame.event.Event(QUIT))

KEYDOWN是键盘按键事件，而K_RIGHT 、K_LEFT 、K_d、K_a等这些表示对应键盘上的按键。

• 确认蛇的移动方向，不能向反方向移动，比如此时蛇是向右移动的，则不能控制它向左移动，只向上或向下

        # 确认方向，判断是否输入了反方向运动
        if changeDirection == 'right' and not direction == 'left':
            direction = changeDirection
        if changeDirection == 'left' and not direction == 'right':
            direction = changeDirection
        if changeDirection == 'up' and not direction == 'down':
            direction = changeDirection
        if changeDirection == 'down' and not direction == 'up':
            direction = changeDirection

• 通过像素的加减确定蛇头的移动向上或向下加减20px相当于向上下移动一步

        # 根据方向移动蛇头
        if direction == 'right':
            snakePosition[0] += 20
        if direction == 'left':
            snakePosition[0] -= 20
        if direction == 'up':
            snakePosition[1] -= 20
        if direction == 'down':
            snakePosition[1] += 20

        # 增加蛇的长度
        snakeSegments.insert(0, list(snakePosition))
        # 判断是否吃到食物
        if snakePosition[0] == foodPostion[0] and snakePosition[1] == foodPostion[1]:
            foodTotal = 0
        else:
            snakeSegments.pop()  # 每次将最后一单位蛇身剔除列表
        # 如果食物为0 重新生成食物
        if foodTotal == 0:
            x = random.randrange(1, 32)
            y = random.randrange(1, 24)
            foodPostion = [int(x * 20), int(y * 20)]
            foodTotal = 1
        # 绘制pygame显示层
        screen.fill(blackColor)

• 设置蛇与食物的颜色长宽

        for position in snakeSegments:  # 蛇身为白色
            # 化蛇
            pygame.draw.rect(screen, pinkColor, Rect(position[0], position[1], 20, 20))
            pygame.draw.rect(screen, whiteColor, Rect(foodPostion[0], foodPostion[1], 20, 20))

• 更新显示到屏幕表面

pygame.display.flip()

• 判断游戏是否结束

         # 判断游戏是否结束
        if snakePosition[0] > 620 or snakePosition[0] < 0:
            gameover()
        elif snakePosition[1] > 460 or snakePosition[1] < 0:
            gameover()
        # 如果碰到自己的身体
        for body in snakeSegments[1:]:
            if snakePosition[0] == body[0] and snakePosition[1] == body[1]:
                gameover()

        # 控制游戏速度
        time_clock.tick(5)

• 入口函数

if __name__ == '__main__':
    main()

然后可以运行代码，如下图

完整代码

import pygame, sys, random
from pygame.locals import *


# 定义颜色
pinkColor = pygame.Color(255, 182, 193)
blackColor = pygame.Color(0, 0, 0)
whiteColor = pygame.Color(255, 255, 255)


# 定义游戏结束的函数
def gameover():
    pygame.quit()
    sys.exit()


def main():
    # 初始化
    pygame.init()
    # 定义一个变量来控制速度
    time_clock = pygame.time.Clock()

    # 创建窗口，定义标题
    screen = pygame.display.set_mode((640, 480))
    pygame.display.set_caption("贪吃蛇")

    # 定义蛇的初始化变量
    snakePosition = [100, 100]  # 蛇头位置
    # 定义一个贪吃蛇的长度列表，其中有几个元素就代表有几段身体
    snakeSegments = [[100, 100], [80, 100], [60, 100], [40, 100], [20, 100]]
    # 初始化食物位置
    foodPostion = [300, 300]
    # 食物数量，1是没被吃，0是被吃了
    foodTotal = 1
    # 初始方向，向右
    direction = 'right'
    # 定义一个改变方向的变量，按键
    changeDirection = direction

    # 通过键盘控制蛇的运动
    while True:
        # 从队列中获取事件
        for event in pygame.event.get():
            # 判断是否为退出事件
            if event.type == QUIT:
                pygame.quit()
                sys.exit()
            # 按键事件
            elif event.type == KEYDOWN:
                # 如果是右键头或者是d，蛇向右移动
                if event.key == K_RIGHT or event.key == K_d:
                    changeDirection = 'right'
                # 如果是左键头或者是a，蛇向左移动
                if event.key == K_LEFT or event.key == K_a:
                    changeDirection = 'left'
                if event.key == K_UP or event.key == K_w:
                    changeDirection = 'up'
                if event.key == K_DOWN or event.key == K_s:
                    changeDirection = 'down'
                # 对应键盘上的Esc键，表示退出
                if event.key == K_ESCAPE:
                    pygame.event.post(pygame.event.Event(QUIT))

        # 确认方向，判断是否输入了反方向运动
        if changeDirection == 'right' and not direction == 'left':
            direction = changeDirection
        if changeDirection == 'left' and not direction == 'right':
            direction = changeDirection
        if changeDirection == 'up' and not direction == 'down':
            direction = changeDirection
        if changeDirection == 'down' and not direction == 'up':
            direction = changeDirection

        # 根据方向移动蛇头
        if direction == 'right':
            snakePosition[0] += 20
        if direction == 'left':
            snakePosition[0] -= 20
        if direction == 'up':
            snakePosition[1] -= 20
        if direction == 'down':
            snakePosition[1] += 20

        # 增加蛇的长度
        snakeSegments.insert(0, list(snakePosition))
        # 判断是否吃到食物
        if snakePosition[0] == foodPostion[0] and snakePosition[1] == foodPostion[1]:
            foodTotal = 0
        else:
            snakeSegments.pop()  # 每次将最后一单位蛇身剔除列表

        # 如果食物为0 重新生成食物
        if foodTotal == 0:
            x = random.randrange(1, 32)
            y = random.randrange(1, 24)
            foodPostion = [int(x * 20), int(y * 20)]
            foodTotal = 1

        # 绘制pygame显示层
        screen.fill(blackColor)


        for position in snakeSegments:  # 蛇身为白色
            # 化蛇
            pygame.draw.rect(screen, pinkColor, Rect(position[0], position[1], 20, 20))
            pygame.draw.rect(screen, whiteColor, Rect(foodPostion[0], foodPostion[1], 20, 20))

        # 更新显示到屏幕表面
        pygame.display.flip()

        # 判断游戏是否结束
        if snakePosition[0] > 620 or snakePosition[0] < 0:
            gameover()
        elif snakePosition[1] > 460 or snakePosition[1] < 0:
            gameover()
        # 如果碰到自己的身体
        for body in snakeSegments[1:]:
            if snakePosition[0] == body[0] and snakePosition[1] == body[1]:
                gameover()

        # 控制游戏速度
        time_clock.tick(5)



#  启动入口函数
if __name__ == '__main__':
    main()