帮我生成一个贪吃蛇的代码
时间: 2023-05-24 07:06:39 浏览: 129
好的,以下是一个简单的贪吃蛇代码:
```
import pygame, random
# 定义窗口大小
windowWidth = 640
windowHeight = 480
gridSize = 20
# 初始化pygame
pygame.init()
# 创建窗口
screen = pygame.display.set_mode((windowWidth, windowHeight))
pygame.display.set_caption("贪吃蛇")
# 定义颜色
black = (0, 0, 0)
white = (255, 255, 255)
red = (255, 0, 0)
green = (0, 255, 0)
# 定义字体
font = pygame.font.SysFont(None, 30)
# 定义贪吃蛇的初始位置和大小
snakePos = [100, 50]
snakeBody = [[100, 50], [80, 50], [60, 50]]
snakeLength = 3
# 定义食物的初始位置
foodPos = [random.randrange(1, (windowWidth/gridSize)) * gridSize,
random.randrange(1, (windowHeight/gridSize)) * gridSize]
# 定义初始方向
direction = "RIGHT"
changeDirection = direction
# 定义运行主循环的变量
gameOver = False
# 定义绘制网格的函数
def drawGrid():
for x in range(0, windowWidth, gridSize):
for y in range(0, windowHeight, gridSize):
rect = pygame.Rect(x, y, gridSize, gridSize)
pygame.draw.rect(screen, white, rect, 1)
# 定义绘制贪吃蛇的函数
def drawSnake(snakeBody):
for pos in snakeBody:
rect = pygame.Rect(pos[0], pos[1], gridSize, gridSize)
pygame.draw.rect(screen, green, rect)
# 定义绘制食物的函数
def drawFood(foodPos):
rect = pygame.Rect(foodPos[0], foodPos[1], gridSize, gridSize)
pygame.draw.rect(screen, red, rect)
# 定义显示文字的函数
def showMessage(msg, color):
text = font.render(msg, True, color)
screen.blit(text, [windowWidth/4, windowHeight/4])
# 主循环
while not gameOver:
# 处理事件
for event in pygame.event.get():
if event.type == pygame.QUIT:
gameOver = True
elif event.type == pygame.KEYDOWN:
if event.key == pygame.K_RIGHT:
changeDirection = "RIGHT"
elif event.key == pygame.K_LEFT:
changeDirection = "LEFT"
elif event.key == pygame.K_UP:
changeDirection = "UP"
elif event.key == pygame.K_DOWN:
changeDirection = "DOWN"
# 确定贪吃蛇移动的方向
if changeDirection == "RIGHT" and direction != "LEFT":
direction = changeDirection
elif changeDirection == "LEFT" and direction != "RIGHT":
direction = changeDirection
elif changeDirection == "UP" and direction != "DOWN":
direction = changeDirection
elif changeDirection == "DOWN" and direction != "UP":
direction = changeDirection
# 根据方向移动贪吃蛇
if direction == "RIGHT":
snakePos[0] += gridSize
elif direction == "LEFT":
snakePos[0] -= gridSize
elif direction == "UP":
snakePos[1] -= gridSize
elif direction == "DOWN":
snakePos[1] += gridSize
# 判断贪吃蛇是否吃到了食物
if snakePos == foodPos:
foodPos = [random.randrange(1, (windowWidth/gridSize)) * gridSize,
random.randrange(1, (windowHeight/gridSize)) * gridSize]
snakeLength += 1
# 添加新的贪吃蛇身体部分
snakeHead = [snakePos[0], snakePos[1]]
snakeBody.insert(0, snakeHead)
if len(snakeBody) > snakeLength:
snakeBody.pop()
# 绘制游戏界面
screen.fill(black)
drawGrid()
drawSnake(snakeBody)
drawFood(foodPos)
# 判断贪吃蛇是否撞墙
if snakePos[0] < 0 or snakePos[0] >= windowWidth or snakePos[1] < 0 or snakePos[1] >= windowHeight:
showMessage("Game Over!", red)
pygame.display.update()
pygame.time.wait(3000)
gameOver = True
# 判断贪吃蛇是否撞到自己
for block in snakeBody[1:]:
if snakePos == block:
showMessage("Game Over!", red)
pygame.display.update()
pygame.time.wait(3000)
gameOver = True
# 更新屏幕
pygame.display.update()
# 退出pygame
pygame.quit()
```