105 lines
4.1 KiB
Python
105 lines
4.1 KiB
Python
|
# ========================================
|
|||
|
|
# =============== 截图控制 ===============
|
|||
|
|
# ========================================
|
|||
|
|
|
|||
|
|
from ..image_controller.image_provider import PixmapProvider # 图片提供器
|
|||
|
|
from ..utils.file_finder import findFiles
|
|||
|
|
|
|||
|
|
import time
|
|||
|
|
from PySide2.QtGui import QGuiApplication, QClipboard, QImage, QPixmap # 截图 剪贴板
|
|||
|
|
|
|||
|
|
Clipboard = QClipboard() # 剪贴板
|
|||
|
|
|
|||
|
|
|
|||
|
|
class _ScreenshotControllerClass:
|
|||
|
|
|
|||
|
|
def getScreenshot(self, wait=0):
|
|||
|
|
"""
|
|||
|
|
延时wait秒后，获取所有屏幕的截图。返回列表(不为空)，每项为：\n
|
|||
|
|
{
|
|||
|
|
"imgID": 图片ID 或 报错信息 "[Error]开头" ,
|
|||
|
|
"screenName": 显示器名称 ,
|
|||
|
|
"width": 截图宽度 ,
|
|||
|
|
"height": 截图高度 ,
|
|||
|
|
}
|
|||
|
|
"""
|
|||
|
|
if wait > 0:
|
|||
|
|
time.sleep(wait)
|
|||
|
|
try:
|
|||
|
|
grabList = []
|
|||
|
|
screensList = QGuiApplication.screens()
|
|||
|
|
for screen in screensList:
|
|||
|
|
name = screen.name()
|
|||
|
|
# 获取截图
|
|||
|
|
pixmap = screen.grabWindow(0)
|
|||
|
|
width = pixmap.width()
|
|||
|
|
height = pixmap.height()
|
|||
|
|
# 检查截图失败
|
|||
|
|
if width <= 0 or height <= 0:
|
|||
|
|
imgID = f"[Error] width={width}, height={height}"
|
|||
|
|
# 检查有效，存入提供器，获取imgID
|
|||
|
|
else:
|
|||
|
|
imgID = PixmapProvider.addPixmap(pixmap)
|
|||
|
|
grabList.append(
|
|||
|
|
{
|
|||
|
|
"imgID": imgID,
|
|||
|
|
"screenName": name,
|
|||
|
|
"width": width,
|
|||
|
|
"height": height,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
if not grabList: # 获取到的截图列表为空
|
|||
|
|
return [{"imgID": f"[Error] grabList is empty."}]
|
|||
|
|
return grabList
|
|||
|
|
except Exception as e:
|
|||
|
|
return [{"imgID": f"[Error] Screenshot: {e}"}]
|
|||
|
|
|
|||
|
|
# 对一张图片做裁切。传入原图imgID和裁切参数，返回裁切后的imgID或[Error]
|
|||
|
|
def getClipImgID(self, imgID, x, y, w, h):
|
|||
|
|
try:
|
|||
|
|
pixmap = PixmapProvider.getPixmap(imgID)
|
|||
|
|
if not pixmap:
|
|||
|
|
return f'[Error] Screenshot: Key "{imgID}" does not exist in the PixmapProvider dict.'
|
|||
|
|
if x < 0 or y < 0 or w <= 0 or h <= 0:
|
|||
|
|
return f"[Error] Screenshot: x/y/w/h value error. {x}/{y}/{w}/{h}"
|
|||
|
|
pixmap = pixmap.copy(x, y, w, h) # 进行裁切
|
|||
|
|
clipID = PixmapProvider.addPixmap(pixmap) # 存入提供器，获取imgID
|
|||
|
|
return clipID
|
|||
|
|
except Exception as e:
|
|||
|
|
return f"[Error] Screenshot: {e}"
|
|||
|
|
|
|||
|
|
# 获取当前剪贴板的内容
|
|||
|
|
# type: imgID paths text
|
|||
|
|
def getPaste(self):
|
|||
|
|
# 获取剪贴板数据
|
|||
|
|
mimeData = Clipboard.mimeData()
|
|||
|
|
res = {"type": ""} # 结果字典
|
|||
|
|
# 检查剪贴板的内容，若是图片，则提取它并扔给OCR
|
|||
|
|
if mimeData.hasImage():
|
|||
|
|
image = Clipboard.image()
|
|||
|
|
pixmap = QPixmap.fromImage(image)
|
|||
|
|
pasteID = PixmapProvider.addPixmap(pixmap) # 存入提供器
|
|||
|
|
res = {"type": "imgID", "imgID": pasteID}
|
|||
|
|
# 若为URL
|
|||
|
|
elif mimeData.hasUrls():
|
|||
|
|
urlList = mimeData.urls()
|
|||
|
|
paths = []
|
|||
|
|
for url in urlList: # 遍历URL列表，提取其中的文件
|
|||
|
|
if url.isLocalFile():
|
|||
|
|
p = url.toLocalFile()
|
|||
|
|
paths.append(p)
|
|||
|
|
paths = findFiles(paths, "image", False) # 过滤，保留图片的路径
|
|||
|
|
if len(paths) == 0: # 没有有效图片
|
|||
|
|
res = {"type": "error", "error": "[Warning] No image in clipboard."}
|
|||
|
|
else: # 将有效图片地址传入OCR，返回地址列表
|
|||
|
|
res = {"type": "paths", "paths": paths}
|
|||
|
|
elif mimeData.hasText():
|
|||
|
|
text = mimeData.text()
|
|||
|
|
res = {"type": "text", "text": text}
|
|||
|
|
else:
|
|||
|
|
res = {"type": "error", "error": "[Warning] Unknow mimeData in clipboard."}
|
|||
|
|
return res # 返回结果字典
|
|||
|
|
|
|||
|
|
|
|||
|
|
ScreenshotController = _ScreenshotControllerClass()
|