blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

添加等待元素隐藏/消失/出现的函数，未完成

Browse Source
This commit is contained in:
g1879 2020-07-20 22:42:42 +08:00
parent 745bc09427
commit a2f55f6f8c
1 changed files with 47 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
DrissionPage/driver_page.py
View File

@ -4,6 +4,7 @@
@Contact : g1879@qq.com
@File : driver_page.py
"""
import time
from glob import glob
from typing import Union, List, Any
from urllib.parse import quote
@ -68,8 +69,11 @@ class DriverPage(object):
self._url_available = self.check_page()
return self._url_available
def ele(self, loc_or_ele: Union[tuple, str, DriverElement], mode: str = None,
timeout: float = None, show_errmsg: bool = False) -> Union[DriverElement, List[DriverElement], None]:
def ele(self,
loc_or_ele: Union[tuple, str, DriverElement],
mode: str = None,
timeout: float = None,
show_errmsg: bool = False) -> Union[DriverElement, List[DriverElement], None]:
"""根据loc获取元素或列表，可用用字符串控制获取方式，可选'id','class','name','tagName'
例：ele.find('id:ele_id')
:param loc_or_ele: 页面元素地址
@ -91,6 +95,47 @@ class DriverPage(object):
return self.ele(loc, mode='all', timeout=timeout, show_errmsg=show_errmsg)
# ----------------以下为独有函数-----------------------
def wait_ele(self, loc_or_ele, mode: str, timeout: float = None):
"""
:param loc_or_ele:
:param mode: show, hidden, del,
:param timeout:
:return:
"""
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as ec
timeout = timeout or self.timeout
is_ele = False
if isinstance(loc_or_ele, DriverElement):
loc_or_ele = loc_or_ele.inner_ele
is_ele = True
elif isinstance(loc_or_ele, WebElement):
is_ele = True
elif isinstance(loc_or_ele, str):
loc_or_ele = get_loc_from_str(loc_or_ele)
if is_ele: # 当传入参数是元素对象时
end_time = time.time() + timeout
if mode == 'del':
while True:
try:
loc_or_ele.is_enabled()
except:
return True
if time.time() > end_time:
return False
elif mode == 'visible':
pass
elif mode == 'invisible':
pass
else: # 当传入参数是控制字符串或元组时
if mode == 'del':
return WebDriverWait(self.driver, timeout).until_not(ec.presence_of_element_located(*loc_or_ele))
elif mode == 'visible':
return WebDriverWait(self.driver, timeout).until(ec.visibility_of_element_located(*loc_or_ele))
elif mode == 'invisible':
return WebDriverWait(self.driver, timeout).until_not(ec.visibility_of_element_located(*loc_or_ele))
def check_page(self) -> Union[bool, None]:
"""检查页面是否符合预期