mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-06 16:07:01 +08:00
feat(events): expose typed payload snapshots
This commit is contained in:
@@ -37,6 +37,21 @@ class EventErrorBehavior(StrEnum):
|
||||
NOTIFY = "notify"
|
||||
|
||||
|
||||
class EventPayloadMode(StrEnum):
|
||||
"""描述事件 payload 是运行时对象还是稳定快照。"""
|
||||
|
||||
RUNTIME = "runtime"
|
||||
SNAPSHOT = "snapshot"
|
||||
EXTENSION = "extension"
|
||||
|
||||
|
||||
class EventValidationMode(StrEnum):
|
||||
"""描述契约校验失败后的兼容策略。"""
|
||||
|
||||
DIAGNOSTIC = "diagnostic"
|
||||
STRICT = "strict"
|
||||
|
||||
|
||||
@dataclass(frozen=True, slots=True)
|
||||
class EventContract:
|
||||
"""冻结一个事件的 payload 与运行语义。"""
|
||||
@@ -49,6 +64,11 @@ class EventContract:
|
||||
delivery: EventDelivery
|
||||
error_behavior: EventErrorBehavior
|
||||
ordering: str
|
||||
input_model: type[BaseModel] | None = None
|
||||
output_model: type[BaseModel] | None = None
|
||||
schema_version: int = 1
|
||||
payload_mode: EventPayloadMode = EventPayloadMode.RUNTIME
|
||||
validation_mode: EventValidationMode = EventValidationMode.DIAGNOSTIC
|
||||
sensitive_fields: tuple[str, ...] = ()
|
||||
legacy_reason: str | None = None
|
||||
|
||||
@@ -66,13 +86,13 @@ _PAYLOAD_MODELS: dict[EventType | ChainEventType, type[BaseModel]] = {
|
||||
EventType.SubscribeAdded: event_schemas.SubscribeAddedEventData,
|
||||
EventType.SubscribeDeleted: event_schemas.SubscribeDeletedEventData,
|
||||
EventType.SubscribeModified: event_schemas.SubscribeModifiedEventData,
|
||||
EventType.DownloadAdded: event_schemas.DownloadAddedEventData,
|
||||
EventType.TransferComplete: event_schemas.TransferResultEventData,
|
||||
EventType.TransferFailed: event_schemas.TransferResultEventData,
|
||||
EventType.SubtitleTransferComplete: event_schemas.TransferResultEventData,
|
||||
EventType.SubtitleTransferFailed: event_schemas.TransferResultEventData,
|
||||
EventType.AudioTransferComplete: event_schemas.TransferResultEventData,
|
||||
EventType.AudioTransferFailed: event_schemas.TransferResultEventData,
|
||||
EventType.DownloadAdded: event_schemas.DownloadAddedContractData,
|
||||
EventType.TransferComplete: event_schemas.TransferResultContractData,
|
||||
EventType.TransferFailed: event_schemas.TransferResultContractData,
|
||||
EventType.SubtitleTransferComplete: event_schemas.TransferResultContractData,
|
||||
EventType.SubtitleTransferFailed: event_schemas.TransferResultContractData,
|
||||
EventType.AudioTransferComplete: event_schemas.TransferResultContractData,
|
||||
EventType.AudioTransferFailed: event_schemas.TransferResultContractData,
|
||||
EventType.HistoryDeleted: event_schemas.HistoryDeletedEventData,
|
||||
EventType.DownloadFileDeleted: event_schemas.DownloadFileDeletedEventData,
|
||||
EventType.DownloadDeleted: event_schemas.DownloadDeletedEventData,
|
||||
@@ -81,7 +101,7 @@ _PAYLOAD_MODELS: dict[EventType | ChainEventType, type[BaseModel]] = {
|
||||
EventType.NoticeMessage: event_schemas.NoticeMessageEventData,
|
||||
EventType.SubscribeComplete: event_schemas.SubscribeCompleteEventData,
|
||||
EventType.SystemError: event_schemas.SystemErrorEventData,
|
||||
EventType.MetadataScrape: event_schemas.MetadataScrapeEventData,
|
||||
EventType.MetadataScrape: event_schemas.MetadataScrapeContractData,
|
||||
EventType.ModuleReload: event_schemas.EmptyEventData,
|
||||
EventType.MessageAction: event_schemas.MessageActionEventData,
|
||||
EventType.WorkflowExecute: event_schemas.WorkflowExecuteEventData,
|
||||
@@ -93,15 +113,15 @@ _PAYLOAD_MODELS: dict[EventType | ChainEventType, type[BaseModel]] = {
|
||||
ChainEventType.TransferRenameBuild: event_schemas.TransferRenameBuildEventData,
|
||||
ChainEventType.TransferIntercept: event_schemas.TransferInterceptEventData,
|
||||
ChainEventType.TransferOverwriteCheck: event_schemas.TransferOverwriteCheckEventData,
|
||||
ChainEventType.ResourceSelection: event_schemas.ResourceSelectionEventData,
|
||||
ChainEventType.ResourceDownload: event_schemas.ResourceDownloadEventData,
|
||||
ChainEventType.ResourceSelection: event_schemas.ResourceSelectionContractData,
|
||||
ChainEventType.ResourceDownload: event_schemas.ResourceDownloadContractData,
|
||||
ChainEventType.DiscoverSource: event_schemas.DiscoverSourceEventData,
|
||||
ChainEventType.MediaRecognizeConvert: event_schemas.MediaRecognizeConvertEventData,
|
||||
ChainEventType.RecommendSource: event_schemas.RecommendSourceEventData,
|
||||
ChainEventType.StorageOperSelection: event_schemas.StorageOperSelectionEventData,
|
||||
ChainEventType.AgentLLMProvider: event_schemas.AgentLLMProviderEventData,
|
||||
ChainEventType.SubscribeEpisodesRefresh: event_schemas.SubscribeEpisodesRefreshEventData,
|
||||
ChainEventType.SubscribeCompletionCheck: event_schemas.SubscribeCompletionCheckEventData,
|
||||
ChainEventType.SubscribeCompletionCheck: event_schemas.SubscribeCompletionCheckContractData,
|
||||
ChainEventType.NameRecognize: event_schemas.NameRecognizeEventData,
|
||||
ChainEventType.MusicNameRecognize: event_schemas.MusicNameRecognizeEventData,
|
||||
ChainEventType.MediaRecognize: event_schemas.MediaRecognizeEventData,
|
||||
@@ -109,6 +129,32 @@ _PAYLOAD_MODELS: dict[EventType | ChainEventType, type[BaseModel]] = {
|
||||
ChainEventType.WorkflowExecution: ActionContext,
|
||||
}
|
||||
|
||||
_SNAPSHOT_EVENTS = {
|
||||
EventType.DownloadAdded,
|
||||
EventType.TransferComplete,
|
||||
EventType.TransferFailed,
|
||||
EventType.SubtitleTransferComplete,
|
||||
EventType.SubtitleTransferFailed,
|
||||
EventType.AudioTransferComplete,
|
||||
EventType.AudioTransferFailed,
|
||||
EventType.MetadataScrape,
|
||||
ChainEventType.ResourceSelection,
|
||||
ChainEventType.ResourceDownload,
|
||||
ChainEventType.SubscribeCompletionCheck,
|
||||
}
|
||||
|
||||
_INPUT_MODELS = {
|
||||
ChainEventType.ResourceSelection: event_schemas.ResourceSelectionInputContractData,
|
||||
ChainEventType.ResourceDownload: event_schemas.ResourceDownloadInputContractData,
|
||||
ChainEventType.SubscribeCompletionCheck: event_schemas.SubscribeCompletionCheckInputContractData,
|
||||
}
|
||||
|
||||
_OUTPUT_MODELS = {
|
||||
ChainEventType.ResourceSelection: event_schemas.ResourceSelectionOutputContractData,
|
||||
ChainEventType.ResourceDownload: event_schemas.ResourceDownloadOutputContractData,
|
||||
ChainEventType.SubscribeCompletionCheck: event_schemas.SubscribeCompletionCheckOutputContractData,
|
||||
}
|
||||
|
||||
_DURABLE_REQUIRED = {
|
||||
EventType.SubscribeAdded,
|
||||
EventType.SubscribeModified,
|
||||
@@ -150,6 +196,14 @@ def _build_contract(event_type: EventType | ChainEventType) -> EventContract:
|
||||
EventErrorBehavior.STOP_CHAIN if is_chain else EventErrorBehavior.NOTIFY
|
||||
),
|
||||
ordering="priority_serial" if is_chain else "priority_queue",
|
||||
input_model=_INPUT_MODELS.get(event_type, payload_model),
|
||||
output_model=_OUTPUT_MODELS.get(event_type),
|
||||
payload_mode=(
|
||||
EventPayloadMode.SNAPSHOT
|
||||
if event_type in _SNAPSHOT_EVENTS
|
||||
else EventPayloadMode.RUNTIME
|
||||
),
|
||||
validation_mode=EventValidationMode.DIAGNOSTIC,
|
||||
sensitive_fields=_SENSITIVE_FIELDS.get(event_type, ()),
|
||||
legacy_reason=(
|
||||
None
|
||||
@@ -195,16 +249,23 @@ def validate_event_payload(
|
||||
) -> tuple[str, ...]:
|
||||
"""在发送边界诊断首批 typed payload,保持原对象和插件 dict 形状不变。"""
|
||||
try:
|
||||
model = get_event_contract(event_type).payload_model
|
||||
contract = get_event_contract(event_type)
|
||||
models = tuple(
|
||||
model
|
||||
for model in (contract.input_model, contract.output_model)
|
||||
if model is not None
|
||||
)
|
||||
except KeyError:
|
||||
# 动态插件在旧 ABI 下可能使用宿主枚举之外的字符串事件。
|
||||
return ()
|
||||
if model is None or payload is None:
|
||||
if not models or payload is None:
|
||||
return ()
|
||||
if isinstance(payload, model):
|
||||
return ()
|
||||
try:
|
||||
model.model_validate(payload)
|
||||
except (ValidationError, TypeError, ValueError) as error:
|
||||
return (str(error),)
|
||||
return ()
|
||||
problems: list[str] = []
|
||||
for model in models:
|
||||
if isinstance(payload, model):
|
||||
continue
|
||||
try:
|
||||
model.model_validate(payload)
|
||||
except (ValidationError, TypeError, ValueError) as error:
|
||||
problems.append(f"{model.__name__}: {error}")
|
||||
return tuple(problems)
|
||||
|
||||
@@ -0,0 +1,116 @@
|
||||
"""事件 payload 的只读契约快照访问。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, ValidationError
|
||||
|
||||
from app.runtime.event.contracts import get_event_contract, normalize_event_type
|
||||
from app.schemas.types import ChainEventType, EventType
|
||||
|
||||
|
||||
@dataclass(frozen=True, slots=True)
|
||||
class EventPayloadSnapshot:
|
||||
"""保存原始 payload 及按登记契约解析出的独立快照。"""
|
||||
|
||||
event_type: EventType | ChainEventType | str
|
||||
raw: Any
|
||||
payload: BaseModel | None
|
||||
input: BaseModel | None
|
||||
output: BaseModel | None
|
||||
known: bool
|
||||
schema_version: int | None
|
||||
payload_mode: str | None
|
||||
validation_mode: str | None
|
||||
errors: tuple[str, ...] = ()
|
||||
|
||||
@property
|
||||
def valid(self) -> bool:
|
||||
"""返回已知事件的全部登记模型是否都通过校验。"""
|
||||
return self.known and not self.errors
|
||||
|
||||
|
||||
def _model_snapshot(
|
||||
model: type[BaseModel] | None,
|
||||
payload: Any,
|
||||
label: str,
|
||||
cache: dict[type[BaseModel], BaseModel | None],
|
||||
errors: list[str],
|
||||
) -> BaseModel | None:
|
||||
"""按单个模型构造快照,同一模型在一次访问中只解析一次。"""
|
||||
if model is None:
|
||||
return None
|
||||
if model in cache:
|
||||
return cache[model]
|
||||
if isinstance(payload, model):
|
||||
cache[model] = payload.model_copy(deep=True)
|
||||
return cache[model]
|
||||
try:
|
||||
cache[model] = model.model_validate(payload)
|
||||
except (ValidationError, TypeError, ValueError) as error:
|
||||
cache[model] = None
|
||||
errors.append(f"{label}:{model.__name__}: {error}")
|
||||
return cache[model]
|
||||
|
||||
|
||||
def snapshot_event_data(
|
||||
event_type: EventType | ChainEventType | str,
|
||||
event_data: Any,
|
||||
) -> EventPayloadSnapshot:
|
||||
"""按事件注册契约解析 payload,未知自定义事件保持原始数据并返回未登记状态。"""
|
||||
normalized = normalize_event_type(event_type)
|
||||
try:
|
||||
contract = get_event_contract(normalized)
|
||||
except KeyError:
|
||||
return EventPayloadSnapshot(
|
||||
event_type=normalized,
|
||||
raw=event_data,
|
||||
payload=None,
|
||||
input=None,
|
||||
output=None,
|
||||
known=False,
|
||||
schema_version=None,
|
||||
payload_mode=None,
|
||||
validation_mode=None,
|
||||
)
|
||||
|
||||
errors: list[str] = []
|
||||
cache: dict[type[BaseModel], BaseModel | None] = {}
|
||||
payload_snapshot = _model_snapshot(
|
||||
contract.payload_model,
|
||||
event_data,
|
||||
"payload",
|
||||
cache,
|
||||
errors,
|
||||
)
|
||||
input_snapshot = _model_snapshot(
|
||||
contract.input_model,
|
||||
event_data,
|
||||
"input",
|
||||
cache,
|
||||
errors,
|
||||
)
|
||||
output_snapshot = _model_snapshot(
|
||||
contract.output_model,
|
||||
event_data,
|
||||
"output",
|
||||
cache,
|
||||
errors,
|
||||
)
|
||||
return EventPayloadSnapshot(
|
||||
event_type=normalized,
|
||||
raw=event_data,
|
||||
payload=payload_snapshot,
|
||||
input=input_snapshot,
|
||||
output=output_snapshot,
|
||||
known=True,
|
||||
schema_version=contract.schema_version,
|
||||
payload_mode=contract.payload_mode.value,
|
||||
validation_mode=contract.validation_mode.value,
|
||||
errors=tuple(errors),
|
||||
)
|
||||
|
||||
|
||||
__all__ = ["EventPayloadSnapshot", "snapshot_event_data"]
|
||||
+18
-9
@@ -7,26 +7,29 @@ import uuid
|
||||
from contextvars import ContextVar
|
||||
from dataclasses import dataclass
|
||||
from queue import Empty, PriorityQueue
|
||||
from typing import Callable, Dict, List, Optional, Tuple, Union, Any, Type
|
||||
from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple, Type, Union
|
||||
|
||||
from app.runtime.config import global_vars
|
||||
from app.runtime.thread import ThreadHelper
|
||||
from app.runtime.log import logger
|
||||
from app.schemas.event import ChainEventData
|
||||
from app.schemas.types import ChainEventType, EventType
|
||||
from app.runtime.rate import ExponentialBackoffRateLimiter
|
||||
from app.foundation.singleton import Singleton
|
||||
from app.runtime.config import global_vars
|
||||
from app.runtime.correlation import get_correlation_id
|
||||
from app.runtime.event.binding import (
|
||||
EventBindingResolver,
|
||||
EventHandlerBinding,
|
||||
HandlerInstanceResolver,
|
||||
)
|
||||
from app.runtime.event.contracts import normalize_event_type, validate_event_payload
|
||||
from app.runtime.event.dispatch import EventDispatcher
|
||||
from app.runtime.event.errors import EventErrorNotifier, EventErrorPolicy
|
||||
from app.runtime.event.registry import EventRegistry
|
||||
from app.runtime.event.contracts import normalize_event_type, validate_event_payload
|
||||
from app.runtime.correlation import get_correlation_id
|
||||
from app.runtime.log import logger
|
||||
from app.runtime.observability import record_metric
|
||||
from app.runtime.rate import ExponentialBackoffRateLimiter
|
||||
from app.runtime.thread import ThreadHelper
|
||||
from app.schemas.event import ChainEventData
|
||||
from app.schemas.types import ChainEventType, EventType
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.runtime.event.snapshot import EventPayloadSnapshot
|
||||
|
||||
DEFAULT_EVENT_PRIORITY = 10 # 事件的默认优先级
|
||||
MIN_EVENT_CONSUMER_THREADS = 1 # 最小事件消费者线程数
|
||||
@@ -84,6 +87,12 @@ class Event:
|
||||
event_name = getattr(self.event_type, "value", self.event_type)
|
||||
return f"<{event_kind}: {event_name}, ID: {self.event_id}, Priority: {self.priority}>"
|
||||
|
||||
def snapshot(self) -> "EventPayloadSnapshot":
|
||||
"""返回插件可安全读取的类型化 payload 快照,原始 event_data 保持不变。"""
|
||||
from app.runtime.event.snapshot import snapshot_event_data
|
||||
|
||||
return snapshot_event_data(self.event_type, self.event_data)
|
||||
|
||||
def __lt__(self, other):
|
||||
"""
|
||||
定义事件对象的比较规则,基于优先级比较
|
||||
|
||||
+203
-5
@@ -1,14 +1,89 @@
|
||||
from collections.abc import Mapping as _Mapping
|
||||
from enum import Enum as _Enum
|
||||
from pathlib import Path
|
||||
from typing import Iterable, Optional, Dict, Any, List, Set, Callable
|
||||
from typing import Annotated as _Annotated
|
||||
from typing import Any, Callable, Dict, Iterable, List, Optional, Set
|
||||
from typing import Union as _Union
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
|
||||
from pydantic import BeforeValidator as _BeforeValidator
|
||||
|
||||
from app.schemas.common import JsonData
|
||||
from app.schemas.types import MediaType, NotificationChannel
|
||||
from app.schemas.context import Context as _ContextSnapshotBase
|
||||
from app.schemas.context import MediaInfo as _MediaInfoSnapshot
|
||||
from app.schemas.context import MetaInfo as _MetaInfoSnapshot
|
||||
from app.schemas.file import FileItem
|
||||
from app.schemas.media import OptionalMediaIdentityMixin, RequiredMediaIdentityMixin
|
||||
from app.schemas.music import MusicInfo as _MusicInfoSnapshot
|
||||
from app.schemas.music import MusicMeta as _MusicMetaSnapshot
|
||||
from app.schemas.subscribe import Subscribe as _SubscribeSnapshot
|
||||
from app.schemas.transfer import TransferInfo
|
||||
from app.schemas.types import MediaSource
|
||||
from app.schemas.types import MediaSource, MediaType, NotificationChannel
|
||||
|
||||
JsonValue = JsonData
|
||||
"""事件扩展字段允许的 JSON 值类型。"""
|
||||
|
||||
|
||||
def _coerce_event_snapshot(value: Any) -> Any:
|
||||
"""把旧运行时对象转换为只用于契约校验的可读取快照。"""
|
||||
if isinstance(value, BaseModel):
|
||||
value = value.model_dump(mode="python")
|
||||
elif hasattr(value, "to_dict") and callable(value.to_dict):
|
||||
value = value.to_dict()
|
||||
elif isinstance(value, _Enum):
|
||||
return value.value
|
||||
elif isinstance(value, Path):
|
||||
return value.as_posix()
|
||||
if isinstance(value, _Mapping):
|
||||
return {str(key): _coerce_event_snapshot(item) for key, item in value.items()}
|
||||
if isinstance(value, (list, tuple, set, frozenset)):
|
||||
return [_coerce_event_snapshot(item) for item in value]
|
||||
return value
|
||||
|
||||
|
||||
EventJsonValue = _Annotated[JsonValue, _BeforeValidator(_coerce_event_snapshot)]
|
||||
MediaSnapshot = _Annotated[
|
||||
_Union[_MusicInfoSnapshot, _MediaInfoSnapshot],
|
||||
_BeforeValidator(_coerce_event_snapshot),
|
||||
]
|
||||
MetaSnapshot = _Annotated[
|
||||
_Union[_MusicMetaSnapshot, _MetaInfoSnapshot],
|
||||
_BeforeValidator(_coerce_event_snapshot),
|
||||
]
|
||||
|
||||
|
||||
class ContextSnapshot(_ContextSnapshotBase):
|
||||
"""资源上下文的稳定事件快照,不替换链路中的运行时 Context。"""
|
||||
|
||||
allowed_episodes: Optional[Set[int]] = None
|
||||
|
||||
model_config = ConfigDict(
|
||||
extra="allow",
|
||||
from_attributes=True,
|
||||
arbitrary_types_allowed=True,
|
||||
)
|
||||
|
||||
@model_validator(mode="before")
|
||||
@classmethod
|
||||
def coerce_runtime_context(cls, value: Any) -> Any:
|
||||
"""允许旧 Context/dataclass 进入校验,同时保持原对象继续投递。"""
|
||||
return _coerce_event_snapshot(value)
|
||||
|
||||
|
||||
class FileContextSnapshot(BaseModel):
|
||||
"""音乐批次中单个文件的元数据上下文快照。"""
|
||||
|
||||
path: str
|
||||
meta: Optional[MetaSnapshot] = None
|
||||
mediainfo: Optional[MediaSnapshot] = None
|
||||
|
||||
model_config = ConfigDict(extra="allow", from_attributes=True)
|
||||
|
||||
@model_validator(mode="before")
|
||||
@classmethod
|
||||
def coerce_runtime_context(cls, value: Any) -> Any:
|
||||
"""把旧文件上下文对象转换成可验证的字典。"""
|
||||
return _coerce_event_snapshot(value)
|
||||
|
||||
|
||||
class Event(BaseModel):
|
||||
@@ -26,7 +101,7 @@ class BaseEventData(BaseModel):
|
||||
事件数据的基类,所有具体事件数据类应继承自此类
|
||||
"""
|
||||
|
||||
pass
|
||||
model_config = ConfigDict(from_attributes=True, arbitrary_types_allowed=True)
|
||||
|
||||
|
||||
class ExtensibleEventData(BaseEventData):
|
||||
@@ -249,7 +324,130 @@ class ChainEventData(BaseEventData):
|
||||
链式事件数据的基类,所有具体事件数据类应继承自此类
|
||||
"""
|
||||
|
||||
pass
|
||||
model_config = ConfigDict(from_attributes=True, arbitrary_types_allowed=True)
|
||||
|
||||
|
||||
class ResourceSelectionContractData(ChainEventData):
|
||||
"""ResourceSelection 的稳定输入/输出契约。"""
|
||||
|
||||
contexts: List[ContextSnapshot] = Field(default_factory=list)
|
||||
downloader: Optional[str] = None
|
||||
origin: Optional[str] = None
|
||||
updated: bool = False
|
||||
updated_contexts: Optional[List[ContextSnapshot]] = None
|
||||
source: Optional[str] = "未知拦截源"
|
||||
|
||||
|
||||
class ResourceSelectionInputContractData(ChainEventData):
|
||||
"""ResourceSelection 事件输入字段。"""
|
||||
|
||||
contexts: List[ContextSnapshot] = Field(default_factory=list)
|
||||
downloader: Optional[str] = None
|
||||
origin: Optional[str] = None
|
||||
|
||||
|
||||
class ResourceSelectionOutputContractData(ChainEventData):
|
||||
"""ResourceSelection 事件插件回写字段。"""
|
||||
|
||||
updated: bool = False
|
||||
updated_contexts: Optional[List[ContextSnapshot]] = None
|
||||
source: Optional[str] = "未知拦截源"
|
||||
|
||||
|
||||
class ResourceDownloadContractData(ChainEventData):
|
||||
"""ResourceDownload 的稳定输入/输出契约。"""
|
||||
|
||||
context: Optional[ContextSnapshot] = None
|
||||
episodes: Optional[Set[int]] = None
|
||||
channel: Optional[NotificationChannel] = None
|
||||
origin: Optional[str] = None
|
||||
downloader: Optional[str] = None
|
||||
options: Optional[Dict[str, EventJsonValue]] = None
|
||||
cancel: bool = False
|
||||
source: str = "未知拦截源"
|
||||
reason: str = ""
|
||||
|
||||
|
||||
class ResourceDownloadInputContractData(ChainEventData):
|
||||
"""ResourceDownload 事件输入字段。"""
|
||||
|
||||
context: Optional[ContextSnapshot] = None
|
||||
episodes: Optional[Set[int]] = None
|
||||
channel: Optional[NotificationChannel] = None
|
||||
origin: Optional[str] = None
|
||||
downloader: Optional[str] = None
|
||||
options: Optional[Dict[str, EventJsonValue]] = None
|
||||
|
||||
|
||||
class ResourceDownloadOutputContractData(ChainEventData):
|
||||
"""ResourceDownload 事件插件回写字段。"""
|
||||
|
||||
cancel: bool = False
|
||||
source: str = "未知拦截源"
|
||||
reason: str = ""
|
||||
|
||||
|
||||
class DownloadAddedContractData(BaseEventData):
|
||||
"""DownloadAdded 的稳定下载上下文契约。"""
|
||||
|
||||
hash: str
|
||||
context: ContextSnapshot
|
||||
username: Optional[str] = None
|
||||
downloader: Optional[str] = None
|
||||
episodes: List[int] = Field(default_factory=list)
|
||||
source: Optional[str] = None
|
||||
idempotency_key: Optional[str] = None
|
||||
|
||||
|
||||
class TransferResultContractData(BaseEventData):
|
||||
"""整理结果事件的稳定媒体与元数据契约。"""
|
||||
|
||||
fileitem: Optional[FileItem] = None
|
||||
meta: Optional[MetaSnapshot] = None
|
||||
mediainfo: Optional[MediaSnapshot] = None
|
||||
transferinfo: Optional[TransferInfo] = None
|
||||
downloader: Optional[str] = None
|
||||
download_hash: Optional[str] = None
|
||||
transfer_history_id: Optional[int] = None
|
||||
idempotency_key: Optional[str] = None
|
||||
|
||||
|
||||
class MetadataScrapeContractData(ExtensibleEventData):
|
||||
"""MetadataScrape 的稳定媒体刮削契约。"""
|
||||
|
||||
fileitem: FileItem
|
||||
file_list: List[str] = Field(default_factory=list)
|
||||
meta: Optional[MetaSnapshot] = None
|
||||
mediainfo: Optional[MediaSnapshot] = None
|
||||
overwrite: bool = False
|
||||
file_contexts: List[FileContextSnapshot] = Field(default_factory=list)
|
||||
|
||||
|
||||
class SubscribeCompletionCheckContractData(ChainEventData):
|
||||
"""订阅完成判定的稳定订阅、媒体和元数据契约。"""
|
||||
|
||||
subscribe: Optional[_SubscribeSnapshot] = None
|
||||
mediainfo: Optional[MediaSnapshot] = None
|
||||
meta: Optional[MetaSnapshot] = None
|
||||
cancel: bool = False
|
||||
source: str = "未知来源"
|
||||
reason: str = ""
|
||||
|
||||
|
||||
class SubscribeCompletionCheckInputContractData(ChainEventData):
|
||||
"""SubscribeCompletionCheck 事件输入字段。"""
|
||||
|
||||
subscribe: Optional[_SubscribeSnapshot] = None
|
||||
mediainfo: Optional[MediaSnapshot] = None
|
||||
meta: Optional[MetaSnapshot] = None
|
||||
|
||||
|
||||
class SubscribeCompletionCheckOutputContractData(ChainEventData):
|
||||
"""SubscribeCompletionCheck 事件插件回写字段。"""
|
||||
|
||||
cancel: bool = False
|
||||
source: str = "未知来源"
|
||||
reason: str = ""
|
||||
|
||||
|
||||
class PluginDataResetEventData(ChainEventData):
|
||||
|
||||
@@ -71,6 +71,7 @@ SCHEMA_EXPORTS = {
|
||||
'ConfigDict': ('app.schemas.workflow', 'ConfigDict'),
|
||||
'ContentType': ('app.schemas.message', 'ContentType'),
|
||||
'Context': ('app.schemas.workflow', 'Context'),
|
||||
'ContextSnapshot': ('app.schemas.event', 'ContextSnapshot'),
|
||||
'ConversationMemory': ('app.schemas.agent', 'ConversationMemory'),
|
||||
'CookieActionResponse': ('app.schemas.servcookie', 'CookieActionResponse'),
|
||||
'CookieData': ('app.schemas.servcookie', 'CookieData'),
|
||||
@@ -87,6 +88,7 @@ SCHEMA_EXPORTS = {
|
||||
'DiscoverMediaSource': ('app.schemas.event', 'DiscoverMediaSource'),
|
||||
'DiscoverSourceEventData': ('app.schemas.event', 'DiscoverSourceEventData'),
|
||||
'Discriminator': ('app.schemas.context', 'Discriminator'),
|
||||
'DownloadAddedContractData': ('app.schemas.event', 'DownloadAddedContractData'),
|
||||
'DownloadAddedData': ('app.schemas.download', 'DownloadAddedData'),
|
||||
'DownloadAddedEventData': ('app.schemas.event', 'DownloadAddedEventData'),
|
||||
'DownloadDeletedEventData': ('app.schemas.event', 'DownloadDeletedEventData'),
|
||||
@@ -109,9 +111,11 @@ SCHEMA_EXPORTS = {
|
||||
'EpisodeFormatRule': ('app.schemas.transfer', 'EpisodeFormatRule'),
|
||||
'ErrorRequest': ('app.schemas.monitoring', 'ErrorRequest'),
|
||||
'Event': ('app.schemas.event', 'Event'),
|
||||
'EventJsonValue': ('app.schemas.event', 'EventJsonValue'),
|
||||
'ExistMediaInfo': ('app.schemas.mediaserver', 'ExistMediaInfo'),
|
||||
'ExtensibleEventData': ('app.schemas.event', 'ExtensibleEventData'),
|
||||
'Field': ('app.schemas.mcp', 'Field'),
|
||||
'FileContextSnapshot': ('app.schemas.event', 'FileContextSnapshot'),
|
||||
'FileItem': ('app.schemas.workflow', 'FileItem'),
|
||||
'FileNameData': ('app.schemas.common', 'FileNameData'),
|
||||
'FileURI': ('app.schemas.file', 'FileURI'),
|
||||
@@ -125,6 +129,7 @@ SCHEMA_EXPORTS = {
|
||||
'JsonData': ('app.schemas.mcp', 'JsonData'),
|
||||
'JsonObject': ('app.schemas.common', 'JsonObject'),
|
||||
'JsonObjectList': ('app.schemas.common', 'JsonObjectList'),
|
||||
'JsonValue': ('app.schemas.event', 'JsonValue'),
|
||||
'LLMAuthStatus': ('app.schemas.llm', 'LLMAuthStatus'),
|
||||
'LLMModelCatalogData': ('app.schemas.llm', 'LLMModelCatalogData'),
|
||||
'LLMModelInfo': ('app.schemas.llm', 'LLMModelInfo'),
|
||||
@@ -196,6 +201,7 @@ SCHEMA_EXPORTS = {
|
||||
'MediaServerPlayData': ('app.schemas.mediaserver', 'MediaServerPlayData'),
|
||||
'MediaServerPlayItem': ('app.schemas.mediaserver', 'MediaServerPlayItem'),
|
||||
'MediaServerSeasonInfo': ('app.schemas.mediaserver', 'MediaServerSeasonInfo'),
|
||||
'MediaSnapshot': ('app.schemas.event', 'MediaSnapshot'),
|
||||
'MediaSource': ('app.schemas.transfer', 'MediaSource'),
|
||||
'MediaSourceInfo': ('app.schemas.event', 'MediaSourceInfo'),
|
||||
'MediaType': ('app.schemas.subscribe', 'MediaType'),
|
||||
@@ -208,6 +214,8 @@ SCHEMA_EXPORTS = {
|
||||
'MessageResponse': ('app.schemas.message', 'MessageResponse'),
|
||||
'MessageType': ('app.schemas.message', 'MessageType'),
|
||||
'MetaInfo': ('app.schemas.transfer', 'MetaInfo'),
|
||||
'MetaSnapshot': ('app.schemas.event', 'MetaSnapshot'),
|
||||
'MetadataScrapeContractData': ('app.schemas.event', 'MetadataScrapeContractData'),
|
||||
'MetadataScrapeEventData': ('app.schemas.event', 'MetadataScrapeEventData'),
|
||||
'MfaChallenge': ('app.schemas.token', 'MfaChallenge'),
|
||||
'MfaStatusData': ('app.schemas.mfa', 'MfaStatusData'),
|
||||
@@ -295,8 +303,14 @@ SCHEMA_EXPORTS = {
|
||||
'RefreshMediaItem': ('app.schemas.mediaserver', 'RefreshMediaItem'),
|
||||
'RequestMetrics': ('app.schemas.monitoring', 'RequestMetrics'),
|
||||
'RequiredMediaIdentityMixin': ('app.schemas.music', 'RequiredMediaIdentityMixin'),
|
||||
'ResourceDownloadContractData': ('app.schemas.event', 'ResourceDownloadContractData'),
|
||||
'ResourceDownloadEventData': ('app.schemas.event', 'ResourceDownloadEventData'),
|
||||
'ResourceDownloadInputContractData': ('app.schemas.event', 'ResourceDownloadInputContractData'),
|
||||
'ResourceDownloadOutputContractData': ('app.schemas.event', 'ResourceDownloadOutputContractData'),
|
||||
'ResourceSelectionContractData': ('app.schemas.event', 'ResourceSelectionContractData'),
|
||||
'ResourceSelectionEventData': ('app.schemas.event', 'ResourceSelectionEventData'),
|
||||
'ResourceSelectionInputContractData': ('app.schemas.event', 'ResourceSelectionInputContractData'),
|
||||
'ResourceSelectionOutputContractData': ('app.schemas.event', 'ResourceSelectionOutputContractData'),
|
||||
'Response': ('app.schemas.response', 'Response'),
|
||||
'RootModel': ('app.schemas.mcp', 'RootModel'),
|
||||
'RuleTestData': ('app.schemas.system', 'RuleTestData'),
|
||||
@@ -349,7 +363,10 @@ SCHEMA_EXPORTS = {
|
||||
'Subscribe': ('app.schemas.workflow', 'Subscribe'),
|
||||
'SubscribeAddedEventData': ('app.schemas.event', 'SubscribeAddedEventData'),
|
||||
'SubscribeCompleteEventData': ('app.schemas.event', 'SubscribeCompleteEventData'),
|
||||
'SubscribeCompletionCheckContractData': ('app.schemas.event', 'SubscribeCompletionCheckContractData'),
|
||||
'SubscribeCompletionCheckEventData': ('app.schemas.event', 'SubscribeCompletionCheckEventData'),
|
||||
'SubscribeCompletionCheckInputContractData': ('app.schemas.event', 'SubscribeCompletionCheckInputContractData'),
|
||||
'SubscribeCompletionCheckOutputContractData': ('app.schemas.event', 'SubscribeCompletionCheckOutputContractData'),
|
||||
'SubscribeDeletedEventData': ('app.schemas.event', 'SubscribeDeletedEventData'),
|
||||
'SubscribeDownloadFileInfo': ('app.schemas.subscribe', 'SubscribeDownloadFileInfo'),
|
||||
'SubscribeEpisodeInfo': ('app.schemas.subscribe', 'SubscribeEpisodeInfo'),
|
||||
@@ -396,6 +413,7 @@ SCHEMA_EXPORTS = {
|
||||
'TransferOverwriteCheckEventData': ('app.schemas.event', 'TransferOverwriteCheckEventData'),
|
||||
'TransferRenameBuildEventData': ('app.schemas.event', 'TransferRenameBuildEventData'),
|
||||
'TransferRenameEventData': ('app.schemas.event', 'TransferRenameEventData'),
|
||||
'TransferResultContractData': ('app.schemas.event', 'TransferResultContractData'),
|
||||
'TransferResultEventData': ('app.schemas.event', 'TransferResultEventData'),
|
||||
'TransferTorrent': ('app.schemas.transfer', 'TransferTorrent'),
|
||||
'TypeAdapter': ('app.schemas.mcp', 'TypeAdapter'),
|
||||
|
||||
+42
-2
@@ -1,6 +1,46 @@
|
||||
"""插件事件订阅与发布接口。"""
|
||||
|
||||
from app.runtime.event.snapshot import EventPayloadSnapshot, snapshot_event_data
|
||||
from app.runtime.events import Event, EventManager, eventmanager
|
||||
from app.schemas.event import (
|
||||
ContextSnapshot,
|
||||
DownloadAddedContractData,
|
||||
FileContextSnapshot,
|
||||
MediaSnapshot,
|
||||
MetadataScrapeContractData,
|
||||
MetaSnapshot,
|
||||
ResourceDownloadContractData,
|
||||
ResourceDownloadInputContractData,
|
||||
ResourceDownloadOutputContractData,
|
||||
ResourceSelectionContractData,
|
||||
ResourceSelectionInputContractData,
|
||||
ResourceSelectionOutputContractData,
|
||||
SubscribeCompletionCheckContractData,
|
||||
SubscribeCompletionCheckInputContractData,
|
||||
SubscribeCompletionCheckOutputContractData,
|
||||
TransferResultContractData,
|
||||
)
|
||||
|
||||
|
||||
__all__ = ["Event", "EventManager", "eventmanager"]
|
||||
__all__ = [
|
||||
"ContextSnapshot",
|
||||
"DownloadAddedContractData",
|
||||
"Event",
|
||||
"EventManager",
|
||||
"EventPayloadSnapshot",
|
||||
"FileContextSnapshot",
|
||||
"MediaSnapshot",
|
||||
"MetadataScrapeContractData",
|
||||
"MetaSnapshot",
|
||||
"ResourceDownloadContractData",
|
||||
"ResourceDownloadInputContractData",
|
||||
"ResourceDownloadOutputContractData",
|
||||
"ResourceSelectionContractData",
|
||||
"ResourceSelectionInputContractData",
|
||||
"ResourceSelectionOutputContractData",
|
||||
"SubscribeCompletionCheckContractData",
|
||||
"SubscribeCompletionCheckInputContractData",
|
||||
"SubscribeCompletionCheckOutputContractData",
|
||||
"TransferResultContractData",
|
||||
"eventmanager",
|
||||
"snapshot_event_data",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user