import pygame
pygame.init()

# 设置窗口尺寸
window_size = (400, 500)
screen = pygame.display.set_mode(window_size)

# 游戏循环
while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
# 接下来，我们需要定义方块的形状。我们可以用数字来表示方块的形状，比如1表示L型、2表示Z型等等。

SHAPES = [
    # L 形状
    [
        [1, 0],
        [1, 0],
        [1, 1],
    ],
    # J 形状
    [
        [0, 2],
        [0, 2],
        [2, 2],
    ],
    # Z 形状
    [
        [3, 3, 0],
        [0, 3, 3],
    ],
    # S 形状
    [
        [0, 4, 4],
        [4, 4, 0],
    ],
    # T 形状
    [
        [5, 5, 5],
        [0, 5, 0],
    ],
    # O 形状
    [
        [6, 6],
        [6, 6],
    ],
    # I 形状
    [
        [7],
        [7],
        [7],
        [7],
    ],
]
# 接下来，我们需要定义方块的类。每个方块都有自己的形状和位置。

class Block:
    def __init__(self, shape, x=0, y=0):
        self.shape = shape
        self.x = x
        self.y = y

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

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

    def move_down(self):
        self.y += 1
# 然后，我们需要定义游戏区域并在其中绘制方块：

# 游戏区域尺寸
BOARD_WIDTH = 10
BOARD_HEIGHT = 20

# 方块大小
BLOCK_SIZE = 20

# 绘制游戏区域
def draw_board():
    for y in range(BOARD_HEIGHT):
        for x in range(BOARD_WIDTH):
            pygame.draw.rect(screen, (255, 255, 255), (x * BLOCK_SIZE, y * BLOCK_SIZE, BLOCK_SIZE, BLOCK_SIZE), 1)

# 绘制方块
def draw_block(block):
    for y, row in enumerate(block.shape):
        for x, value in enumerate(row):
            if value:
                pygame.draw.rect(screen, (255, 255, 255), ((block.x + x) * BLOCK_SIZE, (block.y + y) * BLOCK_SIZE, BLOCK_SIZE, BLOCK_SIZE))

# 创建一个方块
current_block = Block(SHAPES[0], 0, 0)

# 游戏循环
while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()

    # 绘制游戏区域和方块
    draw_board()
    draw_block(current_block)

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