selenium-webdriver的二次封裝(十)

接着上篇隨筆 selenium-配置文件定位元素 ,進行了配置文件設置後,將配置文件運用到定位元素中css

思路:拿到定位的 key 和 value 後,對 webdrvier 中定位進行封裝,使能夠直接運用html

主要的8種定位方法:  web

  • find_element_by_id   
  • find_element_by_name
  • find_element_by_xpath
  • find_element_by_link_text
  • find_element_by_partial_link_text
  • find_element_by_tag_name
  • find_element_by_class_name
  • find_element_by_css_selector
# conding=utf-8

from selenium import webdriver
from auto_learning.config import ReadIni #導入讀取的配置文件的py文件,若是不能導入檢查是不是包(__init__.py文件存在 )文件

class FindElement(object):
    def __init__(self,driver):
        self.driver = driver

    def get_Element(self,pageElement,key):
        # 實例化ReadIni
        file_name = '../config/BaiduElement.ini'
        read_ini = ReadIni.ReadIni(file_name)
        key_value = read_ini.get_value(pageElement,key)
        key = key_value.split(':')[0]
        value = key_value.split(':')[1]

        try:
            if key == 'id':
                return self.driver.find_element_by_id(value)
            elif key == 'class':
                return self.driver.find_element_by_class_name(value)
            elif key == 'name':
                return self.driver.find_element_by_name(value)
            elif key == 'tag':
                return self.driver.find_element_by_tag_name(value)
            elif key == 'linkText':
                return self.driver.find_element_by_link_text(value)
            elif key == 'cssSelector':
                return self.driver.find_element_by_css_selector(value)
            elif key == 'xpath':
                return self.driver.find_element_by_xpath(value)

        except:
            return None

因爲瀏覽器有好多種,因此也須要從新定義瀏覽器的啓動chrome

對構造函數進行修改瀏覽器

    def __init__(self,driverName):
        try:
            if driverName == 'chrome':
                self.driver = webdriver.Chrome()
            elif driverName == 'dege':
                self.driver = webdriver.Edge()
            elif driverName == 'firefox':
                self.driver = webdriver.Firefox()
            elif driverName == 'safari':
                self.driver = webdriver.Safari()

        except:
            error_message = "find not " + driverName
            print(error_message)

        self.driver.maximize_window()
        url = 'https://www.baidu.com'
        self.driver.get(url)

在項目中若是用單一瀏覽器可能會形成堵塞ide

因此就須要多個瀏覽器同時跑腳本,不至於一個瀏覽器的壓力太大函數

# conding=utf-8

from selenium import webdriver
import time
from auto_learning.config import ReadIni #導入讀取的配置文件的py文件,若是不能導入檢查是不是包(__init__.py文件存在 )文件

class FindElement(object):
    def __init__(self,i):
        self.driver = self.get_driver(i)
        time.sleep(3)
        url = 'https://www.baidu.com'
        self.driver.get(url)
        self.driver.maximize_window()

    def get_driver(self,i):
        if i == 0:
            return webdriver.Chrome()
        elif i == 1:
            return webdriver.Edge()
        elif i == 2:
            return webdriver.Firefox()
        elif i == 3:
            return webdriver.Safari()

    def get_Element(self,pageElement,key):
        # 實例化ReadIni
        file_name = '../config/BaiduElement.ini'
        read_ini = ReadIni.ReadIni(file_name)
        key_value = read_ini.get_value(pageElement,key)
        key = key_value.split(':')[0]
        value = key_value.split(':')[1]

        try:
            if key == 'id':
                return self.driver.find_element_by_id(value)
            elif key == 'class':
                return self.driver.find_element_by_class_name(value)
            elif key == 'name':
                return self.driver.find_element_by_name(value)
            elif key == 'tag':
                return self.driver.find_element_by_tag_name(value)
            elif key == 'linkText':
                return self.driver.find_element_by_link_text(value)
            elif key == 'cssSelector':
                return self.driver.find_element_by_css_selector(value)
            elif key == 'xpath':
                return self.driver.find_element_by_xpath(value)

        except:
            return None


if __name__ == '__main__':
    for i in range(3):
        findElement = FindElement(i)
        input_text = findElement.get_Element('BaiduHomeElement','baidu_search_input')
相關文章
相關標籤/搜索