nanobot/nanobot/utils/__init__.py

"""Utility functions for nanobot."""

from __future__ import annotations

import sys
from importlib import import_module
from types import ModuleType

from nanobot.utils.helpers import ensure_dir
from nanobot.utils.path import abbreviate_path

__all__ = ["ensure_dir", "abbreviate_path"]


class _LazyModuleAlias(ModuleType):
    def __init__(self, name: str, target: str) -> None:
        super().__init__(name)
        self.__dict__["_target"] = target

    def _load(self) -> ModuleType:
        module = import_module(self.__dict__["_target"])
        sys.modules[self.__name__] = module
        return module

    def __getattr__(self, name: str) -> object:
        return getattr(self._load(), name)

    def __dir__(self) -> list[str]:
        return sorted(set(super().__dir__()) | set(dir(self._load())))


_LEGACY_MODULE_ALIASES = {
    "webui_thread_disk": "nanobot.webui.thread_disk",
    "webui_transcript": "nanobot.webui.transcript",
    "webui_turn_helpers": "nanobot.session.webui_turns",
}

for _legacy_name, _target_name in _LEGACY_MODULE_ALIASES.items():
    sys.modules.setdefault(
        f"{__name__}.{_legacy_name}",
        _LazyModuleAlias(f"{__name__}.{_legacy_name}", _target_name),
    )