给我写一段PY代码完成一段贪吃蛇游戏

以下是一个简单的贪吃蛇游戏的Python代码示例：

import pygame
import random

# 初始化游戏
pygame.init()

# 游戏窗口大小
window_width = 800
window_height = 600

# 定义颜色
black = pygame.Color(0, 0, 0)
white = pygame.Color(255, 255, 255)
red = pygame.Color(255, 0, 0)
green = pygame.Color(0, 255, 0)
blue = pygame.Color(0, 0, 255)

# 创建游戏窗口
window = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption('贪吃蛇游戏')

# 设置游戏时钟
clock = pygame.time.Clock()

# 定义蛇的初始位置和大小
snake_position = [100, 50]
snake_body = [[100, 50], [90, 50], [80, 50]]
snake_direction = 'RIGHT'

# 定义食物的初始位置
food_position = [random.randrange(1, (window_width // 10)) * 10,
                 random.randrange(1, (window_height // 10)) * 10]
food_spawn = True

# 控制游戏结束的变量
game_over = False

# 游戏主循环
while not game_over:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            game_over = True
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_RIGHT:
                snake_direction = 'RIGHT'
            if event.key == pygame.K_LEFT:
                snake_direction = 'LEFT'
            if event.key == pygame.K_UP:
                snake_direction = 'UP'
            if event.key == pygame.K_DOWN:
                snake_direction = 'DOWN'

    # 根据方向移动蛇的位置
    if snake_direction == 'RIGHT':
        snake_position[0] += 10
    if snake_direction == 'LEFT':
        snake_position[0] -= 10
    if snake_direction == 'UP':
        snake_position[1] -= 10
    if snake_direction == 'DOWN':
        snake_position[1] += 10

    # 增加蛇的身体长度
    snake_body.insert(0, list(snake_position))
    if snake_position[0] == food_position[0] and snake_position[1] == food_position[1]:
        food_spawn = False
    else:
        snake_body.pop()

    # 重新生成食物
    if not food_spawn:
        food_position = [random.randrange(1, (window_width // 10)) * 10,
                         random.randrange(1, (window_height // 10)) * 10]
    food_spawn = True

    # 绘制游戏窗口
    window.fill(black)
    for pos in snake_body:
        pygame.draw.rect(window, green, pygame.Rect(
            pos[0], pos[1], 10, 10))
    pygame.draw.rect(window, white, pygame.Rect(
        food_position[0], food_position[1], 10, 10))

    # 游戏结束条件：碰壁或自身
    if snake_position[0] < 0 or snake_position[0] > window_width-10:
        game_over = True
    if snake_position[1] < 0 or snake_position[1] > window_height-10:
        game_over = True
    for block in snake_body[1:]:
        if snake_position[0] == block[0] and snake_position[1] == block[1]:
            game_over = True

    # 刷新游戏窗口
    pygame.display.update()
    clock.tick(15)

# 退出游戏
pygame.quit()

这段代码使用了pygame库来实现游戏窗口、蛇的移动、食物生成和碰撞检测等功能。通过键盘事件来控制蛇的移动方向，使用循环更新蛇的位置和绘制游戏窗口。游戏结束的条件包括蛇碰壁和自身碰撞