如何使用鼠标和键盘检测来编辑pygame中的数独网格？

import pygame

# 初始化pygame
pygame.init()

# 定义窗口尺寸
WIDTH, HEIGHT = 600, 600
WIN = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("数独编辑器")

# 定义数独网格
grid = [
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0, 0]
]

# 定义字体
FONT = pygame.font.SysFont("Arial", 40)

# 定义颜色
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GRAY = (128, 128, 128)

# 游戏主循环
def main():
    running = True
    while running:
        # 绘制数独网格
        draw_grid()

        # 监听事件
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False
            elif event.type == pygame.MOUSEBUTTONDOWN:
                # 鼠标点击事件
                pos = pygame.mouse.get_pos()
                click(pos)
            elif event.type == pygame.KEYDOWN:
                # 键盘按下事件
                if event.key == pygame.K_1:
                    key = 1
                elif event.key == pygame.K_2:
                    key = 2
                elif event.key == pygame.K_3:
                    key = 3
                elif event.key == pygame.K_4:
                    key = 4
                elif event.key == pygame.K_5:
                    key = 5
                elif event.key == pygame.K_6:
                    key = 6
                elif event.key == pygame.K_7:
                    key = 7
                elif event.key == pygame.K_8:
                    key = 8
                elif event.key == pygame.K_9:
                    key = 9
                elif event.key == pygame.K_DELETE or event.key == pygame.K_BACKSPACE:
                    key = 0
                else:
                    continue
                edit(key)

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

# 绘制数独网格
def draw_grid():
    WIN.fill(WHITE)
    for i in range(9):
        for j in range(9):
            if grid[i][j] != 0:
                pygame.draw.rect(WIN, GRAY, (j * 60, i * 60, 60, 60))
                text_surface = FONT.render(str(grid[i][j]), True, BLACK)
                text_rect = text_surface.get_rect(center=(j * 60 + 30, i * 60 + 30))
                WIN.blit(text_surface, text_rect)

# 处理鼠标点击事件
def click(pos):
    x, y = pos
    row = y // 60
    col = x // 60
    print("点击位置：行{}, 列{}".format(row, col))

# 处理键盘输入事件
def edit(key):
    print("输入数字：{}".format(key))

# 运行游戏
if __name__ == "__main__":
    main()