如何在Selenium中点击并打开页面上的日历？

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 创建浏览器实例
driver = webdriver.Chrome()

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

# 定位到日历元素并点击
calendar_element = driver.find_element(By.ID, "calendar")
calendar_element.click()

# 等待日历选择框加载完成
wait = WebDriverWait(driver, 10)
date_element = wait.until(EC.visibility_of_element_located((By.XPATH, "//div[@class='date-picker']")))

# 定位到具体的日期元素并点击
date_element = driver.find_element(By.XPATH, "//div[@class='date-picker']//td[@data-date='2022-01-01']")
date_element.click()

# 关闭浏览器
driver.quit()