From 1958ad283eebeced6b342f93e60ab3eeec80635c Mon Sep 17 00:00:00 2001
From: donggoing <374766849@qq.com>
Date: Thu, 31 Aug 2023 00:22:45 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0wait=5Funtil=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95=EF=BC=8C=E6=94=AF=E6=8C=81=E8=87=AA=E5=AE=9A=E4=B9=89?=
 =?UTF-8?q?=E7=BB=84=E5=90=88=E7=AD=89=E5=BE=85=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DrissionPage/commons/tools.py  | 37 ++++++++++++++++++++++++++++++++++
 DrissionPage/commons/tools.pyi |  3 +++
 2 files changed, 40 insertions(+)

diff --git a/DrissionPage/commons/tools.py b/DrissionPage/commons/tools.py
index dfdab22..a107c2b 100644
--- a/DrissionPage/commons/tools.py
+++ b/DrissionPage/commons/tools.py
@@ -7,6 +7,7 @@ from platform import system
 from pathlib import Path
 from re import search, sub
 from shutil import rmtree
+from time import perf_counter, sleep
 
 
 def get_usable_path(path):
@@ -177,6 +178,42 @@ def get_chrome_hwnds_from_pid(pid, title):
     EnumWindows(callback, hwnds)
     return hwnds
 
+def wait_until(page, condition, timeout=10, poll=0.1, raise_err=True):
+    """等待返回值不为False或空，直到超时
+    :param page (DrissionPage): DrissionPage对象
+    :param condition (function | str | tuple): 等待条件，返回值不为False则停止等待
+    :param timeout (float, optional): 超时时间
+    :param poll (float, optional): 轮询间隔
+    :param message (str, optional): 超时时的报错信息
+    :param ignored_exceptions (bool, optional): 是否忽略异常
+    :return: DP Element or bool
+    """
+    end_time = perf_counter() + timeout
+    if isinstance(condition, str) or isinstance(condition, tuple):
+        if not callable(getattr(page, 's_ele', None)):
+            raise AttributeError('page对象缺少s_ele方法')
+        condition_method = lambda page: page.s_ele(condition)
+    elif callable(condition):
+        condition_method = condition
+    else:
+        raise ValueError('condition必须是函数或者字符串或者元组')
+    while perf_counter() < end_time:
+        try:
+            value = condition_method(page)
+            if value:
+                return value
+        except Exception as exc:
+            pass
+            
+        sleep(poll)
+        if perf_counter() > end_time:
+            break
+        
+    if raise_err:
+        raise TimeoutError('等待超时')
+    else:
+        return False
+
 # def get_exe_from_port(port):
 #     """获取端口号第一条进程的可执行文件路径
 #     :param port: 端口号
diff --git a/DrissionPage/commons/tools.pyi b/DrissionPage/commons/tools.pyi
index 54b8197..1781bc6 100644
--- a/DrissionPage/commons/tools.pyi
+++ b/DrissionPage/commons/tools.pyi
@@ -6,6 +6,7 @@
 from os import popen
 from pathlib import Path
 from typing import Union
+from types import FunctionType
 
 from chromium_page import ChromiumPage
 
@@ -38,3 +39,5 @@ def get_browser_progress_id(progress: Union[popen, None], address: str) -> Union
 
 
 def get_chrome_hwnds_from_pid(pid: Union[str, int], title: str) -> list: ...
+
+def wait_until(page, condition: Union[FunctionType, str, tuple], timeout: float, poll: float, raise_err: bool): ...