LangBot/src/langbot/pkg/utils/importutil.py

import importlib
import importlib.resources
import os
import typing


def import_modules_in_pkg(pkg: typing.Any) -> None:
    """
    导入一个包内的所有模块
    Args:
        pkg: 要导入的包对象
    """
    pkg_path = os.path.dirname(pkg.__file__)
    import_dir(pkg_path)


def import_modules_in_pkgs(pkgs: typing.List) -> None:
    for pkg in pkgs:
        import_modules_in_pkg(pkg)


def import_dot_style_dir(dot_sep_path: str):
    sec = dot_sep_path.split('.')

    return import_dir(os.path.join(*sec))


def import_dir(path: str, path_prefix: str = 'langbot.'):
    for file in os.listdir(path):
        if file.endswith('.py') and file != '__init__.py':
            full_path = os.path.join(path, file)
            rel_path = full_path.replace(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), '')
            rel_path = rel_path[1:]
            rel_path = rel_path.replace('/', '.')[:-3]
            rel_path = rel_path.replace('\\', '.')
            importlib.import_module(f'{path_prefix}{rel_path}')


def read_resource_file(resource_path: str) -> str:
    with importlib.resources.files('langbot').joinpath(resource_path).open('r', encoding='utf-8') as f:
        return f.read()


def read_resource_file_bytes(resource_path: str) -> bytes:
    return importlib.resources.files('langbot').joinpath(resource_path).read_bytes()


def list_resource_files(resource_path: str) -> list[str]:
    return [f.name for f in importlib.resources.files('langbot').joinpath(resource_path).iterdir()]