Python實(shí)現(xiàn)網(wǎng)頁(yè)截圖（PyQT5）過(guò)程解析

Python實(shí)現(xiàn)?頁(yè)截圖（PyQT5）過(guò)程解析?案說(shuō)明功能要求：實(shí)現(xiàn)?頁(yè)加載后將頁(yè)?截取成長(zhǎng)圖?涉及模塊：PyQT5PIL邏輯說(shuō)明：1：完成窗?設(shè)置，利?PyQT5QWebEngineView加載?頁(yè)地址，待?頁(yè)加載完成后，調(diào)?check_pag；classMainWindow(QMainWindow):def__init__(self,parent=None):super(MainWindow,self).__init__(parent)self.setWindowTitle('易哈佛')self.temp_height=0self.setWindowFlag(Qt.WindowMinMaxButtonsHint,禁?最?化，最?化#self.setWindowFlag(Qt.WindowStaysOnTopHint,窗?頂置窗??邊框False)#True)#self.setWindowFlag(Qt.FramelessWindowHint,True)#defurlScreenShot(self,url):self.browser=QWebEngineView()self.browser.load(QUrl(url))geometry=self.chose_screen()self.setGeometry(geometry)self.browser.loadFinished.connect(self.check_page)self.setCentralWidget(self.browser)defget_page_size(self):size=self.browser.page().contentsSize()self.set_height=size.height()self.set_width=size.width()returnsize.width(),size.height()defchose_screen(self):width,height=750,1370desktop=QApplication.desktop()screen_count=desktop.screenCount()foriinrange(0,screen_count):rect=desktop.availableGeometry(i)s_width,s_height=rect.width(),rect.height()ifs_width>widthands_height>height:returnQRect(rect.left(),rect.top(),width,height)returnQRect(0,0,width,height)if__name__=='__main__':app=QApplication(sys.argv)win=MainWindow()win.show()app.exit(app.exec_())2：收集頁(yè)??度，并計(jì)算分次截屏的次數(shù)和余量?度；實(shí)例化圖?合并?具，設(shè)置定時(shí)器，超時(shí)信號(hào)發(fā)出后，執(zhí)?exe_command；defcheck_page(self):p_width,p_height=self.get_page_size()self.page,self.over_flow_size=divmod(p_height,self.height())ifself.page==0:self.page=1self.ssm=ScreenShotMerge(self.page,self.over_flow_size)self.timer=QTimer(self)self.timer.timeout.connect(self.exe_command)self.timer.setInterval(400)self.timer.start()3：exe_command?來(lái)控制截圖次數(shù)，并在每次截圖完成后控制?頁(yè)向下滑屏幕的?度；所有的頁(yè)?都已截取時(shí)，完成圖?合并。defexe_command(self):ifself.page>0:self.screen_shot()self.run_js()elifself.page<0:self.timer.stop()self.ssm.image_merge()self.close()

elifself.over_flow_size>0:self.screen_shot()self.=1defrun_js(self):script="""varscroll=function(dHeight){vart=document.documentElement.scrollTopvarh=document.documentElement.scrollHeightdHeight=dHeight||0varcurrent=t+dHeightif(current>h){window.scrollTo(0,document.documentElement.clientHeight)}else{window.scrollTo(0,current)}}"""command=script+'\nscroll({})'.format(self.height())self.browser.page().runJavaScript(command)4：screen_shot在每次截圖完成后將圖?保存，并將圖?對(duì)象由圖?合并根據(jù)保存到列表中。defscreen_shot(self):screen=QApplication.primaryScreen()winid=self.browser.winId()pix=screen.grabWindow(int(winid))name='{}/temp.png'.format(self.ssm.root_path)pix.save(name)self.ssm.add_im(name)5：截圖合并?具，在每次截圖完成后將圖?對(duì)象保存，完成余量截圖的重繪和截圖的合并。classScreenShotMerge():def__init__(self,page,over_flow_size):self.im_list=[]self.page=pageself.over_flow_size=over_flow_sizeself.get_path()defget_path(self):self.root_path=Path(__file__).parent.joinpath('temp')ifnotself.root_path.exists():self.root_path.mkdir(parents=True)self.save_path=self.root_path.joinpath('merge.png')defadd_im(self,path):iflen(self.im_list)==self.page:im=self.reedit_image(path)else:im=Image.open(path)im.save('{}/{}.png'.format(self.root_path,len(self.im_list)+1))self.im_list.append(im)defget_new_size(self):max_width=0total_height=0#計(jì)算合成后圖?的寬度（以最寬的為準(zhǔn)）和?度f(wàn)orimginself.im_list:width,height=img.sizeifwidth>max_width:max_width=widthtotal_height+=heightreturnmax_width,total_heightdefimage_merge(self,):iflen(self.im_list)>1:max_width,total_height=self.get_new_size()#產(chǎn)??張空?圖new_img=Image.new('RGB',(max_width-15,total_height),255)x=y=0forimginself.im_list:width,height=img.sizenew_img.paste(img,(x,y))y+=heightnew_img.save(self.save_path)print('截圖成功:',self.save_path)

else:obj=self.im_list[0]width,height=obj.sizeleft,top,right,bottom=0,0,width,heightbox=(left,top,right,bottom)region=obj.crop(box)new_img=Image.new('RGB',(width,height),255)new_img.paste(region,box)new_img.save(self.save_path)print('截圖成功:',self.save_path)defreedit_image(self,path):obj=Image.open(path)width,height=obj.sizeleft,top,right,bottom=0,height-self.over_flow_size,width,heightbox=(left,top,right,bottom)region=obj.crop(box)returnregion截圖功能完整代碼#!/usr/bin/envpython#-*-coding:UTF-8-*-#Author:Leslie-ximportsysfromPyQt5.QtCoreimport*fromPyQt5.QtWidgetsimport*fromPyQt5.QtWebEngineWidgetsimport*fromPILimportImagefrompathlibimportPathclassScreenShotMerge():def__init__(self,page,over_flow_size):self.im_list=[]self.page=pageself.over_flow_size=over_flow_sizeself.get_path()defget_path(self):self.root_path=Path(__file__).parent.joinpath('temp')ifnotself.root_path.exists():self.root_path.mkdir(parents=True)self.save_path=self.root_path.joinpath('merge.png')defadd_im(self,path):iflen(self.im_list)==self.page:im=self.reedit_image(path)else:im=Image.open(path)im.save('{}/{}.png'.format(self.root_path,len(self.im_list)+1))self.im_list.append(im)defget_new_size(self):max_width=0total_height=0#計(jì)算合成后圖?的寬度（以最寬的為準(zhǔn)）和?度f(wàn)orimginself.im_list:width,height=img.sizeifwidth>max_width:max_width=widthtotal_height+=heightreturnmax_width,total_heightdefimage_merge(self,):iflen(self.im_list)>1:max_width,total_height=self.get_new_size()#產(chǎn)??張空?圖new_img=Image.new('RGB',(max_width-15,total_height),255)x=y=0forimginself.im_list:width,height=img.sizenew_img.paste(img,(x,y))y+=heightnew_img.save(self.save_path)print('截圖成功:',self.save_path)else:obj=self.im_list[0]width,height=obj.sizeleft,top,right,bottom=0,0,width,heightbox=(left,top,right,bottom)region=obj.crop(box)

new_img=Image.new('RGB',(width,height),255)new_img.paste(region,box)new_img.save(self.save_path)print('截圖成功:',self.save_path)defreedit_image(self,path):obj=Image.open(path)width,height=obj.sizeleft,top,right,bottom=0,height-self.over_flow_size,width,heightbox=(left,top,right,bottom)region=obj.crop(box)returnregionclassMainWindow(QMainWindow):def__init__(self,parent=None):super(MainWindow,self).__init__(parent)self.setWindowTitle('易哈佛')self.temp_height=0self.setWindowFlag(Qt.WindowMinMaxButtonsHint,False)#禁?最?化，最?化#self.setWindowFlag(Qt.WindowStaysOnTopHint,True)#窗?頂置self.setWindowFlag(Qt.FramelessWindowHint,True)#窗??邊框defurlScreenShot(self,url):self.browser=QWebEngineView()self.browser.load(QUrl(url))geometry=self.chose_screen()self.setGeometry(geometry)self.browser.loadFinished.connect(self.check_page)self.setCentralWidget(self.browser)defget_page_size(self):size=self.browser.page().contentsSize()self.set_height=size.height()self.set_width=size.width()returnsize.width(),size.height()defchose_screen(self):width,height=750,1370desktop=QApplication.desktop()screen_count=desktop.screenCount()foriinrange(0,screen_count):rect=desktop.availableGeometry(i)s_width,s_height=rect.width(),rect.height()ifs_width>widthands_height>height:returnQRect(rect.left(),rect.top(),width,height)returnQRect(0,0,width,height)defcheck_page(self):p_width,p_height=self.get_page_size()self.page,self.over_flow_size=divmod(p_height,self.height())ifself.page==0:self.page=1self.ssm=ScreenShotMerge(self.page,self.over_flow_size)self.timer=QTimer(self)self.timer.timeout.connect(self.exe_command)self.timer.setInterval(400)self.timer.start()defexe_command(self):ifself.page>0:self.screen_shot()self.run_js()elifself.page<0:self.timer.stop()self.ssm.image_merge()self.close()elifself.over_flow_size>