mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-19 19:46:55 +08:00
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
import inspect
|
|
|
|
from app.core.event import eventmanager, Event
|
|
from app.log import logger
|
|
from app.schemas.types import EventType
|
|
|
|
|
|
class ConfigReloadMixin:
|
|
"""配置重载混入类
|
|
|
|
继承此 Mixin 类的类，会在配置变更时自动调用 on_config_changed 方法。
|
|
在类中定义 CONFIG_WATCH 集合，指定需要监听的配置项
|
|
重写 on_config_changed 方法实现具体的重载逻辑
|
|
可选地重写 get_reload_name 方法提供模块名称（用于日志显示）
|
|
"""
|
|
|
|
def __init_subclass__(cls, **kwargs):
|
|
super().__init_subclass__(**kwargs)
|
|
|
|
config_watch = getattr(cls, 'CONFIG_WATCH', None)
|
|
if not config_watch:
|
|
return
|
|
|
|
# 检查 on_config_changed 方法是否为异步
|
|
is_async = inspect.iscoroutinefunction(cls.on_config_changed)
|
|
|
|
method_name = 'handle_config_changed'
|
|
|
|
# 创建事件处理函数
|
|
def create_handler(is_async):
|
|
if is_async:
|
|
async def wrapper(self: ConfigReloadMixin, event: Event):
|
|
if not event:
|
|
return
|
|
changed_keys = getattr(event.event_data, "key", set()) & config_watch
|
|
if not changed_keys:
|
|
return
|
|
logger.info(f"配置 {', '.join(changed_keys)} 变更，重载 {self.get_reload_name()}...")
|
|
await self.on_config_changed()
|
|
else:
|
|
def wrapper(self: ConfigReloadMixin, event: Event):
|
|
if not event:
|
|
return
|
|
changed_keys = getattr(event.event_data, "key", set()) & config_watch
|
|
if not changed_keys:
|
|
return
|
|
logger.info(f"配置 {', '.join(changed_keys)} 变更，重载 {self.get_reload_name()}...")
|
|
self.on_config_changed()
|
|
|
|
return wrapper
|
|
|
|
# 创建并设置处理函数
|
|
handler = create_handler(is_async)
|
|
handler.__module__ = cls.__module__
|
|
handler.__qualname__ = f'{cls.__name__}.{method_name}'
|
|
setattr(cls, method_name, handler)
|
|
# 添加为事件处理器
|
|
eventmanager.add_event_listener(EventType.ConfigChanged, handler)
|
|
|
|
def on_config_changed(self):
|
|
"""子类重写此方法实现具体重载逻辑"""
|
|
pass
|
|
|
|
def get_reload_name(self):
|
|
"""功能/模块名称"""
|
|
return self.__class__.__name__
|