《一頭扎進》系列之Python+Selenium框架實戰篇4- 價值好幾K的框架，呵！這個框架有點意思啊！！！

1.簡介

　　前面文章，我們實現了框架的一部分功能，包括日志類和瀏覽器引擎類的封裝，今天我們繼續封裝一個基類和介紹如何實現POM。關於基類，是這樣定義的：把一些常見的頁面操作的selenium封裝到base_page.py這個類文件，以后每個POM中的頁面類，都繼承這個基類，這樣每個頁面類都有基類的方法，這個我們會在這篇文章由宏哥實現。

2.項目層級結構

1. 上一篇中我們已經創建好了項目層級結構，具體項目層級結構如下圖。這里不再贅述，相關文件也如下：

3. 定位和截圖類封裝

1. 在實現封裝基類里，我們實現了元素八大方式的定位和截圖類封裝。

2. 基類base_page.py的具體實現代碼，這里就封裝了幾個常用方法，其他方法，你自己去練習封裝下。

3.1 代碼實現：

3.2 參考代碼：

# -*- coding:utf-8 -*-

# 1.先設置編碼，utf-8可支持中英文，如上，一般放在第一行

# 2.注釋：包括記錄創建時間，創建人，項目名稱。
'''
Created on 2019-12-20
@author: 北京-宏哥   QQ交流群：705269076
Project: 《《一頭扎進》系列之Python+Selenium框架設計篇4- 價值好幾K的框架，不看別后悔，過時不候
'''

# 3.導入模塊

import time
from selenium.common.exceptions import NoSuchElementException
import os.path
from automation_framework_demo.framework.logger import Logger

# create a logger instance
logger = Logger(logger="BasePage").getlog()

