用python实现贪吃蛇

class Point:
    row = 0
    col = 0
    def __init__(self, row, col):
        self.row = row
        self.col = col

    def copy(self):
        return Point(row=self.row, col=self.col)

import pygame
import random

pygame.init()
W = 800
H = 600
ROW = 60
COL = 80


size = (W, H)
window = pygame.display.set_mode(size)
pygame.display.set_caption('贪吃蛇')

bg_color = (255, 255, 255)
snake_color = (128, 128, 128)
head = Point(row=int(ROW/2), col=int(COL/2))
head_color = (0, 128, 128)
snakes = [
    Point(row=head.row, col=head.col+1),
    Point(row=head.row, col=head.col+2),
    Point(row=head.row, col=head.col+3),
]


#生成食物
def gen_food():
    while 1:
        pos = Point(row=random.randint(0, ROW-1), col=random.randint(0, COL-1))
        #是否碰到蛇了
        is_coll = False

        #碰到头
        if head.row == pos.row and head.col == pos.col:
            is_coll = True

        for snake in snakes:
            if snake.row == pos.row and snake.col == pos.col:
                is_coll = True
                break

        if not is_coll:
            break
    return pos

#定义坐标

food = gen_food()
food_color = (0, 0, 0)


direct = 'left'
def rect(point, color):
    cell_width = W/COL
    cell_height = H/ROW

    left = point.col*cell_width
    top = point.row*cell_height

    pygame.draw.rect(window, color, (left, top, cell_width, cell_height))


#游戏循环
quit = True
clock = pygame.time.Clock()
while quit:
    #处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            quit = False
        elif event.type == pygame.KEYDOWN:
            if event.key == 273:
                if direct == 'right' or direct == 'left':
                    direct = 'up'
            elif event.key == 274:
                if direct == 'right' or direct == 'left':
                    direct = 'down'
            elif event.key == 276:
                if direct == 'up' or direct == 'down':
                    direct = 'left'
            elif event.key == 275:
                if direct == 'up' or direct == 'down':
                    direct = 'right'

    #吃东西
    eat = (head.row == food.row and head.col == food.col)

    #重新产生食物
    if eat:
        food = gen_food()

    #处理身子
    #1,把原来的head插入到sneak的第一个位置
    snakes.insert(0, head.copy())
    #2，把sneak的最后一个数据删掉
    if not eat:
        snakes.pop()

    #移动
    if direct == 'left':
        head.col -= 1
    elif direct == 'right':
        head.col += 1
    elif direct == 'up':
        head.row -= 1
    elif direct == 'down':
        head.row += 1

    #检测
    dead = False
    #1，撞墙
    if head.col < 0 or head.row < 0 or head.col >= COL or head.row >= ROW:
        dead = True

    #2，撞自己
    for snake in snakes:
        if head.col == snake.col and head.row == snake.row:
            dead = True
            break

    if dead :
        print("死了")
        quit = False

    #渲染
    #背景
    pygame.draw.rect(window, bg_color, (0, 0, W, H))
    #蛇头
    for snake in snakes:
        rect(snake, snake_color)
    rect(head, head_color)
    rect(food, food_color)


    #
    pygame.display.flip()


    #fps,即蛇的速度
    clock.tick(18)