如何让pygame只需点击一次鼠标就能恢复？

import pygame
import sys

pygame.init()

screen = pygame.display.set_mode((800, 600))
pygame.display.set_caption("Restore On Click")

need_restore = False

def handle_click():
    global need_restore
    need_restore = True

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
            handle_click()

    if need_restore:
        # 执行需要恢复的操作
        # ...
        need_restore = False

    pygame.display.flip()