from selenium import webdriver
from selenium.webdriver import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.support.wait import WebDriverWait
# 点击元素直到成功
def click_success(locator): # locator=(By.XXX,'dddd')
def _predicate(driver):
try:
element = driver.find_element(*locator)
element.click()
return True
except:
return False
return _predicate
class DriverOperate:
def __init__(self, browser='chrome'):
self.browser = browser.lower()
if self.browser == 'chrome':
options = webdriver.ChromeOptions()
self.driver = webdriver.Chrome(options=options)
if self.browser == 'firfox':
options = webdriver.FirefoxOptions()
self.driver = webdriver.Firefox(options=options)
if self.browser == 'Edge':
options = webdriver.EdgeOptions()
self.driver = webdriver.Edge(options=options)
if self.browser == 'ie':
options = webdriver.IeOptions()
self.driver = webdriver.Ie(options=options)
if self.browser == 'ie':
options = webdriver.safari.optins.Options()
self.driver = webdriver.Safari(options=options)
else:
raise BaseException(f'{self.browser}浏览器不支持')
# 打开网址
def get(self, url):
self.driver.get(url)
def find_element(self, ele_info):
# 为了更好的描述元素信息,方便记录日志
# 我们设计元素信息是这样传递的:{"name":"登录链接","type":"linktext","value":"登录","timeout":5}
name = ele_info['name']
type = ele_info['type'].lower()
value = ele_info['value']
timeout = ele_info.get('timeout', 5)
locator = self.get_by(ele_info)
try:
wait = WebDriverWait(driver=self.driver, timeout=5)
element = wait.until(expected_conditions.presence_of_element_located(locator))
return element
except BaseException as e:
raise BaseException(f'点击元素【{name}】,通过【{type}】,值是【{value}】,点击失败{e}')
def find_elements(self, ele_info):
# 为了更好的描述元素信息,方便记录日志
# 我们设计元素信息是这样传递的:{"name":"登录链接","type":"linktext","value":"登录","timeout":5}
name = ele_info['name']
type = ele_info['type'].lower()
value = ele_info['value']
timeout = ele_info.get('timeout', 5)
locator = self.get_by(ele_info)
try:
wait = WebDriverWait(driver=self.driver, timeout=5)
element = wait.until(expected_conditions.presence_of_all_elements_located(locator))
return element
except BaseException as e:
raise BaseException(f'点击元素【{name}】,通过【{type}】,值是【{value}】,点击失败{e}')
def get_by(self, ele_info):
name = ele_info['name']
type = ele_info['type'].lower()
value = ele_info['value']
timeout = ele_info.get('timeout', 5)
if type == 'id':
locator = (By.ID, value)
elif type == 'name':
locator = (By.NAME, value)
elif type == 'classname':
locator = (By.CLASS_NAME, value)
elif type == 'tagname':
locator = (By.TAG_NAME, value)
elif type == 'linktext':
locator = (By.LINK_TEXT, value)
elif type == 'partiallinktext':
locator = (By.PARTIAL_LINK_TEXT, value)
elif type == 'css':
locator = (By.CSS_SELECTOR, value)
elif type == 'xpath':
locator = (By.XPATH, value)
else:
raise BaseException(f'定位类型{type}不支持')
return locator
# 点击二次封装
def click(self, ele_info):
name = ele_info['name']
type = ele_info['type'].lower()
value = ele_info['value']
timeout = ele_info.get('timeout', 5)
lactor = self.get_by(ele_info)
try:
wait = WebDriverWait(self.driver, timeout=5)
wait.until(click_success(lactor))
except BaseException as e:
raise BaseException(f'点击元素【{name}】,通过【{type}】,值是【{value}】,点击失败{e}')
def send_keys(self, ele_info, text, is_clear=True):
name = ele_info['name']
type = ele_info['type'].lower()
value = ele_info['value']
timeout = ele_info.get('timeout', 5)
lactor = self.get_by(ele_info)
element = self.find_element(ele_info)
try:
if is_clear:
# element.clear() # 默认清除方法可能清不掉
element.send_keys(Keys.CONTROL,'a')
element.send_keys(Keys.DELETE)
element.send_keys(text)
except BaseException as e:
raise BaseException(f'向元素【{name}】,通过【{type}】,值是【{value}】,输入【{text}】失败{e}')
一些没完成的代码
最新推荐文章于 2024-08-06 17:45:33 发布