使用selenium进行网页截图,本文使用方法皆为显式情境下调试,未涉及隐式(无界面)与协议(设备模拟)。
首先定义驱动driver
from selenium import webdriver
driver = webdriver.Chrome()
#可指定目标页面网址,不指定页面默认为空白页
#url1='???'
#driver.get(url1)
拼接图片涉及
from PIL import Image
import numpy as np
一、基本操作
首先是最基础的截图,界面截图与某一部件截图
#部件截图
driver.find_element_by_id('captchaimg').screenshot('C:/xxxx/xx.png')
#屏幕截图
driver.get_screenshot_as_file('C:/xxxx/xx.png')
二、整页截图
以上方法只能截取当前页面的图片,那么当网页长度超出屏幕长度时呢?(即需要滚动)
2.1截取再拼接
window_height = self.driver.execute_script("return window.screen.height") #屏幕高度
self.driver.set_window_size(650, window_height) #设置浏览器宽高
time.sleep(1)
page_height = self.driver.execute_script('return document.documentElement.scrollHeight') # 页面高度
print(window_height, page_height)
n1 = page_height// window_height
if n1 == 0: #判断是否需要滚动
self.driver.get_screenshot_as_file(r'{}\{}.png'.format(self.path_last,Account_number )) #指定截图保存位置
else:
self.driver.set_window_size(650, window_height)
for j in range(n1+1):
self.driver.execute_script("window.scrollBy(0,{})".format(j*window_height))
time.sleep(0.5)
#driver.find_element(By.ID,'frmRecPODetail').screenshot(r'.png'.format(i)) #只截取部分
self.driver.get_screenshot_as_file(r'{}\{}.png'.format(self.path_mid, j)) # 截屏
#拼图时间到
target_img=os.listdir(self.path_mid)
base_mat = np.atleast_2d(Image.open(os.path.join(self.path_mid,target_img[0])))
for i in target_img[1:]:
mat = np.atleast_2d(Image.open(os.path.join(self.path_mid,i))) # 打开截图并转为二维矩阵
base_mat = np.append(base_mat, mat, axis=0) # 拼接图片的二维矩阵
Image.fromarray(base_mat).save(os.path.join(self.path_last,'{}.png'.format(Account_number)))
2.2利用javascript进行打印截图
# 打开谷歌浏览器
chrome_options = webdriver.ChromeOptions()
settings = {
"recentDestinations": [{
"id": "Save as PDF",
"origin": "local",
"account": ""
}],
"selectedDestinationId": "Save as PDF",
"version": 2, # 另存为pdf,1 是默认打印机
"isHeaderFooterEnabled": True, # 是否勾选页眉和页脚
# "customMargins": {},
# "marginsType": 2,#边距(2是最小值、0是默认)
# "scaling": 100,
# "scalingType": 3,
# "scalingTypePdf": 3,
# "isLandscapeEnabled": True, # 若不设置该参数,默认值为纵向
"isCssBackgroundEnabled": True,
"mediaSize": {
"height_microns": 297000,
"name": "ISO_A4",
"width_microns": 210000,
"custom_display_name": "A4"
},
}
chrome_options.add_argument('--enable-print-browser')
# chrome_options.add_argument('--headless') #headless模式下,浏览器窗口不可见,应当可提高效率
prefs = {
'printing.print_preview_sticky_settings.appState': json.dumps(settings),
'savefile.default_directory': r'{}'.format(self.path_last) # 此处填写你希望文件保存的路径,可填写your file path默认下载地址
}
chrome_options.add_argument('--kiosk-printing') # 静默打印,无需用户点击打印页面的确定按钮
chrome_options.add_experimental_option('prefs', prefs)
self.driver = webdriver.Chrome(chrome_options=chrome_options,executable_path=self.path_dri)
self.driver.execute_script(
'document.title="{}.pdf";window.print();'.format(Account_number)) # 利用js修改网页的title,该title最终就是PDF文件名,利用js的window.print可以快速调出浏览器打印窗口,避免使用热键ctrl+P
三、某部件长截图
在网页中,或许存在某一部件,该部件采取滚动条的形式(即需要滚动该部件进行截屏)文章来源:https://www.toymoban.com/news/detail-511511.html
同理,使用单位置截屏滚动再拼接的方法。文章来源地址https://www.toymoban.com/news/detail-511511.html
tabe = self.driver.find_element_by_xpath('sdasd')
tr0 = tabe.find_elements_by_xpath('tbody')[0] #找到位置
#我这里是有多行tr,故统计tr数量
tr0 = tr0.find_elements_by_xpath('tr')
self.driver.set_window_size(1350,650) #调整浏览器尺寸,也可以直接打印部件
n1 = (len(tr0)-1)//5 #5是我同窗可显示数目
if n1==0: #展示数量小于5,无需滚动
self.pic += 1
self.driver.execute_script('document.documentElement.scrollTop=10000')
self.driver.get_screenshot_as_file(r'{}\{}.png'.format(self.path1,'0'))
else:
for i in range(n1+1):
self.driver.execute_script("arguments[0].scrollIntoView(false)",tr0[i*5])
self.driver.execute_script('document.documentElement.scrollTop=10000')
# driver.find_element(By.ID,'frmRecPODetail').screenshot(r'C:\xx\{}.png'.format(i)) #只截取部分
self.driver.get_screenshot_as_file(r'{}\{}.png'.format(self.path1,i)) # 截屏
self.pic+=1
else:
if (len(tr0)-1)%5!=0:
self.driver.execute_script("arguments[0].scrollIntoView(false)",tr0[i*5+(len(tr0)-1)%5])
self.driver.execute_script('document.documentElement.scrollTop=10000') # 将页面滚动到底部
# driver.find_element(By.ID,'frmRecPODetail').screenshot(r'C:\xx\{}.png'.format(i+1))
self.driver.get_screenshot_as_file(r'{}\{}.png'.format(self.path1,i+1)) # 截屏
self.pic += 1
else:
self.driver.execute_script('document.documentElement.scrollTop=10000') # 将页面滚动到底部
self.driver.get_screenshot_as_file(r'{}\{}.png'.format(self.path1,i)) # 截屏,替换最后一张
#若需要拼接,需要注意第一张图片在循环之外打开(以它为底)
#拼接部分
# target_img=os.listdir(self.path3)
# base_mat = np.atleast_2d(Image.open(os.path.join(self.path3,target_img[0])))
# for i in target_img[1:]:
# mat = np.atleast_2d(Image.open(os.path.join(self.path3,i))) # 打开截图并转为二维矩阵
# base_mat = np.append(base_mat, mat, axis=0) # 拼接图片的二维矩阵
# Image.fromarray(base_mat).save(os.path.join(self.path1,'521.png'))
到了这里,关于selenium进行网页长截图,模块长截图的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!