msn_quick.py

# 单次快捷任务 任务处理器

from utils.config import Config
from ocr.engine import MsnFlag
from ocr.msn import Msn
from ocr.output_panel import OutputPanel  # 输出器
from ui.win_notify import Notify  # 通知弹窗

import tkinter as tk
import time
from pyperclip import copy as pyperclipCopy

from utils.logger import GetLog
Log = GetLog()


class MsnQuick(Msn):

    # __init__ 在主线程内初始化，其余方法在子线程内被调用
    def __init__(self):
        self.isNeedClear = Config.get('isNeedClear')  # 是否需要清空输出面板
        self.outputPanel = OutputPanel()  # 输出到面板
        # 初始化文块处理器
        self.procList = []
        tbpuClass = Config.get('tbpu').get(
            Config.get('tbpuName'), None)
        if tbpuClass:
            self.procList.append(tbpuClass())
        Log.info(f'快捷文本处理器初始化完毕！')

    def onStart(self, num):
        # 重置进度提示
        Config.main.setRunning(MsnFlag.running)  # 先设running，再设进度条动画来覆盖
        progressbar = Config.main.progressbar  # 进度条组件
        progressbar["maximum"] = 50  # 重置进度条长度，值越小加载动画越快
        progressbar['mode'] = 'indeterminate'  # 进度条为来回动模式
        progressbar.start()  # 进度条开始加载动画
        Config.set('tipsTop2', '快捷识图中……')
        # 输出初始信息
        if self.isNeedClear:  # 清空面板
            Config.main.panelClear()
        else:  # 无需清空面板，则输出日志信息
            self.outputPanel.print('\n\n')
            startStr = f"快捷识图 {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))}"
            self.outputPanel.print(startStr, highlight='blue')
            self.outputPanel.print('\n\n')

    def onGet(self, numData, ocrData):
        # ==================== 分析文块 ====================
        if ocrData['code'] == 100:  # 成功
            tbList = ocrData['data']  # 获取文块
            # 将文块组导入每一个文块处理器，获取输出文块组
            for proc in self.procList:
                tbList, s = proc.run(tbList, None)
            tbLen = len(tbList)
            if tbLen == 0:
                self.outputPanel.print('不存在有效文字\n')
                Notify('未发现文字', '')
                return
            tbTexts = [tb['text'] for tb in tbList]  # 提取文字
            tbStr = '\n'.join(tbTexts)
            self.outputPanel.print(tbStr)  # 输出到面板
            if Config.get('isNeedCopy'):  # 需要复制
                pyperclipCopy(tbStr)  # 复制到剪贴板
                Notify('已复制文字', tbStr)
            else:
                Notify('识别完成', tbStr)
            # 计算置信度
            tbScore = sum([tb['score'] for tb in tbList])
            Config.set(
                'tipsTop1', f'耗时 {round(numData["time"], 2)}s  置信 {str(tbScore/tbLen)[:4]}')
        elif ocrData['code'] == 101:  # 无文字
            Config.set('tipsTop1', f'耗时：{round(numData["time"], 2)}s  无文字')
            self.outputPanel.print('未发现文字\n')
            Notify('未发现文字', '')
        else:  # 识别失败
            Config.set('tipsTop1', f'耗时：{round(numData["time"], 2)}s  识别失败')
            self.outputPanel.print(
                f'识别失败，错误码：{ocrData["code"]}\n错误信息：{str(ocrData["data"])}\n')
            Notify(
                '识别失败', f'错误码：{ocrData["code"]}\n错误信息：{str(ocrData["data"])}\n')

    def onStop(self, num):
        Config.main.setRunning(MsnFlag.none)