方案说明
功能要求:实现网页加载后将页面截取成长图片
涉及模块:PyQT5 PIL
逻辑说明:
1:完成窗口设置,利用PyQT5 QWebEngineView加载网页地址,待网页加载完成后,调用check_pag;
class MainWindow(QMainWindow): def __init__(self, parent=None): super(MainWindow, self).__init__(parent) self.setWindowTitle('易哈佛') self.temp_height = 0 self.setWindowFlag(Qt.WindowMinMaxButtonsHint, False) # 禁用最大化,最小化 # self.setWindowFlag(Qt.WindowStaysOnTopHint, True) # 窗口顶置 self.setWindowFlag(Qt.FramelessWindowHint, True) # 窗口无边框 def urlScreenShot(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) def get_page_size(self): size = self.browser.page().contentsSize() self.set_height = size.height() self.set_width = size.width() return size.width(), size.height() def chose_screen(self): width, height = 750, 1370 desktop = QApplication.desktop() screen_count = desktop.screenCount() for i in range(0, screen_count): rect = desktop.availableGeometry(i) s_width, s_height = rect.width(), rect.height() if s_width > width and s_height > height: return QRect(rect.left(), rect.top(), width, height) return QRect(0, 0, width, height) if __name__ == '__main__': app = QApplication(sys.argv) win = MainWindow() win.show() app.exit(app.exec_())
2:收集页面高度,并计算分次截屏的次数和余量高度;实例化图片合并工具,设置定时器,超时信号发出后,执行exe_command;
def check_page(self): p_width, p_height = self.get_page_size() self.page, self.over_flow_size = divmod(p_height, self.height()) if self.page == 0: self.page = 1 self.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用来控制截图次数,并在每次截图完成后控制网页向下滑屏幕的高度;所有的页面都已截取时,完成图片合并。
def exe_command(self): if self.page > 0: self.screen_shot() self.run_js() elif self.page < 0: self.timer.stop() self.ssm.image_merge() self.close() elif self.over_flow_size > 0: self.screen_shot() self.page -= 1 def run_js(self): script = """ var scroll = function (dHeight) { var t = document.documentElement.scrollTop var h = document.documentElement.scrollHeight dHeight = dHeight || 0 var current = t + dHeight if (current > h) { window.scrollTo(0, document.documentElement.clientHeight) } else { window.scrollTo(0, current) } } """ command = script + '\n scroll({})'.format(self.height()) self.browser.page().runJavaScript(command)
4:screen_shot在每次截图完成后将图片保存,并将图片对象由图片合并根据保存到列表中。
def screen_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:截图合并工具,在每次截图完成后将图片对象保存,完成余量截图的重绘和截图的合并。
class ScreenShotMerge(): def __init__(self, page, over_flow_size): self.im_list = [] self.page = page self.over_flow_size = over_flow_size self.get_path() def get_path(self): self.root_path = Path(__file__).parent.joinpath('temp') if not self.root_path.exists(): self.root_path.mkdir(parents=True) self.save_path = self.root_path.joinpath('merge.png') def add_im(self, path): if len(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) def get_new_size(self): max_width = 0 total_height = 0 # 计算合成后图片的宽度(以最宽的为准)和高度 for img in self.im_list: width, height = img.size if width > max_width: max_width = width total_height += height return max_width, total_height def image_merge(self, ): if len(self.im_list) > 1: max_width, total_height = self.get_new_size() # 产生一张空白图 new_img = Image.new('RGB', (max_width - 15, total_height), 255) x = y = 0 for img in self.im_list: width, height = img.size new_img.paste(img, (x, y)) y += height new_img.save(self.save_path) print('截图成功:', self.save_path) else: obj = self.im_list[0] width, height = obj.size left, top, right, bottom = 0, 0, width, height box = (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) def reedit_image(self, path): obj = Image.open(path) width, height = obj.size left, top, right, bottom = 0, height - self.over_flow_size, width, height box = (left, top, right, bottom) region = obj.crop(box) return region
截图功能完整代码
#!/usr/bin/env python # -*- coding:UTF-8 -*- # Author:Leslie-x import sys from PyQt5.QtCore import * from PyQt5.QtWidgets import * from PyQt5.QtWebEngineWidgets import * from PIL import Image from pathlib import Path class ScreenShotMerge(): def __init__(self, page, over_flow_size): self.im_list = [] self.page = page self.over_flow_size = over_flow_size self.get_path() def get_path(self): self.root_path = Path(__file__).parent.joinpath('temp') if not self.root_path.exists(): self.root_path.mkdir(parents=True) self.save_path = self.root_path.joinpath('merge.png') def add_im(self, path): if len(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) def get_new_size(self): max_width = 0 total_height = 0 # 计算合成后图片的宽度(以最宽的为准)和高度 for img in self.im_list: width, height = img.size if width > max_width: max_width = width total_height += height return max_width, total_height def image_merge(self, ): if len(self.im_list) > 1: max_width, total_height = self.get_new_size() # 产生一张空白图 new_img = Image.new('RGB', (max_width - 15, total_height), 255) x = y = 0 for img in self.im_list: width, height = img.size new_img.paste(img, (x, y)) y += height new_img.save(self.save_path) print('截图成功:', self.save_path) else: obj = self.im_list[0] width, height = obj.size left, top, right, bottom = 0, 0, width, height box = (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) def reedit_image(self, path): obj = Image.open(path) width, height = obj.size left, top, right, bottom = 0, height - self.over_flow_size, width, height box = (left, top, right, bottom) region = obj.crop(box) return region class MainWindow(QMainWindow): def __init__(self, parent=None): super(MainWindow, self).__init__(parent) self.setWindowTitle('易哈佛') self.temp_height = 0 self.setWindowFlag(Qt.WindowMinMaxButtonsHint, False) # 禁用最大化,最小化 # self.setWindowFlag(Qt.WindowStaysOnTopHint, True) # 窗口顶置 self.setWindowFlag(Qt.FramelessWindowHint, True) # 窗口无边框 def urlScreenShot(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) def get_page_size(self): size = self.browser.page().contentsSize() self.set_height = size.height() self.set_width = size.width() return size.width(), size.height() def chose_screen(self): width, height = 750, 1370 desktop = QApplication.desktop() screen_count = desktop.screenCount() for i in range(0, screen_count): rect = desktop.availableGeometry(i) s_width, s_height = rect.width(), rect.height() if s_width > width and s_height > height: return QRect(rect.left(), rect.top(), width, height) return QRect(0, 0, width, height) def check_page(self): p_width, p_height = self.get_page_size() self.page, self.over_flow_size = divmod(p_height, self.height()) if self.page == 0: self.page = 1 self.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() def exe_command(self): if self.page > 0: self.screen_shot() self.run_js() elif self.page < 0: self.timer.stop() self.ssm.image_merge() self.close() elif self.over_flow_size > 0: self.screen_shot() self.page -= 1 def run_js(self): script = """ var scroll = function (dHeight) { var t = document.documentElement.scrollTop var h = document.documentElement.scrollHeight dHeight = dHeight || 0 var current = t + dHeight if (current > h) { window.scrollTo(0, document.documentElement.clientHeight) } else { window.scrollTo(0, current) } } """ command = script + '\n scroll({})'.format(self.height()) self.browser.page().runJavaScript(command) def screen_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) if __name__ == '__main__': url = 'http://blog.sina.com.cn/lm/rank/focusbang//' app = QApplication(sys.argv) win = MainWindow() win.urlScreenShot(url) win.show() app.exit(app.exec_())
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。