如何在pygame中重启游戏？

import pygame
import sys

pygame.init()

screen = pygame.display.set_mode((width, height))

while True:
    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_r:
                # 重启游戏
                main()

def main():
    # 游戏逻辑代码
    ...

main()