from selenium import webdriver
class extend():
"""
You can add new function here for you own.
"""
def __init__(self):
self.driver = self.init_driver()
def init_driver(self):
global driver
if driver is None:
print('======initial driver=====')
chromeOptions = webdriver.ChromeOptions()
chromeOptions.add_experimental_option('useAutomationExtension', False)
driver = webdriver.Chrome(chrome_options=chromeOptions,desired_capabilities=chromeOptions.to_capabilities())
driver.maximize_window()
return driver
在 selenium 执行 UI 自动化测试的过程中，会经常遇到以下问题：
基于这些问题，我先封装一个通用的查找元素方法如下：
def find_elements(self,driver,locator,is_displayed=True,text=''):
"""
Find elements using the given condition..
:param driver:
:param locator: like id=btn_search
:param is_displayed: True : Only the elements shown on the page will be returned. False: Elements will also be returned no matter is shown or hidden.
:param text: Only the elements contain the specified text will be returned.
:return: Found element list.
"""
method, value = locator.split('=',maxsplit=1)
if 'css' in method:
method = By.CSS_SELECTOR
elif 'class' in method:
method = By.CLASS_NAME
elif 'text' in method:
method = By.PARTIAL_LINK_TEXT
elif 'tag' in method:
method = By.TAG_NAME
elements = driver.find_elements(by=method,value=value)
new_elements = []
for element in elements:
if text != '':
if self.is_text_in_element(element,text):
if is_displayed:
if element.is_displayed():
new_elements.append(element)
else:
new_elements.append(element)
else:
if is_displayed:
if element.is_displayed():
new_elements.append(element)
else:
new_elements.append(element)
return new_elements
def find_element(self,driver,locator,is_displayed=True,index=0,text=''):
"""
Find element using the given condition..
:param driver:
:param para_list: [method, value] eg : ['id','add_btn']
:param is_displayed: True : Only the element shown on the page will be returned. False: Element will also be returned no matter is shown or hidden.
:param index: If more than one element is found, driver will return the specified element by index. e.g: element 0,1,2 is found and the index is set to by 2, the element 2 will be returned.
:param text: Only the element contains the specified text will be returned.
:return: Found element.
"""
elements = self.find_elements(driver,locator,is_displayed=is_displayed,text=text)
if len(elements)>0 and index < len(elements):
return elements[index]
else:
return elements[0]
扩展常用的操作步骤：
def click_index(self,driver,locator,index=0):
"""
When more than one web element are found by the given value, you can use the index the specified which element you want to click.
:param driver:
:param index: default to 0, means to click the first element found.
:return:
"""
elements = self.find_elements(driver, locator)
elements[int(index)].click()
time.sleep(2)
def fill(self,driver,locator,text, is_displayed = True,index=0):
"""
Fill specified text to the target web elelemnt.
:param driver:
:param para_list: [method, value]
:param text: the text you want to input
:param is_displayed:
:param index:
:return:
"""
text = self.replaceRandomValue(text)
elements = self.find_elements(driver, locator,is_displayed,text='')
elements[index].clear()
elements[index].send_keys(text)
def click_text(self,driver,text,type=''):
"""
click a web element by it's text and type.
:param driver:
:param text:
:param type: by default is empty. you can specified the type to it's tag name. e.g : span, li, a .
:return:
"""
if len(type) ==0:
try :
driver.find_element_by_link_text(text).click()
except:
try:
driver.find_element_by_partial_link_text(text).click()
except:
elements = driver.find_elements(by='xpath', value="//*[contains(.,'" + text + "')]")
for element in elements:
element.click()
else:
elements = self.find_elements(driver, ['tag', type], is_displayed=True, text=text)
if len(elements):
elements[0].click()
time.sleep(2)
def try_click(self,driver,locator):
"""
Try to click on a specified web element. Sometimes a message will be popped up on a web page. We need this function to close the pop up window.
:param driver:
:param para_list:
:return:
"""
for i in range(3):
try:
self.find_element(driver, locator).click()
break
except:
pass
time.sleep(2)
def scroll_down(self,height=0):
if height:
self.driver.execute_script('window.scrollTo(0,%s)' %height)
else:
self.driver.execute_script('window.scrollTo(0,document.body.scrollHeight)')
def scroll_to_element(self,locator):
for i in range(1,50):
height = i*200
self.driver.execute_script('window.scrollTo(0,%s)' % height)
self.wait(1)
if len(self.find_elements(locator)):
break
from app.core.extend import extend
new_driver = extend() # 初始化自己封装的driver 库
new_driver.driver.get('https://www.baidu.com') # 仍可以通过 .driver 来调用原来 selenium 的driver 方法
new_driver.fill('id=kw','testerhome')
new_driver.click_index('id=su')