mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-31 13:07:56 +08:00
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
"""Chain 兼容门面所需运行时依赖的显式上下文。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from dataclasses import dataclass, field
|
|
from typing import Any, Optional
|
|
|
|
from app.application.chain.data import ChainDataPorts
|
|
from app.application.chain.events import ChainDurableEventWriter
|
|
from app.application.configuration import ChainRuntimeConfig
|
|
from app.runtime.stop import StopState, runtime_stop_state
|
|
|
|
MessageQueueFactory = Callable[[Callable[..., Any]], Any]
|
|
ModuleDispatcherFactory = Callable[..., Any]
|
|
LegacyTransferCommand = Callable[..., Any]
|
|
ChainRuntimeContextProvider = Callable[[], "ChainRuntimeContext"]
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ChainRuntimeContext:
|
|
"""集中声明 Chain 调度、事件、消息和缓存所需的最小运行时对象。"""
|
|
|
|
module_manager: Any
|
|
plugin_manager: Any
|
|
event_manager: Any
|
|
message_oper: Any
|
|
message_helper: Any
|
|
file_cache: Any
|
|
async_file_cache: Any
|
|
message_queue_factory: MessageQueueFactory
|
|
module_dispatcher_factory: ModuleDispatcherFactory
|
|
legacy_transfer_command: Optional[LegacyTransferCommand] = None
|
|
data_ports: Optional[ChainDataPorts] = None
|
|
durable_event_writer: Optional[ChainDurableEventWriter] = None
|
|
configuration: ChainRuntimeConfig = field(
|
|
default_factory=lambda: ChainRuntimeConfig(media_extensions=())
|
|
)
|
|
stop_state: StopState = field(default_factory=lambda: runtime_stop_state)
|
|
|
|
|
|
def _unconfigured_chain_runtime_context() -> ChainRuntimeContext:
|
|
"""拒绝在组合根装配前隐式抓取全局管理器。"""
|
|
raise RuntimeError("Chain 运行上下文尚未由启动组合根配置")
|
|
|
|
|
|
_context_provider: ChainRuntimeContextProvider = _unconfigured_chain_runtime_context
|
|
|
|
|
|
def configure_chain_runtime_context_provider(
|
|
provider: Optional[ChainRuntimeContextProvider],
|
|
) -> None:
|
|
"""由组合根替换 Chain 上下文来源；传入空值恢复未配置状态。"""
|
|
global _context_provider
|
|
_context_provider = provider or _unconfigured_chain_runtime_context
|
|
|
|
|
|
def get_chain_runtime_context() -> ChainRuntimeContext:
|
|
"""返回当前组合根提供的 Chain 运行上下文。"""
|
|
return _context_provider()
|