截图功能的设定

from selenium import webdriver

import os
import time


def create_img(driver, imgname):
    base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

    '''存放截图路经'''
    phone_path = os.path.join(
        base_dir + '\\TestReport\\screenshot\\' + imgname)

    driver.get_screenshot_as_file(phone_path)


if __name__ == "__main__":
    driver = webdriver.Chrome()
    driver.get("http://www.baidu.com")
    time.sleep(0.2)
    create_img(driver, 'baidu.png')
    driver.quit()