python实现打开Url并截图保存

安装第三方库

selenium、openpyxl

源码

from selenium import webdriver
from time import sleep
from openpyxl import load_workbook


class Excel_Get_img(object):

    def read_excel_return_dict_get_img(self):
        '''读取excel转换为字典'''
        # 读取excel
        wb = load_workbook(filename='case_data.xlsx')
        # 指定表
        sheet = wb['Sheet1']
        # 获得最大列和最大行
        # print(sheet.max_row)
        # print(sheet.max_column)
        img_name = []
        # 循环遍历图片名称
        for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, min_col=1, max_col=1):
            # print(row)
            for c in row:
                # print(c.value)
                img_name.append(c.value)
        # print(img_name)

        url = []
        #  循环遍历url
        for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, min_col=2, max_col=2):
            # print(row)
            for cell in row:
                # print(cell.value)
                url.append(cell.value)
        # print(url)

        # 将图片名称和url整合成字典格式
        res = dict(zip(img_name, url))
        print(res)

        # 循环遍历字典
        for r in res.items():
            print(r)
            # print(type(r))
            # print(r[0])
            # print(r[1])
            url = r[1]
            driver = webdriver.Chrome()
            driver.maximize_window()
            driver.get(url)
            sleep(3)
            driver.get_screenshot_as_file(f'.//img//{r[0]}.png')
            print(f'{r[0]}.png')
            driver.quit()


if __name__ == '__main__':
    e = Excel_Get_img()
    e.read_excel_return_dict_get_img()

excel格式