如何在鼠标单击Python时更改矩形颜色(pygame)

import pygame
from pygame.locals import *

pygame.init()

window_size = (800, 600)
screen = pygame.display.set_mode(window_size)
pygame.display.set_caption("Change Rectangle Color")

rect_position = (200, 200)
rect_color = (255, 0, 0)  # 初始颜色为红色

rect = pygame.Rect(rect_position, (100, 100))

running = True
while running:
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False
        elif event.type == MOUSEBUTTONDOWN:
            if rect.collidepoint(event.pos):
                rect_color = (0, 255, 0)  # 当鼠标点击矩形时，将颜色更改为绿色

    screen.fill((255, 255, 255))  # 清空屏幕
    pygame.draw.rect(screen, rect_color, rect)  # 绘制矩形
    pygame.display.flip()  # 更新屏幕显示

pygame.quit()