python实现自动网页截图并裁剪图片_Python

python实现自动网页截图并裁剪图片

2021-03-24 00:17Erick-LONG Python

这篇文章主要为大家详细介绍了python实现自动网页截图并裁剪图片，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

本文实例为大家分享了python自动网页截图并裁剪图片的具体代码，供大家参考，具体内容如下

代码：

									# coding=utf-8

									import time

									from selenium import webdriver

									from selenium.webdriver.chrome.options import Options

									from PIL import Image

									import os

									all_urls = ['http:/****edit']

									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://x*****e")

									  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('***')

									  pwd.send_keys('***')

									  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

									def get_screen(driver,urls):

									  count = 1

									  for url in urls:

									    driver.get(url)

									    print('正在抓取--> %s'% url)

									    count +=1

									    time.sleep(2)

									    uid = url.split('/')[-2]

									    cid = url.split('/')[-5]

									    driver.get_screenshot_as_file("./screen_shot/{}-{}.png".format(uid,cid))

									    print("创意--> {}-{}.png 已经保存".format(uid,cid))

									    print('还剩 %s 个'% str(len(urls)-count))

									def crop_img():

									  for img in os.listdir('./screen_shot'):

									    if img.endswith('.png'):

									      print('%s裁剪中。。'% img)

									      im = Image.open('./screen_shot/%s'% img)

									      x = 755

									      y = 162

									      w = 383

									      h = 346

									      region = im.crop((x, y, x+w, y+h))

									      region.save("./screenshot_final/%s" % img)

									if __name__ == '__main__':

									  driver = login()

									  get_screen(driver,all_urls)

									  driver.quit()

									  print('所有抓取结束')

									  crop_img()

									  print('所有裁剪结束')