mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-05 05:16:03 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
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()]
|