请用PYTHON帮我写一个俄罗斯方块的代码

好的，以下是一个基本的俄罗斯方块游戏的 Python 代码示例，使用 Pygame 库进行绘制和交互：

import pygame
import random

# 初始化 Pygame
pygame.init()

# 设置游戏窗口大小
screen_width = 800
screen_height = 600
screen = pygame.display.set_mode((screen_width, screen_height))

# 定义方块大小和颜色
block_size = 30
block_colors = [(0, 0, 0), (255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0)]

# 定义方块形状
shapes = [
    [[1, 1, 1], [0, 1, 0]],
    [[2, 2], [2, 2]],
    [[0, 3, 3], [3, 3, 0]],
    [[4, 4, 0], [0, 4, 4]],
    [[5, 5, 5, 5]]
]

# 定义方块初始位置
start_x = screen_width // 2 - block_size * 2
start_y = 0

# 定义游戏循环变量
game_over = False

# 定义方块类
class Block:
    def __init__(self, x, y, shape):
        self.x = x
        self.y = y
        self.shape = shape
        self.color = random.choice(block_colors)

    def draw(self):
        for i in range(len(self.shape)):
            for j in range(len(self.shape[i])):
                if self.shape[i][j] != 0:
                    pygame.draw.rect(screen, self.color, (self.x + j * block_size, self.y + i * block_size, block_size, block_size))

    def move_down(self):
        self.y += block_size

    def move_right(self):
        self.x += block_size

    def move_left(self):
        self.x -= block_size

# 定义键盘事件处理函数
def handle_events(block):
    global game_over

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            game_over = True
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                block.move_left()
            elif event.key == pygame.K_RIGHT:
                block.move_right()
            elif event.key == pygame.K_DOWN:
                block.move_down()

# 主程序循环
while not game_over:
    # 创建一个新的方块
    block = Block(start_x, start_y, random.choice(shapes))

    # 主循环，处理方块移动和绘制
    while True:
        # 处理键盘事件
        handle_events(block)

        # 绘制背景
        screen.fill((255, 255, 255))

        # 绘制方块
        block.draw()

        # 更新屏幕
        pygame.display.update()

        # 延迟一段时间
        pygame.time.delay(500)

        # 移动方块
        block.move_down()

        # 如果方块到达底部，结束循环
        if block.y + len(block.shape) * block_size >= screen_height:
            break

    # 如果游戏结束，退出循环
    if game_over:
        break

# 退出 Pygame
pygame.quit()

请注意，这只是一个基本的示例，可能需要进一步调整和完善，以实现更完整和复杂的俄罗斯方块游戏。