Selenium自动化中循环操作的元素定位与显式等待策略

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from time import sleep # 仅在必要时使用，应尽量替换为显式等待

# 假设 sb 是一个封装了 WebDriver 实例的对象，并且可以通过 sb.driver 访问原始的 WebDriver 对象
# 如果 sb 本身就支持显式等待，请参考其文档进行集成。
# 在这里，我们假设需要直接使用原始的 driver 对象。

def select_first_category(sb):
    # 避免使用硬编码的 sleep，用显式等待替代
    # sleep(1) # 移除或替换此行

    # 定义等待超时时间
    wait_timeout = 15 # 例如，等待15秒

    try:
        # 等待元素 #mat-select-value-1 可点击
        # 使用 By.CSS_SELECTOR 定位器
        first_category_dropdown = WebDriverWait(sb.driver, wait_timeout).until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, '#mat-select-value-1'))
        )
        first_category_dropdown.click() # 点击下拉菜单
        print("First category dropdown clicked successfully.")

        # 等待“Application Centre”选项出现并可点击
        # 注意：此处假设 span:contains("Application Centre") 是一个有效的CSS选择器，
        # 但Selenium原生CSS选择器不支持 :contains()。
        # 更稳健的方法是使用 XPath 或根据实际HTML结构调整。
        # 示例使用 XPath：
        application_centre_option = WebDriverWait(sb.driver, wait_timeout).until(
            EC.element_to_be_clickable((By.XPATH, '//span[contains(text(), "Application Centre")]'))
        )
        application_centre_option.click() # 点击“Application Centre”选项
        print("Application Centre option selected successfully.")

    except Exception as e:
        print(f"Error in select_first_category: {e}")
        # 可以选择重新尝试或进行错误处理
        raise # 重新抛出异常，以便上层调用捕获

    select_second_category(sb) # 继续下一个步骤

def go_to_homepage(sb):
    wait_timeout = 15
    try:
        # 点击返回主页的图片/链接
        homepage_link = WebDriverWait(sb.driver, wait_timeout).until(
            EC.element_to_be_clickable((By.XPATH, '/html/body/app-root/div/header/div[1]/div/a/img'))
        )
        homepage_link.click()
        print("Back to Booking There is No Dates >>>>> Success")

        # 等待主页上的某个关键元素加载完成，例如“Start New Booking”按钮
        WebDriverWait(sb.driver, wait_timeout).until(
            EC.element_to_be_clickable((By.XPATH, '/html/body/app-root/div/div/app-dashboard/section[1]/div/div[2]/div/button'))
        )
        print("Homepage loaded and 'Start New Booking' button is ready.")

    except Exception as e:
        print(f"Error navigating to homepage or waiting for elements: {e}")
        raise

    click_new_booking(sb)

def Check_Appointment(sb):
    wait_timeout = 15
    no_appointment_message = "no appointment" # 确保此消息与实际页面文本匹配

    while True:
        try:
            # 等待包含预约信息的元素出现并可见
            appointment_status_element = WebDriverWait(sb.driver, wait_timeout).until(
                EC.visibility_of_element_located((By.XPATH, '/html/body/app-root/div/div/app-eligibility-criteria/section/form/mat-card[1]/form/div[4]'))
            )
            element_text = appointment_status_element.text # 获取元素文本

            if no_appointment_message in element_text:
                print("We are sorry but no appointment slots are currently available.")
                go_to_homepage(sb) # 重新开始流程
            else:
                print("Earliest available slot for Applicants")
                # playsound('./Music.mp3') # 如果有音频播放，保留
                print("Attention Alarm >>>>> Success")
                get_appointment_data(sb)
                break  # 找到预约后跳出循环

        except TimeoutException:
            print("Timed out waiting for appointment status element. Retrying or navigating back.")
            go_to_homepage(sb) # 元素未在预期时间内出现，可能需要重新尝试

        except Exception as e:
            print(f"An unexpected error occurred in Check_Appointment: {e}")
            go_to_homepage(sb) # 发生其他错误也尝试重新开始