# 导入pygame库，sys操控python运行的环境
import pygame, sys, random
# 这个模块包含所有pygame所使用的常量
from pygame.locals import *

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

# 自定义退出游戏函数
def gameExit():
    pygame.quit()
    sys.exit()
# 自定义游戏结束函数
def gameOver():
    #playsurface.blit(font.render("Score:" + str(score), -1, (255, 255, 255)), [30, 20])
    pygame.quit()

# 定义main函数--》定义我们的入口函数
def main():
    # 初始化pygame
    pygame.init()
    # 定义一个变量来控制速度
    fpsClock = pygame.time.Clock()
    # 创建pygame显示层，创建一个界面
    playsurface = pygame.display.set_mode((640, 480))  #生成主屏幕
    pygame.display.set_caption('贪吃蛇')
    # 初始化变量
    # 贪吃蛇初始坐标位置   （先以100,100为基准）
    snakePosition = [100, 100]
    # 初始化贪吃蛇的长度列表中有个元素就代表有几段身体
    snakeBody = [[100, 100], [80, 100], [60, 100]]
    # 初始化目标方向额位置
    targetPosition = [300, 300]
    # 目标方块的标记 目的：判断是否吃掉了这个目标方块，1-没有吃 0-吃了
    targetflag = 1
    # 初始化方向  往右
    direction = 'right'
    # 定义一个方向变量（人为控制  按键）
    changeDirection = direction
    font=pygame.font.SysFont(None,50)
    score=0
    while True:
        for event in pygame.event.get():  # 从队列中获取事件
            if event.type == QUIT:
                gameExit()
            elif event.type == KEYDOWN:  # 按键按下时，会触发该事件
                if event.key == K_RIGHT:
                    changeDirection = 'right'
                if event.key == K_LEFT:
                    changeDirection = 'left'
                if event.key == K_UP:
                    changeDirection = 'up'
                if event.key == K_DOWN:
                    changeDirection = 'down'
                    # 对应键盘上的esc文件
                if event.key == K_ESCAPE:
                    gameExit()

        # 确定方向
        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
            score+=1
        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))

        playsurface.blit(font.render("Score:"+str(score),-1,(255,255,255)),[30,20])
        # 更新显示到屏幕表面
        pygame.display.flip()
        # 判断是否游戏结束
        if snakePosition[0] > 620 or snakePosition[0] < 0:
            gameOver()
        elif snakePosition[1] > 460 or snakePosition[1] < 0:
            gameOver()
        # 控制游戏速度
        fpsClock.tick(3)


#   启动入口函数
if __name__ == '__main__':
    main()