mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-09 23:30:13 +08:00
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""站点搜索调用的轻量结果观察上下文。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from contextlib import contextmanager
|
|
from contextvars import ContextVar
|
|
from dataclasses import dataclass
|
|
from typing import Iterator, Optional
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class SiteSearchObservation:
|
|
"""记录一次站点搜索是否发出请求及其可治理结果。"""
|
|
|
|
attempted: bool = False
|
|
outcome: str = "skipped"
|
|
error: Optional[str] = None
|
|
raw_count: Optional[int] = None
|
|
has_more: Optional[bool] = None
|
|
|
|
|
|
_current_observation: ContextVar[Optional[SiteSearchObservation]] = ContextVar(
|
|
"site_search_observation",
|
|
default=None,
|
|
)
|
|
|
|
|
|
@contextmanager
|
|
def capture_site_search_observation() -> Iterator[SiteSearchObservation]:
|
|
"""为当前同步 worker 捕获一次索引器搜索结果。"""
|
|
observation = SiteSearchObservation()
|
|
token = _current_observation.set(observation)
|
|
try:
|
|
yield observation
|
|
finally:
|
|
_current_observation.reset(token)
|
|
|
|
|
|
def report_site_search_outcome(
|
|
*,
|
|
attempted: bool,
|
|
outcome: str,
|
|
error: Optional[str] = None,
|
|
) -> None:
|
|
"""由索引器在不改变公开返回合同的前提下发布调用结果。"""
|
|
observation = _current_observation.get()
|
|
if observation is None:
|
|
return
|
|
if outcome == "skipped" and observation.outcome == "deferred":
|
|
return
|
|
observation.attempted = attempted
|
|
observation.outcome = outcome
|
|
observation.error = error
|
|
|
|
|
|
def report_site_search_page(*, raw_count: int, has_more: Optional[bool] = None) -> None:
|
|
"""发布过滤前的页事实；短页不自动等价于末页，旧 List 返回 ABI 保持不变。"""
|
|
observation = _current_observation.get()
|
|
if observation is not None:
|
|
observation.raw_count = max(raw_count, observation.raw_count or 0)
|
|
if has_more is not None:
|
|
observation.has_more = has_more
|