class BasePage(object):
    """
    定義一個頁面基類，讓所有頁面都繼承這個類，封裝一些常用的頁面操作方法到這個類
    """
    def __init__(self, driver):
        self.driver = driver

    # quit browser and end testing
    def quit_browser(self):
        self.driver.quit()

     # 瀏覽器前進操作
    def forward(self):
        self.driver.forward()
        logger.info("Click forward on current page.")

    # 瀏覽器后退操作
    def back(self):
        self.driver.back()
        logger.info("Click back on current page.")

    # 隱式等待
    def wait(self, seconds):
        self.driver.implicitly_wait(seconds)
        logger.info("wait for %d seconds." % seconds)

    # 點擊關閉當前窗口
    def close(self):
        try:
            self.driver.close()
            logger.info("Closing and quit the browser.")
        except NameError as e:
            logger.error("Failed to quit the browser with %s" % e)

    # 保存圖片
    def get_windows_img(self):
        """
        在這里我們把file_path這個參數寫死，直接保存到我們項目根目錄的一個文件夾.\Screenshots下
        """
        file_path = os.path.dirname(os.path.abspath('.')) + '/screenshots/'
        rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
        screen_name = file_path + rq + '.png'
        try:
            self.driver.get_screenshot_as_file(screen_name)
            logger.info("Had take screenshot and save to folder : /screenshots")
        except NameError as e:
            logger.error("Failed to take screenshot! %s" % e)
            self.get_windows_img()

    # 定位元素方法
    def find_element(self, selector):
        """
         這個地方為什么是根據=>來切割字符串，請看頁面里定位元素的方法
         submit_btn = "id=>su"
         login_lnk = "xpath => //*[@id='u1']/a[7]"  # 百度首頁登錄鏈接定位
         如果采用等號，結果很多xpath表達式中包含一個=，這樣會造成切割不准確，影響元素定位
        :param selector:
        :return: element
        """
        element = ''
        if '=>' not in selector:
            return self.driver.find_element_by_id(selector)
        selector_by = selector.split('=>')[0]
        selector_value = selector.split('=>')[1]
        print(selector_value)
        if selector_by == "i" or selector_by == 'id':
            try:
                element = self.driver.find_element_by_id(selector_value)
                logger.info("Had find the element \' %s \' successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                logger.error("NoSuchElementException: %s" % e)
                self.get_windows_img()  # take screenshot
        elif selector_by == "n" or selector_by == 'name':
            element = self.driver.find_element_by_name(selector_value)
        elif selector_by == "c" or selector_by == 'class_name':
            element = self.driver.find_element_by_class_name(selector_value)
        elif selector_by == "l" or selector_by == 'link_text':
            element = self.driver.find_element_by_link_text(selector_value)
        elif selector_by == "p" or selector_by == 'partial_link_text':
            element = self.driver.find_element_by_partial_link_text(selector_value)
        elif selector_by == "t" or selector_by == 'tag_name':
            element = self.driver.find_element_by_tag_name(selector_value)
        elif selector_by == "x" or selector_by == 'xpath':
            try:
                element = self.driver.find_element_by_xpath(selector_value)
                logger.info("Had find the element \' %s \' successful "
                            "by %s via value: %s " % (element.text, selector_by, selector_value))
            except NoSuchElementException as e:
                logger.error("NoSuchElementException: %s" % e)
                self.get_windows_img()
        elif selector_by == "s" or selector_by == 'selector_selector':
            element = self.driver.find_element_by_css_selector(selector_value)
        else:
            raise NameError("Please enter a valid type of targeting elements.")
        print(element)
        return element

    # 輸入
    def type(self, selector, text):
        el = self.find_element(selector)
        el.clear()
        try:
            el.send_keys(text)
            logger.info("Had type \' %s \' in inputBox" % text)
        except NameError as e:
            logger.error("Failed to type in input box with %s" % e)
            self.get_windows_img()

    # 清除文本框
    def clear(self, selector):

        el = self.find_element(selector)
        try:
            el.clear()
            logger.info("Clear text in input box before typing.")
        except NameError as e:
            logger.error("Failed to clear in input box with %s" % e)
            self.get_windows_img()

    # 點擊元素
    def click(self, selector):

        el = self.find_element(selector)
        try:
            el.click()
            logger.info("The element \' %s \' was clicked." % el)
        except NameError as e:
            logger.error("Failed to click the element with %s" % e)

    # 或者網頁標題
    def get_page_title(self):
        logger.info("Current page title is %s" % self.driver.title)
        return self.driver.title

    @staticmethod
    def sleep(seconds):
        time.sleep(seconds)
        logger.info("Sleep for %d seconds" % seconds)

4. pageObjects文件夾下相關代碼

1.頁面對象中，百度主頁的元素定位和簡單的操作函數，頁面類主要是元素定位和頁面操作寫成函數，供測試類調用。

baidu_homepage.py

4.1 代碼實現：

4.2 參考代碼：

# -*- coding:utf-8 -*-

# 1.先設置編碼，utf-8可支持中英文，如上，一般放在第一行

# 2.注釋：包括記錄創建時間，創建人，項目名稱。
'''
Created on 2019-12-20
@author: 北京-宏哥   QQ交流群：705269076
Project: 《《一頭扎進》系列之Python+Selenium框架設計篇4- 價值好幾K的框架，不看別后悔，過時不候
'''
# 3.導入模塊

from automation_framework_demo.framework.base_page import BasePage

class HomePage(BasePage):
    input_box = "id=>kw"
    search_submit_btn = "xpath=>//*[@id='su']"

    #百度新聞入口
    #news_link = "xpath=>//*[@id='u1']/a[@name='tj_trnews']"
    news_link = "xpath=>//*[@id='u1']/a[@name='tj_trnews']"

    def type_search(self, text):
        self.type(self.input_box, text)

    def send_submit_btn(self):
        self.click(self.search_submit_btn)

    def click_news(self,):
        self.click(self.news_link)
        self.sleep(2)

這里注意下元素定位寫法，=>和base_page.py中find_element()方法元素定位切割有關系，網上有些人寫根據逗號切割或者等號切割，在實際使用xpath定位，發現單獨逗號或者單獨等號切割都不精確，造成元素定位失敗。

5. 測試類的寫法舉例

1.新建一個測試類baidu_search1.py。

baidu_search1.py

5.1 代碼實現：

5.2 參考代碼：

# -*- coding:utf-8 -*-

# 1.先設置編碼，utf-8可支持中英文，如上，一般放在第一行

# 2.注釋：包括記錄創建時間，創建人，項目名稱。
'''
Created on 2019-12-20
@author: 北京-宏哥   QQ交流群：705269076
Project: 《《一頭扎進》系列之Python+Selenium框架設計篇4- 價值好幾K的框架，不看別后悔，過時不候
'''

# 3.導入模塊

import time
import unittest
from automation_framework_demo.framework.browser_engine import BrowserEngine
from automation_framework_demo.pageobjects.baidu_homepage import HomePage


class BaiduSearch(unittest.TestCase):

    
    def setUpClass(cls):
        """
        測試固件的setUp()的代碼，主要是測試的前提准備工作
        :return:
        """
        browse = BrowserEngine(cls)
        cls.driver = browse.open_browser(cls)

    
    def tearDownClass(cls):
        """
        測試結束后的操作，這里基本上都是關閉瀏覽器
        :return:
        """
        cls.driver.quit()

    def test_baidu_search(self):
        """
        這里一定要test開頭，把測試邏輯代碼封裝到一個test開頭的方法里。
        :return:
        """
       # self.driver.find_element_by_id('kw').send_keys('selenium')
       # time.sleep(1)
        homepage = HomePage(self.driver)
        homepage.type_search('selenium')  # 調用頁面對象中的方法
        homepage.send_submit_btn()  # 調用頁面對象類中的點擊搜索按鈕方法
        time.sleep(2)
        homepage.get_windows_img()  # 調用基類截圖方法
        print(self.driver.title)
        try:
            assert('selenium' in HomePage.get_page_title(self))
            print('Test Pass.')
        except Exception as e:
            print('Test Fail.', format(e))

    def test_search2(self):
        homepage = HomePage(self.driver)
        homepage.type_search('python')  # 調用頁面對象中的方法
        homepage.send_submit_btn()  # 調用頁面對象類中的點擊搜索按鈕方法
        time.sleep(2)
        homepage.get_windows_img()  # 調用基類截圖方法


if __name__ == '__main__':
    unittest.main()

5.3 運行結果：

運行代碼后，控制台打印如下圖的結果

5.4 代碼說明

homepage = HomePage(self.driver)

      上面這行代碼要注意，意思是：到一個頁面，第一件事情是初始化這個頁面的一個頁面對象實例。注意，一定要帶self.driver，不然會報錯，這個self.driver，可以這樣理解：在當前測試類里面，self.driver是來自瀏覽器引擎類中方法得到的，在初始化一個頁面對象

的時候，也把這個來自瀏覽器引擎類的driver給賦值給當前的頁面對象，這樣，才能執行頁面對象或者基類里面的相關driver方法。寫多了selenium的自動化腳本，你會明白，最重要的是保持前后driver的唯一性。

5.5 生成圖片

1.測試結果：會在logs文件夾生成一個日志文件，也會在screenshots文件夾生成一個png圖片。日志看過了，這里我們看一下圖片

2.圖片內容如下圖：

6. 小結

好了，今天的分享就到這里吧！！！謝謝各位的耐心閱讀。有問題加群交流討論

 

您的肯定就是我進步的動力。如果你感覺還不錯，就請鼓勵一下吧！記得隨手點波  推薦  不要忘記哦！！！

別忘了點 推薦 留下您來過的痕跡