用Python实现贪吃蛇双人大战

Python贪吃蛇双人大战
晚上家里小朋友要玩贪吃蛇游戏，还要跟我对战，一时半会我没想到去哪里下这样一个游戏，忽然灵机一动，可以自己写一个，顺便还可以跟小朋友展示一下程序员的厉害，于是开工。

原始版本
这是一个很基础的程序，自然不用从头写，在网上随便一搜，找到有人共享的代码，链接如下：https://www.cnblogs.com/qiu2013/p/6087627.html，据说是来源于《Raspberry Pi 用户指南》的代码，我也没有去查。代码如下：

#!/usr/bin/env python
import pygame,sys,time,random
from pygame.locals import *
# 定义颜色变量
redColour = pygame.Color(255,0,0)
blackColour = pygame.Color(0,0,0)
whiteColour = pygame.Color(255,255,255)
greyColour = pygame.Color(150,150,150)

# 定义gameOver函数
def gameOver(playSurface):
    gameOverFont = pygame.font.Font('arial.ttf',72)
    gameOverSurf = gameOverFont.render('Game Over', True, greyColour)
    gameOverRect = gameOverSurf.get_rect()
    gameOverRect.midtop = (320, 10)
    playSurface.blit(gameOverSurf, gameOverRect)
    pygame.display.flip()
    time.sleep(5)
    pygame.quit()
    sys.exit()

# 定义main函数
def main():
    # 初始化pygame
    pygame.init()
    fpsClock = pygame.time.Clock()
    # 创建pygame显示层
    playSurface = pygame.display.set_mode((640,480))
    pygame.display.set_caption('Raspberry Snake')

    # 初始化变量
    snakePosition = [100,100]
    snakeSegments = [[100,100],[80,100],[60,100]]
    raspberryPosition = [300,300]
    raspberrySpawned = 1
    direction = 'right'
    changeDirection = direction
    while True:
        # 检测例如按键等pygame事件
        for event in pygame.event.get():
            if event.type == QUIT:
                pygame.quit()
                sys.exit()
            elif event.type == KEYDOWN:
                # 判断键盘事件
                if event.key == K_RIGHT or event.key == ord('d'):
                    changeDirection = 'right'
                if event.key == K_LEFT or event.key == ord('a'):
                    changeDirection = 'left'
                if event.key == K_UP or event.key == ord('w'):
                    changeDirection = 'up'
                if event.key == K_DOWN or event.key == ord('s'):
                    changeDirection = 'down'
                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] == raspberryPosition[0] and snakePosition[1] == raspberryPosition[1]:
            raspberrySpawned = 0
        else:
            snakeSegments.pop()
        # 如果吃掉树莓，则重新生成树莓
        if raspberrySpawned == 0:
            x = random.randrange(1,32)
            y = random.randrange(1,24)
            raspberryPosition = [int(x*20),int(y*20)]
            raspberrySpawned = 1
        # 绘制pygame显示层
        playSurface.fill(blackColour)
        for position in snakeSegments:
            pygame.draw.rect(playSurface,whiteColour,Rect(position[0],position[1],20,20))
            pygame.draw.rect(playSurface,redColour,Rect(raspberryPosition[0], raspberryPosition[1],20,20))

        # 刷新pygame显示层
        pygame.display.flip()
        # 判断是否死亡
        if snakePosition[0] > 620 or snakePosition[0] < 0:
            gameOver(playSurface)
        if snakePosition[1] > 460 or snakePosition[1] < 0:
            for snakeBody in snakeSegments[1:]:
                if snakePosition[0] == snakeBody[0] and snakePosition[1] == snakeBody[1]:
                    gameOver(playSurface)
        # 控制游戏速度
        fpsClock.tick(5)

if __name__ == "__main__":
    main()

此代码实现了基本功能，主循环中先判断按键事件，然后调整蛇的位置，若蛇吃到了豆子（这个代码里叫树莓，我嫌名字太长，改成了习惯的豆子），则增加蛇的长度，并重新生成豆子，接着刷新显示，最后判断是否死亡，若死亡则调用gameOver。

当然这个是满足不了小朋友的需求的，小朋友尝试了一下，马上提取了如下需求：

要跟我一起玩，也就是要有两条蛇，每人控制一个，看谁吃得多。
蛇死了之后不要结束，太麻烦，改为重新开始。
蛇的颜色要能自己定。
要能看出来蛇头，即蛇头需要用不同的颜色。
豆子数量太少，每次才一个，要一下子出现很多豆子，可以随便吃。
看来天下用户都一样，总是各种需求。于是为了便于以后的修改，我把蛇相关的操作提取了一个蛇的类如下。
蛇类 

class Snake:    
    def __init__(self, color, headColor, ctrlKeys):
        self.color = color
        self.headColor = headColor
        self.ctrlKeys = ctrlKeys #按[上，下，左，右]的顺序
        self.direction = random.choice([-2,2,-1,1]) # 方向[-2,2,-1,1]分别表示[上，下，左，右]
        x = random.randrange(5,SCREEN_WIDTH/20-5)
        y = random.randrange(5,SCREEN_HEIGHT/20-5)
        self.headPos = [int(x*20),int(y*20)]
        self.segments = [self.headPos]
        self.moveAndAdd()
        self.moveAndAdd()      
    def changeDirection(self, pressKey):
        directions = [-2,2,-1,1]
        for direct, key in zip(directions, self.ctrlKeys):
            if key == pressKey and direct + self.direction != 0:
                self.direction = direct      
    def moveAndAdd(self):
        # 根据方向移动蛇头的坐标
        if self.direction == 1:
            self.headPos[0] += 20
        if self.direction == -1:
            self.headPos[0] -= 20
        if self.direction == -2:
            self.headPos[1] -= 20
        if self.direction == 2:
            self.headPos[1] += 20
        self.segments.insert(0,list(self.headPos)) # 在蛇头插入一格
    def pop(self):
        self.segments.pop()  # 在蛇尾减去一格
    def show(self, playSurface):
        # 画蛇身
        for pos in self.segments[1:]:
            pygame.draw.rect(playSurface,self.color,Rect(pos[0],pos[1],20,20))
        # 画蛇头
        pygame.draw.rect(playSurface,self.headColor,Rect(self.headPos[0],self.headPos[1],20,20))
    def respawnIfDead(self):
        if self.headPos[0] > SCREEN_WIDTH-20 or self.headPos[0] < 0 or self.headPos[1] > SCREEN_HEIGHT-20 or self.headPos[1] < 0:       
            x = random.randrange(5,SCREEN_WIDTH/20-5)
            y = random.randrange(5,SCREEN_HEIGHT/20-5)
            self.direction = random.choice([-2,2,-1,1])
            self.headPos = [int(x*20),int(y*20)]
            self.segments = [self.headPos]
            self.moveAndAdd()
            self.moveAndAdd()       

后边的请去原作者那里看

———————————————
版权声明：本文为CSDN博主「tyst08」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/tyst08/article/details/105521505