截图翻译-python实现
因为看到TextShot和translators两个python的github项目，想着可以做一个截图翻译软件，于是合了一下 pyqt5 截屏,图片识别文字 的代码
准备工作
需要先安装 Tesseract OCR，并且加入Path路径
Tesseract OCR下载地址
python需要的库
Pillow==9.1.0
PyQt5==5.15.6
PyQt5-sip==12.10.1
PyQt5-stubs==5.15.6.0
py-notifier==0.3.2
pyperclip==1.8.2
pytesseract==0.3.9
win10toast==0.9; platform_system == "Windows"
translators==5.1.1
实现代码：
import sys
import pyperclip
import pytesseract
from PIL import Image
import translators as ts
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtCore import *
try:
from pynotifier import Notification
except ImportError:
pass
class QTextEditDemo(QWidget):
def __init__(self):
super(QTextEditDemo,self).__init__()
self.initUI()
def initUI(self):
self.setWindowTitle("截图翻译")
self.resize(300,320)
self.move(0,200)
self.textEdit = QTextEdit()
self.textEdit1 = QTextEdit()
self.buttonText = QPushButton("截图")
layout = QVBoxLayout()
layout.addWidget(self.textEdit)
layout.addWidget(self.textEdit1)
layout.addWidget(self.buttonText)
self.setLayout(layout)
self.buttonText.clicked.connect(self.onClick_ButtonText)
# self.textEdit.setPlainText("Hello World")
# 最小化
# self.buttonText1.clicked.connect(self.showMinimized)
def onClick_ButtonText(self):
self.hide()
win.show()
class WScreenShot(QWidget):
def __init__(self, parent=None):
super(WScreenShot, self).__init__(parent)
self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint)
self.setStyleSheet('''background-color:black; ''')
self.setWindowOpacity(0.4)
# 1 透明度的有效范围从1.0 完全不透明到0.0 完全透明
# 2 默认情况下是不透明状态，属性值为1.0
desktopRect = QDesktopWidget().screenGeometry()
# 获取屏幕的信息
# screenGeometry（）函数提供有关可用屏幕几何的信息
self.setGeometry(desktopRect)
# setGeometry (9,9, 50, 25)
# 从屏幕上（9，9）位置开始（即为最左上角的点），显示一个50*25的界面（宽50，高25）
# setGeometry之后一定要调用show函数，否则可能看不到控件存在
self.setCursor(Qt.ArrowCursor) # 设置鼠标形状
self.blackMask = QBitmap(desktopRect.size())
self.blackMask.fill(Qt.black)
self.mask = self.blackMask.copy()
self.isDrawing = False
self.startPoint = QPoint()
self.endPoint = QPoint()
self.setMouseTracking(False)
self.startX, self.startY = 0, 0 # the point where you start
self.endX, self.endY = 0, 0 # the point where you end
def paintEvent(self, event):
if self.isDrawing:
self.mask = self.blackMask.copy()
pp = QPainter(self.mask)
pen = QPen()
pen.setStyle(Qt.NoPen)
pp.setPen(pen)
brush = QBrush(Qt.white)
pp.setBrush(brush)
pp.drawRect(QRect(self.startPoint, self.endPoint)) # 画图
self.setMask(QBitmap(self.mask))
def mousePressEvent(self, event):
if event.button() == Qt.LeftButton:
self.startPoint = event.pos()
self.startX, self.startY = event.x(), event.y()
# print("mousepress: ", self.startPoint)
self.endPoint = self.startPoint
self.isDrawing = True
def mouseMoveEvent(self, event):
if self.isDrawing:
self.endPoint = event.pos()
self.endX, self.endY = event.x(), event.y()
self.update()
def mouseReleaseEvent(self, event):
result_str = ""
fy_str = ""
if event.button() == Qt.LeftButton:
self.endPoint = event.pos()
# print("mouse release: ", self.endPoint)
# PySide2
# screenshot = QPixmap.grabWindow(QApplication.desktop().winId())
# PyQt5
# screenshot = QApplication.primaryScreen().grabWindow(0)
# 通用
screenshot = QApplication.primaryScreen().grabWindow(QApplication.desktop().winId())
# rect = QRect(self.startPoint, self.endPoint)
# rect = QRect(self.startX, self.startY + 28, self.endX-self.startX, self.endY-self.startY)
rect = QRect(
min(self.startX, self.endX),
min(self.startY, self.endY),
abs(self.startX - self.endX),
abs(self.startY - self.endY)
)
# print("last rect:", rect)
outputRegion = screenshot.copy(rect)
try:
outputRegion.save('111.jpg', format='JPG', quality=100)
except Exception as e:
print("保存截图失败！！！")
notify("保存截图失败！！！")
# print("ok")
self.close()
try:
pil_img = Image.open('111.jpg')
except Exception as e:
print("读取截图失败！！！")
notify("读取截图失败！！！")
try:
result = pytesseract.image_to_string(
pil_img, timeout=5, lang=(sys.argv[1] if len(sys.argv) > 1 else None)
).strip()
except RuntimeError as error:
print(f"ERROR: An error occurred when trying to process the image: {error}")
notify(f"识别失败！！！")
return
if result:
result_str = result
# 内容复制到剪贴板
pyperclip.copy(result)
print(f'INFO: Copied "{result}" to the clipboard')
# notify(f'Copied "{result}" to the clipboard')
else:
print(f"INFO: Unable to read text from image, did not copy")
notify(f"Unable to read text from image, did not copy")
main_demo.textEdit.setPlainText(result_str)
try:
fy_str = ts.google(result_str,to_language='zh')
except Exception as e:
print("翻译出错！！！")
notify("翻译出错！！！")
main_demo.textEdit1.setPlainText(fy_str)
main_demo.show()
def notify(msg):
try:
Notification(title="TextShot", description=msg).send()
except (SystemError, NameError):
trayicon = QSystemTrayIcon(
QIcon(
QPixmap.fromImage(QImage(1, 1, QImage.Format_Mono))
)
)
trayicon.show()
trayicon.showMessage("TextShot", msg, QSystemTrayIcon.NoIcon)
trayicon.hide()
if __name__ == "__main__":
app = QApplication(sys.argv)
main_demo = QTextEditDemo()
main_demo.show()
try:
pytesseract.get_tesseract_version()
except EnvironmentError:
print(
"ERROR: Tesseract is either not installed or cannot be reached.\n"
"Have you installed it and added the install directory to your system path?"
)
sys.exit()
win = WScreenShot()
sys.exit(app.exec_())
最后可以用 pyinstaller 打包一下程序，然后到dist目录中项目文件夹下找到exe执行程序，然后把这个执行程序创建桌面快捷方式就可以了
# -w是无显示终端
pyinstaller xxx.py -w
下载体验：
资源下载
