快捷導(dǎo)航

Python?selenium?find_element()示例詳解

更新時(shí)間：2022年07月08日 09:58:03 作者：☆星空物語☆

selenium定位元素的函數(shù)/方法可以分為兩類:find_element及find_elements,下面這篇文章主要給大家介紹了關(guān)于Python?selenium?find_element()的相關(guān)資料,需要的朋友可以參考下

selenium有以下定位方式：

driver.find_element_by_id(‘id')
driver.find_element_by_xpath('xpath')
driver.find_element_by_link_text('link_text')
driver.find_element_by_partial_link_text('partial_link_text')
driver.find_element_by_name('name')
driver.find_element_by_tag_name('tag_name')
driver.find_element_by_class_name('class_name')
driver.find_element_by_css_selector('css_selector')

但我們看下源碼：

D:\Program Files\Python27\Lib\site-packages\selenium\webdriver\remote\webdriver.py

可以看到，上面那些定位方式，實(shí)際上都是調(diào)用的driver.find_element(by, value)

我們繼續(xù)看下find_element()的源碼，大部分方法最終全是通過By.CSS_SELECTOR來實(shí)現(xiàn)的查找

那By又都有哪些條件呢？

D:\Program Files\Python27\Lib\site-packages\selenium\webdriver\common\by.py

繼續(xù)看源碼，可以看到下面8種

ID = "id"
XPATH = "xpath"
LINK_TEXT = "link text"
PARTIAL_LINK_TEXT = "partial link text"
NAME = "name"
TAG_NAME = "tag name"
CLASS_NAME = "class name"
CSS_SELECTOR = "css selector"

所以我們以后就都只用find_element()就好了，因?yàn)樽罱K實(shí)際上也都是調(diào)用的這個(gè)方法。

簡單封裝一下，以后就記住find_element()一個(gè)方法即可。

from selenium.common.exceptions import TimeoutException
from selenium.webdriver.support.ui import WebDriverWait
from appium import webdriver

class BasePage(object):
    def split_locator(self, locator):
        """
        分解定位表達(dá)式，如'css,.username',拆分后返回'css selector'和定位表達(dá)式'.username'(class為username的元素)
        :param locator: 定位方法+定位表達(dá)式組合字符串，如'css,.username'
        :return: locator_dict[by], value:返回定位方式和定位表達(dá)式
        """
        by = locator.split(',')[0]
        value = locator.split(',')[1]
        locator_dict = {
            'id': 'id',
            'name': 'name',
            'class': 'class name',
            'tag': 'tag name',
            'link': 'link text',
            'plink': 'partial link text',
            'xpath': 'xpath',
            'css': 'css selector',
        }
        if by not in locator_dict.keys():
            raise NameError("wrong locator!'id','name','class','tag','link','plink','xpath','css',exp:'id,username'")
        return locator_dict[by], value


    def wait_element(self, locator, sec=30):
        """
        等待元素出現(xiàn)
        :param locator: 定位方法+定位表達(dá)式組合字符串，用逗號分隔，如'css,.username'
        :param sec:等待秒數(shù)
        """
        by, value = self.split_locator(locator)
        try:
            WebDriverWait(self.driver, sec, 1).until(lambda x: x.find_element(by=by, value=value),message='element not found!!!')
            log.info(u'等待元素：%s' % locator)
        return True
        except TimeoutException:
            return False
        except Exception, e:
            raise e

    
    def get_element(self, locator, sec=60):
        """
        獲取一個(gè)元素
        :param locator: 定位方法+定位表達(dá)式組合字符串，用逗號分隔，如'css,.username'
        :param sec:等待秒數(shù)
        :return: 元素可找到返回element對象，否則返回False
        """
        if self.wait_element(locator, sec):
            by, value = self.split_locator(locator)
            try:
                element = self.driver.find_element(by=by, value=value)
                log.info(u'獲取元素：%s' % locator)
                return element
            except Exception, e:
                raise e
        else:
            return False

    def get_elements(self, locator):
        """
        獲取一組元素
        :param locator: 定位方法+定位表達(dá)式組合字符串，用逗號分隔，如'css,.username'
        :return: elements
        """
        by, value = self.split_locator(locator)
        try:
            elements = WebDriverWait(self.driver, 60, 1).until(lambda x: x.find_elements(by=by, value=value))
            log.info(u'獲取元素列表：%s' % locator)
            return elements
        except Exception, e:
            raise e

以后調(diào)用，先定義元素，如