123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on 2018年6月8日
|
|
author: Irony
|
|
site: https://pyqt.site , https://github.com/PyQt5
|
|
email: 892768447@qq.com
|
|
file: ProbeWindow
|
|
description: 简单探测窗口和放大截图
|
|
"""
|
|
|
|
import win32gui
|
|
|
|
try:
|
|
from PyQt5.QtCore import Qt, QRect
|
|
from PyQt5.QtGui import QPainter, QPen, QCursor, QColor
|
|
from PyQt5.QtWidgets import QLabel, QWidget, QApplication
|
|
except ImportError:
|
|
from PySide2.QtCore import Qt, QRect
|
|
from PySide2.QtGui import QPainter, QPen, QCursor, QColor
|
|
from PySide2.QtWidgets import QLabel, QWidget, QApplication
|
|
|
|
|
|
class FrameWidget(QWidget):
|
|
# 一个全屏的透明窗口
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(FrameWidget, self).__init__(*args, **kwargs)
|
|
self.setWindowFlags(Qt.Tool | Qt.FramelessWindowHint |
|
|
Qt.WindowStaysOnTopHint)
|
|
self.setAttribute(Qt.WA_TranslucentBackground, True)
|
|
self.showFullScreen() # 全屏
|
|
self._rect = QRect() # 被探测的窗口的矩形位置
|
|
|
|
def setRect(self, x, y, w, h):
|
|
# 更新矩形框
|
|
self._rect.setX(x)
|
|
self._rect.setY(y)
|
|
self._rect.setWidth(w - x)
|
|
self._rect.setHeight(h - y)
|
|
self.update()
|
|
|
|
def paintEvent(self, event):
|
|
super(FrameWidget, self).paintEvent(event)
|
|
if self._rect.isValid(): # 画边框
|
|
painter = QPainter(self)
|
|
painter.setPen(QPen(Qt.red, 4))
|
|
painter.drawRect(self._rect)
|
|
|
|
|
|
class Label(QLabel):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(Label, self).__init__(*args, **kwargs)
|
|
self.ismd = False # 是不是否按下
|
|
self.setAlignment(Qt.AlignCenter)
|
|
self.setText('鼠标按住不放拖动到外面')
|
|
self.resize(240, 240)
|
|
self.frameWidget = FrameWidget() # 边框
|
|
|
|
def closeEvent(self, event):
|
|
self.frameWidget.close()
|
|
super(Label, self).closeEvent(event)
|
|
|
|
def mousePressEvent(self, event):
|
|
super(Label, self).mousePressEvent(event)
|
|
self.ismd = True # 按下
|
|
# 设置鼠标样式为十字
|
|
self.setCursor(Qt.CrossCursor)
|
|
|
|
def mouseReleaseEvent(self, event):
|
|
super(Label, self).mouseReleaseEvent(event)
|
|
self.ismd = False
|
|
self.frameWidget.setRect(0, 0, 0, 0)
|
|
# 设置鼠标样式为普通
|
|
self.setCursor(Qt.ArrowCursor)
|
|
self.clear()
|
|
self.setText('鼠标按住不放拖动到外面')
|
|
|
|
def mouseMoveEvent(self, event):
|
|
super(Label, self).mouseMoveEvent(event)
|
|
# 得到鼠标在屏幕中的位置
|
|
pos = self.mapToGlobal(event.pos())
|
|
hwnd = win32gui.WindowFromPoint((pos.x(), pos.y()))
|
|
self.frameWidget.setRect(*win32gui.GetWindowRect(hwnd))
|
|
# 截图
|
|
screen = QApplication.primaryScreen()
|
|
if screen is not None:
|
|
image = screen.grabWindow(0,
|
|
pos.x() - 60, pos.y() - 60, 120, 120)
|
|
if not image.isNull():
|
|
self.setPixmap(image.scaled(240, 240))
|
|
|
|
def paintEvent(self, event):
|
|
super(Label, self).paintEvent(event)
|
|
# 中正间画十字
|
|
painter = QPainter(self)
|
|
painter.setPen(Qt.red)
|
|
x = int(self.width() / 2)
|
|
y = int(self.height() / 2)
|
|
painter.drawLine(x, 0, x, self.height())
|
|
painter.drawLine(0, y, self.width(), y)
|
|
if self.ismd:
|
|
# 画坐标点
|
|
pos = QCursor.pos()
|
|
ret = win32gui.GetPixel(win32gui.GetWindowDC(
|
|
win32gui.GetDesktopWindow()), pos.x(), pos.y())
|
|
r, g, b = ret & 0xff, (ret >> 8) & 0xff, (ret >> 16) & 0xff
|
|
print(r, g, b)
|
|
painter.setPen(Qt.white)
|
|
painter.drawText(self.rect(), Qt.AlignLeft |
|
|
Qt.AlignBottom, '({}, {})\nRGB: ({}, {}, {})\n{}'.format(
|
|
pos.x(), pos.y(), r, g, b, QColor(r, g, b).name()))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
|
|
app = QApplication(sys.argv)
|
|
app.setQuitOnLastWindowClosed(True)
|
|
w = Label()
|
|
w.show()
|
|
sys.exit(app.exec_())
|