open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/dptech-corp/Uni-Lab-OS.git synced 2026-02-07 07:25:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

成功动态导入的不再需要使用静态导入

Browse Source
This commit is contained in:
Xuwznln
2025-06-29 11:52:59 +08:00
parent 498b3cad6a
commit 68871358c2
1 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
unilabos/utils/import_manager.py
View File

@@ -205,6 +205,7 @@ class ImportManager:
# 尝试动态导入 # 尝试动态导入
dynamic_info = None dynamic_info = None
static_info = None
if use_dynamic: if use_dynamic:
try: try:
dynamic_info = self._get_dynamic_class_info(module_path) dynamic_info = self._get_dynamic_class_info(module_path)
@@ -216,10 +217,9 @@ class ImportManager:
f"{module_path} 失败（将使用静态分析，" f"{module_path} 失败（将使用静态分析，"
f"建议修复导入错误，以实现更好的注册表识别效果！）: {e}" f"建议修复导入错误，以实现更好的注册表识别效果！）: {e}"
) )
use_dynamic = False use_dynamic = False
if not use_dynamic: if not use_dynamic:
# 尝试静态分析 # 尝试静态分析
static_info = None
try: try:
static_info = self._get_static_class_info(module_path) static_info = self._get_static_class_info(module_path)
result["static_analysis_success"] = True result["static_analysis_success"] = True
@@ -408,18 +408,32 @@ class ImportManager:
def _get_return_type_from_method(self, method) -> str: def _get_return_type_from_method(self, method) -> str:
"""从方法中获取返回类型""" """从方法中获取返回类型"""
if hasattr(method, "__annotations__") and "return" in method.__annotations__:
return self._get_type_string(method.__annotations__["return"])
signature = inspect.signature(method) signature = inspect.signature(method)
return self._get_type_string(signature.return_annotation) return self._get_type_string(signature.return_annotation)
def _get_type_string(self, annotation) -> str: def _get_type_string(self, annotation) -> str:
"""将类型注解转换为字符串""" """将类型注解转换为Class Library中可搜索的类名"""
if annotation == inspect.Parameter.empty: if annotation == inspect.Parameter.empty:
return "Any" # 如果没有注解，返回Any return "Any" # 如果没有注解，返回Any
if annotation is None: if annotation is None:
return "None" # 明确的None类型 return "None" # 明确的None类型
if hasattr(annotation, "__origin__"):
# 处理typing模块的类型
origin = annotation.__origin__
if origin in (list, set, tuple):
if hasattr(annotation, "__args__") and annotation.__args__:
if len(annotation.__args__):
arg0 = annotation.__args__[0]
if isinstance(arg0, int):
return "Int64MultiArray"
elif isinstance(arg0, float):
return "Float64MultiArray"
return "list"
elif origin is dict:
return "dict"
elif origin is Optional:
return "Unknown"
return f"Unknown"
annotation_str = str(annotation) annotation_str = str(annotation)
# 处理typing模块的复杂类型 # 处理typing模块的复杂类型
if "typing." in annotation_str: if "typing." in annotation_str: