python使用技巧-调用pyqt5窗口截图
安装pyqt5
安装通过 PIP 进行即可。
pip install pyqt5
还可以通过pycharm安装pyqt5。
以下为屏幕截图所需代码，把底部注释代码去掉即可运行，点击运行会弹出窗口。
点击截图即可使用，用鼠标选定区域截图，弹出保存窗口，如没有保存则弹出提示。
代码实现
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Created by GJ on 2017/11/21
import sys
from PyQt5.QtGui import QPainter, QIcon
from PyQt5.QtWidgets import QMainWindow, QApplication, QPushButton, QFileDialog, QSystemTrayIcon, QAction, QMenu, \
QMessageBox
from PyQt5.QtCore import Qt, pyqtSignal
import logging
import os
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
formatter = logging.Formatter('%(levelname)s - %(message)s')
ch.setFormatter(formatter)
logger.addHandler(ch)
class MyWin(QMainWindow):
def __init__(self):
super(MyWin, self).__init__()
self.initUi()
def initUi(self):
# 窗口大小设置为600*500
self.setWindowTitle("截图")
self.resize(100, 100)
self.btn = QPushButton('截图', self)
self.btn.setGeometry(20, 20, 60, 60)
self.btn.clicked.connect(self.click_btn)
self.addSystemTray() # 设置系统托盘
self.setFixedSize(self.width(), self.height());
def addSystemTray(self):
minimizeAction = QAction("Mi&nimize", self, triggered=self.hide)
maximizeAction = QAction("Ma&ximize", self, triggered=self.showMaximized)
restoreAction = QAction("&Restore", self, triggered=self.showNormal)
quitAction = QAction("&Quit", self,triggered=self.close)
self.trayIconMenu = QMenu(self)
self.trayIconMenu.addAction(minimizeAction)
self.trayIconMenu.addAction(maximizeAction)
self.trayIconMenu.addAction(restoreAction)
self.trayIconMenu.addSeparator()
self.trayIconMenu.addAction(quitAction)
self.trayIcon = QSystemTrayIcon(self)
self.trayIcon.setIcon(QIcon("icon.png"))
self.setWindowIcon(QIcon("icon.png"))
self.trayIcon.setContextMenu(self.trayIconMenu)
self.trayIcon.show()
def click_btn(self):
self.showMinimized()
self.screenshot = ScreenShotsWin()
self.screenshot.showFullScreen()
class ScreenShotsWin(QMainWindow):
# 定义一个信号
oksignal = pyqtSignal()
def __init__(self):
super(ScreenShotsWin, self).__init__()
self.initUI()
self.start = (0, 0) # 开始坐标点
self.end = (0, 0) # 结束坐标点
self.cwd = os.getcwd() # 获取当前程序文件位置
def initUI(self):
# self.showFullScreen()
self.setWindowOpacity(0.4)
self.btn_ok = QPushButton('保存', self)
self.oksignal.connect(lambda: self.screenshots(self.start, self.end))
def screenshots(self, start, end):
'''
截图功能
:param start:截图开始点
:param end:截图结束点
:return:
'''
logger.debug('开始截图,%s, %s', start, end)
x = min(start[0], end[0])
y = min(start[1], end[1])
width = abs(end[0] - start[0])
height = abs(end[1] - start[1])
des = QApplication.desktop()
screen = QApplication.primaryScreen()
if screen:
self.setWindowOpacity(0.0)
pix = screen.grabWindow(des.winId(), x, y, width, height)
# fileName = QFileDialog.getSaveFileName(self, '保存图片', '图片', "png文件(*.png)")
fileName = QFileDialog.getSaveFileName(self, '保存图片', '.', ".png")
if fileName[0]:
pix.save(fileName[0] + fileName[1])
else:
QMessageBox.information(self, "提示", "没有保存图片,请重新保存")
self.close()
def paintEvent(self, event):
'''
给出截图的辅助线
:param event:
:return:
'''
logger.debug('开始画图')
x = self.start[0]
y = self.start[1]
w = self.end[0] - x
h = self.end[1] - y
pp = QPainter(self)
pp.drawRect(x, y, w, h)
def mousePressEvent(self, event):
# 点击左键开始选取截图区域
if event.button() == Qt.LeftButton:
self.start = (event.pos().x(), event.pos().y())
logger.debug('开始坐标：%s', self.start)
def mouseReleaseEvent(self, event):
# 鼠标左键释放开始截图操作
if event.button() == Qt.LeftButton:
self.end = (event.pos().x(), event.pos().y())
logger.debug('结束坐标：%s', self.end)
self.oksignal.emit()
logger.debug('信号提交')
# 进行重新绘制
self.update()
def mouseMoveEvent(self, event):
# 鼠标左键按下的同时移动鼠标绘制截图辅助线
if event.buttons() and Qt.LeftButton:
self.end = (event.pos().x(), event.pos().y())
# 进行重新绘制
self.update()
# if __name__ == '__main__':
# app = QApplication(sys.argv)
# dbb = MyWin()
# dbb.show()
# sys.exit(app.exec_())
注释主函数可以在别的函数中调用。
# 初始化MainWindow_JT窗口 MainWindow_JT = QMainWindow() JT_ui = screenshots.MyWin() JT_ui.initUi()
###打开截图窗口
def jietu():
# windows.show()
JT_ui.show()