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
|