import pygame
from pygame import locals
import random
# 初始化pygame，为使用硬件做准备
pygame.init()

# 创建一个窗口
screen = pygame.display.set_mode((660, 480))
FPSCLOCK = pygame.time.Clock()  # pygame时钟，控制游戏速度（帧数）

# 背景
background = pygame.image.load('bg.png')
right = pygame.image.load('right.png')      # 头 朝右
food = pygame.image.load('apple.png')       # 食物 苹果
body = pygame.image.load('body.png')        # 身体
left = pygame.image.load('left.png')        # 头 朝左
up = pygame.image.load('up.png')            # 头 朝上
down = pygame.image.load('down.png')        # 头 朝下
ttf1 = pygame.font.Font('neuropol.ttf',18)
x, y = 240, 120
position = [(180, 90), (180, 120), (210, 120), (x, y)]
score = 0
setheading = "right"
snake_head = right
apple_x=360
apple_y=300
a = 3
while True:
    for event in pygame.event.get():
        if event.type == locals.QUIT:
            # 接收到退出事件后退出程序
            exit()
        if event.type == locals.KEYDOWN:
            if event.key == locals.K_RIGHT and setheading != "left":
                setheading = 'right'
                snake_head = right
            if event.key == locals.K_LEFT and setheading != "right":
                setheading = 'left'
                snake_head = left
            if event.key == locals.K_UP and setheading != "down":
                setheading = 'up'
                snake_head = up
            if event.key == locals.K_DOWN and setheading != "up":
                setheading = 'down'
                snake_head = down

    # 设置贪吃蛇的头部坐标
    if setheading == "right":
        x += 30
    elif setheading == "left":
        x -= 30
    elif setheading == "up":
        y -= 30
    else:
        y += 30
    if x==apple_x and y==apple_y:
        apple_x=random.randrange(0,660,30)
        apple_y=random.randrange(0,480,30)
        score+=10
        a=a+1
    else:
        position.pop(0)
    if x<0 or y<0 or x>630 or y>450:
        exit()
    position.append((x, y))
    # 将背景图画上去
    screen.blit(background, (0, 0))
    # 将贪吃蛇的头画上去
    screen.blit(snake_head, position[-1])
    # 将贪吃蛇的身体画上去
    for i in range(len(position)-1):
        screen.blit(body, position[i])
    # 将果实画上去
    screen.blit(food, (apple_x,apple_y))
    info = "score:"+str(score)
    info1 = ttf1.render(info, True, (0,0,0))
    screen.blit(info1,(540,10))
    # 刷新画面

    pygame.display.update()
    FPSCLOCK.tick(a)