# pygame游戏库，sys操控python运行的环境

import pygame, sys, random

from pygame.locals import *

# 1,定义颜色变量,0-255  0黑色  255白色
redColor = pygame.Color(255, 0, 0)
# 背景为黑色
blackColor = pygame.Color(0, 0, 0)
# 贪吃蛇为白色
whiteColor = pygame.Color(255, 255, 255)

# 定义游戏结束的函数
def gameover():
    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('贪吃蛇')

    snakePosition = [100, 100]                          # 贪吃蛇初始坐标位置
    snakeBody = [[100, 100], [80, 100], [60, 100]]      # 初始化贪吃蛇的长度列表中有个元素就代表有几段身体
    targetPosition = [300, 300]                         # 初始化目标方向和位置

    # 判断是否吃掉了这个目标方块1 就是没有吃 0就是吃掉
    targetflag = 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:

                if event.key == K_d:
                    changeDirection = 'right'

                if event.key == K_a:
                    changeDirection = 'left'

                if event.key == K_w:
                    changeDirection = 'up'

                if event.key == K_s:
                    changeDirection = 'down'

                    # 对应键盘上的esc文件

                if event.key == K_ESCAPE:
                    pygame.event.post(pygame.event.Event(QUIT))

        # 确定方向

        if changeDirection == 'left' and not direction == 'right':
            direction = changeDirection

        if changeDirection == 'right' and not direction == 'left':
            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

        # 增加蛇的长度
        snakeBody.insert(0, list(snakePosition))

        # 如果贪吃蛇和目标方块的位置重合
        if snakePosition[0] == targetPosition[0] and snakePosition[1] == targetPosition[1]:

            targetflag = 0

        else:

            snakeBody.pop()

        if targetflag == 0:
            x = random.randrange(1, 32)

            y = random.randrange(1, 24)

            targetPosition = [int(x * 20), int(y * 20)]

            targetflag = 1

        # 填充背景颜色
        playsurface.fill(blackColor)

        for position in snakeBody:
            # 第一个参数serface指定一个serface编辑区，在这个区域内绘制。第二个参数color：颜色。
            # 第三个参数:rect:返回一个矩形(xy),(width,height)。第四个参数：width：表示线条的粗细  width0填充  实心
            pygame.draw.rect(playsurface, redColor, Rect(position[0], position[1], 20, 20))

            pygame.draw.rect(playsurface, whiteColor, Rect(targetPosition[0], targetPosition[1], 20, 20))

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

        # 判断是否游戏结束
        if snakePosition[0] > 620 or snakePosition[0] < 0:
            gameover()

        elif snakePosition[1] > 460 or snakePosition[1] < 0:
            gameover()

        # 控制蛇移动的速度
        fpsClock.tick(10)


#   启动入口函数

if __name__ == '__main__':
    main()