# * coding=utf8
# Python(pygame)实现代码雨
# 今日头条: 明哥读世界
# 微博：明哥读世界
# 个人博客:https://www.mylasting.com

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()