本文介紹把截圖類方法封裝到BasePage.py檔案裡，這個檔案是在前面Selenium方法二次封裝文章裡建立的，具體程式碼請到前面這篇裡找。我們截圖類寫死了把截圖圖片儲存到根目錄下的Screenshots資料夾裡，圖片名稱是當前系統時間，圖片字尾名是png。

新的BasePage.py內容如下：

# coding=utf-8
import os
import time

from test.logger import Logger

mylog = Logger(logger='BasePage').getlog()
class BasePage(object):
    """
    主要是把常用的幾個Selenium方法封裝到BasePage這個類，我們這裡演示以下幾個方法
    back()
    forward()
    get()
    quit()
    """

    def __init__(self, driver):
        """
        寫一個建構函式，有一個引數driver
        :param driver:
        """
        self.driver = driver

    def back(self):
        """
        瀏覽器後退按鈕
        :param none:
        """
        self.driver.back()

    def forward(self):
        """
        瀏覽器前進按鈕
        :param none:
        """
        self.driver.forward()

    def open_url(self, url):
        """
        開啟url站點
        :param url:
        """
        self.driver.get(url)

    def quit_browser(self):
        """
        關閉並停止瀏覽器服務
        :param none:
        """
        self.driver.quit()

    def take_screenshot(self):
        """
        截圖並儲存在根目錄下的Screenshots資料夾下
        :param none:
        """
        file_path = os.path.dirname(os.getcwd()) + '/Screenshots/'
        rq = time.strftime('%Y%m%d%H%M%S',time.localtime(time.time()))
        screen_name = file_path + rq + '.png'
        try :
            self.driver.get_screenshot_as_file(screen_name)
            mylog.info("開始截圖並儲存")

        except Exception as e:
            mylog.error("出現異常",format(e))


 

測試類相關程式碼如下：

# coding=utf-8
import time
from selenium import webdriver

from test.basepage import BasePage


class TestScreenshot(object):
    driver = webdriver.Chrome()
    driver.maximize_window()
    driver.implicitly_wait(10)
    basepage = BasePage(driver)

    def test_take_screen(self):
        self.basepage.open_url("https://www.baidu.com")
        time.sleep(1)
        self.basepage.take_screenshot()
        self.basepage.quit_browser()

test = TestScreenshot()
test.test_take_screen()
 

本文就介紹了截圖類方法新增到BasePage裡，介紹瞭如何儲存到根目錄下的Screenshots資料夾。