python自動化封裝

# -*- coding:utf-8 -*-
from selenium import webdriver
import time

OVER_TIME = 5
BASE_URL = "https://passport.cnblogs.com/user/signin?ReturnUrl=https%3A%2F%2Fwww.cnblogs.com%2F"


class Driver(object):

    def __new__(cls, *args, **kw):
        """
        使用單例模式將類設置為運行時只有一個實例，在其他Python類中使用基類時，
        可以創建多個對象，保證所有的對象都是基於一個瀏覽器
        """
        if not hasattr(cls, '_instance'):
            orig = super(Driver, cls)
            cls._instance = orig.__new__(cls, *args, **kw)
        return cls._instance

    def start(self, url=BASE_URL, driver_name="Firefox"):
        """
        啟動瀏覽器
        :param url: 測試地址
        :param driver_name: 在啟動時設置瀏覽器的類型
        :return:
        """
        if driver_name == "Firefox":
            self.driver = webdriver.Firefox()
        elif driver_name == "Ie":
            self.driver = webdriver.Ie()
        else:
            self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(OVER_TIME)
        self.driver.get(url)
        self.driver.maximize_window()

    def get_url(self):
        """返回瀏覽器的地址"""
        return BASE_URL

    def find_element(self, by, value):
        """
        這里添加了一個OVER_TIME作為查找元素的超時次數，根據系統的實際情況設置OVER_TIME的大小
        """
        for i in range(OVER_TIME):
            try:
                return self.driver.find_element(by=by, value=value)
            except Exception as e:
                print(e)

    def find_elements(self, by, value):
        """與find_element一致"""
        for i in range(OVER_TIME):
            try:
                return self.driver.find_elements(by=by, value=value)
            except Exception as e:
                print(e)

    def find_display_elements(self, by, value):
        """
        查找狀態為displayed的元素集合，當查找一類元素時，
        經常出現有些元素是不可見的情況，此函數屏蔽那些不可見的元素
        """
        for i in range(OVER_TIME):
            try:
                elements = self.driver.find_elements(by=by, value=value)
                num = elements.__len__()
            except Exception as e:
                print(e)
                time.sleep(1)
            if num >= 1:
                break
        display_element = []
        # 將可見的元素放到列表中， 並返回
        for j in range(num):
            element = elements.__getitem__(j)
            if element.is_displayed():
                display_element.append(element)
        return display_element

    def is_element_present(self, By, Value):
        """判斷元素是否存在"""
        try:
            self.driver.find_element(by=By, value=Value)
            return True
        except Exception as e:
            print(e)
            return False

    def close(self):
        self.driver.close()

    def quit(self):
        u"""退出瀏覽器"""
        self.driver.quit()

if __name__ == "__main__":
    page = Driver()
    page.start()