Vladless
106 lines
4.3 KiB
Python
106 lines
4.3 KiB
Python
import importlib
|
||
import pkgutil
|
||
|
||
from pathlib import Path
|
||
|
||
from aiogram import Router
|
||
|
||
from logger import logger
|
||
|
||
from .modules_manager import manager
|
||
|
||
|
||
modules_hub = Router(name="modules_hub")
|
||
|
||
|
||
def _is_safe_module_name(name: str) -> bool:
|
||
return bool(name and name.isidentifier() and "." not in name and "/" not in name and "\\" not in name)
|
||
|
||
|
||
def load_modules_from_folder(folder: str = "modules") -> list[Router]:
|
||
routers = []
|
||
base_path = Path(folder)
|
||
|
||
for _finder, name, _ispkg in pkgutil.iter_modules([str(base_path)]):
|
||
name = (name or "").strip()
|
||
if not _is_safe_module_name(name):
|
||
logger.warning(f"[Modules] Пропуск недопустимого имени модуля: {name!r}")
|
||
continue
|
||
if not manager.should_autostart(name):
|
||
logger.info(f"[Modules] Пропуск автозапуска модуля '{name}' (отключён).")
|
||
continue
|
||
|
||
module_path = f"{folder}.{name}.router"
|
||
try:
|
||
mod = importlib.import_module(module_path)
|
||
router = getattr(mod, "router", None)
|
||
if isinstance(router, Router):
|
||
modules_hub.include_router(router)
|
||
manager.adopt(name, router)
|
||
routers.append(router)
|
||
logger.info(f"[Modules] Загружен модуль: {module_path}")
|
||
else:
|
||
logger.warning(f"[Modules] В модуле {module_path} не найден router")
|
||
except Exception as e:
|
||
logger.error(f"[Modules] Ошибка при загрузке {module_path}: {e}")
|
||
return routers
|
||
|
||
|
||
def load_module_webhooks(folder: str = "modules") -> list[dict]:
|
||
webhooks = []
|
||
base_path = Path(folder)
|
||
|
||
for _finder, name, _ispkg in pkgutil.iter_modules([str(base_path)]):
|
||
name = (name or "").strip()
|
||
if not _is_safe_module_name(name):
|
||
continue
|
||
if not manager.should_autostart(name):
|
||
logger.info(f"[Modules] Пропуск вебхуков модуля '{name}' (отключён).")
|
||
continue
|
||
|
||
module_path = f"{folder}.{name}"
|
||
try:
|
||
router_module = importlib.import_module(f"{module_path}.router")
|
||
if hasattr(router_module, "get_webhook_data"):
|
||
webhook_data = router_module.get_webhook_data()
|
||
if isinstance(webhook_data, dict) and "path" in webhook_data and "handler" in webhook_data:
|
||
webhooks.append(webhook_data)
|
||
logger.info(f"[Modules] Найден вебхук в модуле {name}: {webhook_data['path']}")
|
||
except Exception as e:
|
||
logger.error(f"[Modules] Ошибка при загрузке вебхуков из {module_path}: {e}")
|
||
return webhooks
|
||
|
||
|
||
def load_module_fast_flow_handlers(folder: str = "modules") -> dict:
|
||
handlers = {}
|
||
base_path = Path(folder)
|
||
|
||
for _finder, name, _ispkg in pkgutil.iter_modules([str(base_path)]):
|
||
name = (name or "").strip()
|
||
if not _is_safe_module_name(name):
|
||
continue
|
||
if not manager.should_autostart(name):
|
||
logger.info(f"[Modules] Пропуск fast-flow модуля '{name}' (отключён).")
|
||
continue
|
||
|
||
module_path = f"{folder}.{name}"
|
||
try:
|
||
router_module = importlib.import_module(f"{module_path}.router")
|
||
if hasattr(router_module, "get_fast_flow_handler"):
|
||
fast_flow_data = router_module.get_fast_flow_handler()
|
||
if (
|
||
fast_flow_data
|
||
and isinstance(fast_flow_data, dict)
|
||
and "payment_key" in fast_flow_data
|
||
and "handler" in fast_flow_data
|
||
):
|
||
payment_key = fast_flow_data["payment_key"]
|
||
handler = fast_flow_data["handler"]
|
||
handlers[payment_key] = handler
|
||
logger.info(f"[Modules] Найден обработчик быстрого флоу в модуле {name}: {payment_key}")
|
||
elif fast_flow_data is None:
|
||
logger.info(f"[Modules] Быстрое флоу отключено в модуле {name}")
|
||
except Exception as e:
|
||
logger.error(f"[Modules] Ошибка при загрузке быстрого флоу из {module_path}: {e}")
|
||
return handlers
|