中級篇之9-把截圖類方法封裝到前面的BasePage.py

本文介紹把截圖類方法封裝到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:

"""

def forward(self):

"""

瀏覽器前進按鈕

:param none:

"""

def open_url(self, url):

"""

打開url站點

:param url:

"""

def quit_browser(self):

"""

關閉並停止瀏覽器服務

:param none:

"""

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 :

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)

test = TestScreenshot()

test.test_take_screen()

運行後，可以在根目錄下Screenshots文件夾里找到百度首頁截圖。

本文就介紹了截圖類方法添加到BasePage里，介紹了如何保存到根目錄下的Screenshots文件夾。

歡迎關注凱哥公眾號：凱哥Java

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！