自动网页截图并指定元素位置裁剪图片并保存到excel表格

# coding=utf-8
import os
import time
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from PIL import Image
import xlsxwriter, xlrd
import pandas as pd


def login():
    chrome_options = Options()
    chrome_options.add_argument('--headless')
    driver = webdriver.Chrome(executable_path='./chromedriver', chrome_options=chrome_options)
    driver.set_window_size(1200, 741)
    driver.implicitly_wait(2)
    print('初始化中...')
    driver.get("http://xxxve")
    print('填写登录信息中...')
    acc = driver.find_element_by_id('login-email')
    pwd = driver.find_element_by_id('login-pass')
    btn = driver.find_element_by_tag_name('button')
    acc.send_keys('zhxxxm')
    pwd.send_keys('LONxxxxx$')
    btn.click()
    print('跳转到验证码页面中...')
    time.sleep(2)
    capta = driver.find_element_by_id('code')
    capta_input = input('请输入两步验证码：')
    capta.send_keys(capta_input)
    btn1 = driver.find_element_by_tag_name('button')
    btn1.click()
    time.sleep(2)
    print('跳转到创意编辑页面中...')
    return driver


faild_url = []


def parse_img(driver, url):
    cid = url.split('/')[-2]
    try:
        driver.get_screenshot_as_file("./screen_shot/{}.png".format(cid))
        element = driver.find_element_by_class_name("AdvertViewer-item")
        # print(element.location)  # 打印元素坐标
        # print(element.size)  # 打印元素大小

        left = element.location['x']
        top = element.location['y']
        right = element.location['x'] + element.size['width']
        bottom = element.location['y'] + element.size['height']

        im = Image.open("./screen_shot/{}.png".format(cid))
        im = im.crop((left, top, right, bottom))
        im.save("./screen_shot/{}.png".format(cid))
        print("创意-->{}.png 已经保存".format(cid))
    except Exception as e:
        faild_url.append(url)


def get_screen(driver, url, tem):
    try:
        driver.get(url)
        time.sleep(10)
        parse_img(driver,url)

    except Exception as e:
        print(e, url)
        faild_url.append(url)


def new_xlsx():
    # 打开excel文件
    data = xlrd.open_workbook('全xxx意.xlsx')
    book = xlsxwriter.Workbook('全行xxx意-fina.xlsx')
    # 获取第一张工作表（通过索引的方式）
    tables = data.sheets()
    for table in tables:
        worksheet = book.add_worksheet(table.name)
        nrows = table.nrows

        # 表头
        rows = table.row_values(0)
        for i, v in enumerate(rows):
            if v != '':
                worksheet.write(0, i, v)
        # 表体
        for k in range(1, nrows - 1):
            rows = table.row_values(k)
            for i, v in enumerate(rows):
                if v != '':
                    if str(int(rows[6])) + '.png' in os.listdir('./screen_shot/'):
                        image_width, image_height = Image.open('./screen_shot/' + str(int(rows[6])) + '.png').size
                        worksheet.set_column('H:H', width=58)
                        worksheet.set_row(k, height=image_height*0.8)
                        if i == 7:
                            worksheet.insert_image('H' + str(k + 1), './screen_shot/' + str(int(rows[6])) + '.png',
                                                   {'x_offset': 6, 'y_offset': 3})
                        else:
                            worksheet.write(k, i, v)
                    else:
                        worksheet.write(k, i, v)

    book.close()


if __name__ == '__main__':
    df = pd.read_excel('./全行xxxx.xlsx')
    driver1 = login()
    link_list = df['link'].tolist()
    tem_list = df['样式'].tolist()
    dict_url = zip(link_list, tem_list)

    for url, tem in dict_url:
        count = 1
        get_screen(driver1, url, tem)
        count += 1
        print('还剩 %s 个' % str(len(link_list) - count))
    driver1.quit()

    print('失败的url:', faild_url)
    print('所有抓取结束')
    new_xlsx()
    print('插入表格结束')