如何在python中用动态代码关闭弹出窗口

from selenium import webdriver

# 创建浏览器对象
driver = webdriver.Chrome()

# 打开网页
driver.get("https://www.example.com")

# 点击按钮或链接，触发弹出窗口
driver.find_element_by_id("button_id").click()

# 获取当前窗口句柄
main_window = driver.current_window_handle

# 获取所有窗口句柄
all_windows = driver.window_handles

# 切换到弹出窗口
for window in all_windows:
    if window != main_window:
        driver.switch_to.window(window)
        break

# 关闭弹出窗口
driver.close()

# 切换回主窗口
driver.switch_to.window(main_window)

# 关闭浏览器
driver.quit()