File size: 2,071 Bytes
c8d30bc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | # ThreatHunter tools package
# 保留 package-level export 契約，但以 proxy 方式延後真正 import，
# 避免單純 `import tools` / `hasattr(tools, "...")` 就觸發 CrewAI 副作用。
from importlib import import_module
from typing import Any
class _LazyToolProxy:
def __init__(self, module_name: str, attr_name: str) -> None:
self._module_name = module_name
self._attr_name = attr_name
self._loaded: Any | None = None
def _load(self) -> Any:
if self._loaded is None:
module = import_module(self._module_name)
self._loaded = getattr(module, self._attr_name)
return self._loaded
def __call__(self, *args, **kwargs):
return self._load()(*args, **kwargs)
def __getattr__(self, name: str) -> Any:
return getattr(self._load(), name)
def __repr__(self) -> str:
return f"<LazyToolProxy {self._module_name}:{self._attr_name}>"
_EXPORTS = {
"search_nvd": ("tools.nvd_tool", "search_nvd"),
"search_osv": ("tools.osv_tool", "search_osv"),
"search_osv_batch": ("tools.osv_tool", "search_osv_batch"),
"search_otx": ("tools.otx_tool", "search_otx"),
"check_cisa_kev": ("tools.kev_tool", "check_cisa_kev"),
"search_exploits": ("tools.exploit_tool", "search_exploits"),
"fetch_epss_score": ("tools.epss_tool", "fetch_epss_score"),
"get_epss_score": ("tools.epss_tool", "get_epss_score"),
"lookup_attck_by_cwe": ("tools.attck_tool", "lookup_attck_by_cwe"),
"lookup_attck_by_description": ("tools.attck_tool", "lookup_attck_by_description"),
"get_attck_for_cve": ("tools.attck_tool", "get_attck_for_cve"),
"read_memory": ("tools.memory_tool", "read_memory"),
"write_memory": ("tools.memory_tool", "write_memory"),
"history_search": ("tools.memory_tool", "history_search"),
}
__all__ = list(_EXPORTS)
for _name, (_module_name, _attr_name) in _EXPORTS.items():
globals()[_name] = _LazyToolProxy(_module_name, _attr_name)
def __dir__() -> list[str]:
return sorted(set(globals()) | set(__all__))
|