| |
| |
| |
|
|
| 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__)) |
|
|
