33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
|
# OCR输出器的基类。按指定的格式，将传入的文本输出到指定地方。
|
|||
|
|
|
|||
|
|
from .tools import getDataText
|
|||
|
|
from ...platform import Platform
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Output:
|
|||
|
|
def __init__(self, argd):
|
|||
|
|
self.dir = argd["outputDir"] # 输出路径（文件夹）
|
|||
|
|
self.fileName = argd["outputFileName"] # 文件名
|
|||
|
|
self.outputPath = f"{self.dir}/{self.fileName}.txt" # 输出路径
|
|||
|
|
self.ignoreBlank = argd["ignoreBlank"] # 忽略空白文件
|
|||
|
|
|
|||
|
|
def print(self, res): # 输出图片信息
|
|||
|
|
if not res["code"] == 100 and self.ignoreBlank:
|
|||
|
|
return # 忽略空白图片
|
|||
|
|
textOut = f"图片路径：{res['path']}\n代码：{res['code']}\n"
|
|||
|
|
if res["code"] == 100:
|
|||
|
|
textOut += getDataText(res["data"]) # 获取拼接结果
|
|||
|
|
elif res["code"] == 101:
|
|||
|
|
textOut += "无文字"
|
|||
|
|
else:
|
|||
|
|
textOut += f"错误原因：{res['data']}"
|
|||
|
|
print(textOut)
|
|||
|
|
|
|||
|
|
def openOutputFile(self): # 打开输出文件
|
|||
|
|
if self.outputPath and os.path.exists(self.outputPath):
|
|||
|
|
Platform.startfile(self.outputPath)
|
|||
|
|
|
|||
|
|
def onEnd(self): # 结束输出。
|
|||
|
|
pass
|