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)
