Змейка

import pygame 
 
SIZE_BLOCK = 20#размер кубиков
frame_color = [0, 255, 204]#цвет фона
white = (255,255,255)#белые кубики
BLUE = (204,255,255)#голуые кубики
header_color = (0, 204, 153)
snake_color = [0,102,0]
count_bloks = 20# длина ряда
header_margin = 70
margin = 1# отступые между квадратами 
 
size = [SIZE_BLOCK * count_bloks + 2 * SIZE_BLOCK + margin * count_bloks,
        SIZE_BLOCK * count_bloks + 2 * SIZE_BLOCK + margin * count_bloks + header_margin]
print(size)
 
screen = pygame.display.set_mode(size)#s
pygame.display.set_caption('Змейка')#Name window 
timer = pygame.time.Clock
 
class SnakeBlock:
    def __init__(self, x, y):
        self.x = x
        self.y = y
 
def draw_block(color, row, draw):
    pygame.draw.rect(screen, color, [SIZE_BLOCK + column * SIZE_BLOCK + margin * (column + 1), header_margin + SIZE_BLOCK + row * SIZE_BLOCK + margin * (row+1), SIZE_BLOCK, SIZE_BLOCK])
 
snake_block = [SnakeBlock(9, 8), SnakeBlock(9, 9),SnakeBlock(9, 10)]
 
d_row = 0
d_col = 1
 
while True:# цикл работы
 
    for event in pygame.event.get():# выход из игры
        if event.type == pygame.QUIT:
            pygame.quit()
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP and d_col != 0:
                d_row=-1
                d_col=0
            elif event.key == pygame.K_DOWN and d_col != 0:
                d_row=1
                d_col=0
            elif event.key == pygame.K_LEFT and d_col != 0:
                d_row=0
                d_col=-1
            elif event.key == pygame.K_RIGHT and d_col != 0:
                d_row=0
                d_col=1
            
 
    screen.fill(frame_color)
    pygame.draw.rect(screen, header_color, [0, 0, size[0], header_margin])
 
    for row in range(count_bloks):#ряды
        for column in range(count_bloks):# колонки
            if (row + column) % 2 == 0:
                color = BLUE
            else:
                color = white
 
            draw_block(color, row, column)
 
    for block in snake_block:
        draw_block(snake_color, block.x, block.y)
        block.x += d_row
        block.y += d_col
 
    pygame.display.flip()
    timer.tick(2