43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
|
# ===============================================
|
|||
|
|
# =============== OCR 插件接口管理 ===============
|
|||
|
|
# ===============================================
|
|||
|
|
|
|||
|
|
from umi_log import logger
|
|||
|
|
|
|||
|
|
ApiDict = {}
|
|||
|
|
AllDict = {}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# TODO: 静态插件
|
|||
|
|
# 由插件控制器调用，初始化OCR插件的接口。传入动态插件
|
|||
|
|
def initOcrPlugins(plugins):
|
|||
|
|
global ApiDict, AllDict
|
|||
|
|
for p in plugins:
|
|||
|
|
ApiDict[p] = plugins[p]["api_class"]
|
|||
|
|
AllDict[p] = plugins[p]
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 生成一个ocr api实例，成功返回对象，失败返回 [Error] 开头的字符串
|
|||
|
|
def getApiOcr(apiKey, argd):
|
|||
|
|
# 检测argd，恢复int类型
|
|||
|
|
for k in argd:
|
|||
|
|
n = argd[k]
|
|||
|
|
if isinstance(n, float):
|
|||
|
|
rounded = round(n)
|
|||
|
|
if abs(n - rounded) <= 1e-7:
|
|||
|
|
argd[k] = rounded
|
|||
|
|
if apiKey in ApiDict:
|
|||
|
|
try:
|
|||
|
|
return ApiDict[apiKey](argd) # 实例化后返回
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"生成api实例{apiKey}失败。", exc_info=True, stack_info=True)
|
|||
|
|
return f"[Error] Failed to generate API instance {apiKey}: {e}"
|
|||
|
|
return f'[Error] "{apiKey}" not in ApiDict.'
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 返回一个API的局部配置字典
|
|||
|
|
def getLocalOptions(apiKey):
|
|||
|
|
if apiKey in AllDict:
|
|||
|
|
return AllDict[apiKey]["local_options"]
|
|||
|
|
return {}
|