如何在Python中添加碰撞事件之间的降温？

import pygame
import time

# 初始化Pygame
pygame.init()

# 设置屏幕宽高和背景颜色
width, height = 800, 600
background_color = (255, 255, 255)

# 创建屏幕对象
screen = pygame.display.set_mode((width, height))
pygame.display.set_caption("碰撞事件")

# 创建两个矩形对象
rect1 = pygame.Rect(100, 100, 50, 50)
rect2 = pygame.Rect(200, 200, 50, 50)

# 设置移动速度和冷却时间
move_speed = 5
cooldown_time = 2

# 初始化冷却状态
cooldown = False
cooldown_start_time = 0

# 游戏主循环
running = True
while running:
    # 事件处理
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
    
    # 键盘按键处理
    keys = pygame.key.get_pressed()
    if keys[pygame.K_LEFT]:
        rect1.x -= move_speed
    elif keys[pygame.K_RIGHT]:
        rect1.x += move_speed
    elif keys[pygame.K_UP]:
        rect1.y -= move_speed
    elif keys[pygame.K_DOWN]:
        rect1.y += move_speed

    # 矩形碰撞检测
    if rect1.colliderect(rect2):
        if not cooldown:
            # 触发碰撞事件
            print("碰撞发生！")
            
            # 开始冷却计时
            cooldown = True
            cooldown_start_time = time.time()
        
    # 冷却检测
    if cooldown:
        current_time = time.time()
        if current_time - cooldown_start_time >= cooldown_time:
            cooldown = False
            print("冷却结束")

    # 绘制屏幕
    screen.fill(background_color)
    pygame.draw.rect(screen, (255, 0, 0), rect1)
    pygame.draw.rect(screen, (0, 255, 0), rect2)
    pygame.display.flip()

# 退出游戏
pygame.quit()