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 {}
|