Playwright实现条件循环+每10秒刷新，等待定位器文本状态变更

核心逻辑 #

通过循环检查目标元素的状态文本，每10秒点击刷新按钮，直到状态变为Completed（任务成功）或Failed（测试失败），同时添加超时限制避免无限等待。

TypeScript 实现代码 #

async function waitForTaskCompletion(page: Page, statusSelector: string, refreshBtnSelector: string) {
  const maxWaitMinutes = 30; // 最大等待时长，可按需调整
  const startTime = Date.now();
  const maxWaitTime = maxWaitMinutes * 60 * 1000;

  while (true) {
    // 检查是否超时
    if (Date.now() - startTime > maxWaitTime) {
      throw new Error(`任务等待超时，已超过${maxWaitMinutes}分钟`);
    }

    // 获取当前状态文本
    const statusText = await page.locator(statusSelector).textContent();
    if (!statusText) {
      await page.click(refreshBtnSelector);
      await page.waitForTimeout(10000);
      continue;
    }

    const currentStatus = statusText.trim();
    switch (currentStatus) {
      case 'Completed':
        console.log('任务执行完成');
        return;
      case 'Failed':
        throw new Error(`任务执行失败，状态：${currentStatus}`);
      case 'Pending':
      case 'Running':
        console.log(`任务当前状态：${currentStatus}，10秒后刷新页面`);
        await page.waitForTimeout(10000);
        await page.click(refreshBtnSelector);
        break;
      default:
        console.log(`检测到未知状态：${currentStatus}，继续等待刷新`);
        await page.waitForTimeout(10000);
        await page.click(refreshBtnSelector);
    }
  }
}

// 调用示例（替换为实际选择器）
// await waitForTaskCompletion(page, '#task-status', '#refresh-btn');

Python 实现代码 #

import time
from playwright.async_api import Page

async def wait_for_task_completion(page: Page, status_selector: str, refresh_btn_selector: str):
    max_wait_minutes = 30  # 最大等待时长，可按需调整
    start_time = time.time()
    max_wait_time = max_wait_minutes * 60

    while True:
        # 检查是否超时
        if time.time() - start_time > max_wait_time:
            raise Exception(f"任务等待超时，已超过{max_wait_minutes}分钟")

        # 获取当前状态文本
        status_text = await page.locator(status_selector).text_content()
        if not status_text:
            await page.click(refresh_btn_selector)
            await page.wait_for_timeout(10000)
            continue

        current_status = status_text.strip()
        if current_status == 'Completed':
            print("任务执行完成")
            return
        elif current_status == 'Failed':
            raise Exception(f"任务执行失败，状态：{current_status}")
        elif current_status in ['Pending', 'Running']:
            print(f"任务当前状态：{current_status}，10秒后刷新页面")
            await page.wait_for_timeout(10000)
            await page.click(refresh_btn_selector)
        else:
            print(f"检测到未知状态：{current_status}，继续等待刷新")
            await page.wait_for_timeout(10000)
            await page.click(refresh_btn_selector)

# 调用示例（替换为实际选择器）
# await wait_for_task_completion(page, "#task-status", "#refresh-btn")

关键说明 #

• 替换代码中的statusSelector和refreshBtnSelector为你页面中对应元素的CSS/XPath选择器
• waitForTimeout（TypeScript）/wait_for_timeout（Python）用于等待10秒，是Playwright原生的等待方法
• 加入了最大等待时长限制，避免无限循环，可根据实际任务耗时调整
• 遇到Failed状态时直接抛出异常，会触发Playwright测试用例失败
• 若目标状态元素初始未加载，可在循环开头添加await page.wait_for_selector(status_selector, state="visible", timeout=0)（Python）或await page.waitForSelector(statusSelector, { state: 'visible', timeout: 0 })（TypeScript），确保元素出现后再检查文本

内容的提问来源于stack exchange，提问作者David Pongo