feat(context): add recognition context metadata

This commit is contained in:
InfinityPacer
2026-05-08 14:38:04 +08:00
committed by jxxghp
parent 14b366a648
commit 2f231fe632
2 changed files with 21 additions and 1 deletions

View File

@@ -832,6 +832,14 @@ class Context:
torrent_info: TorrentInfo = None
# 媒体识别失败次数
media_recognize_fail_count: int = 0
# 候选资源来源rss、spider、search、unknown。
resource_source: str = "unknown"
# 候选匹配来源tmdbid、doubanid、imdbid、title、plugin、unknown。
match_source: str = "unknown"
# 候选自身是否已经识别出有效媒体 ID。
candidate_recognized: bool = False
# 当前 media_info 是否为目标媒体回填,而不是候选自身识别结果。
media_info_is_target: bool = False
def to_dict(self):
"""
@@ -841,5 +849,9 @@ class Context:
"meta_info": self.meta_info.to_dict() if self.meta_info else None,
"torrent_info": self.torrent_info.to_dict() if self.torrent_info else None,
"media_info": self.media_info.to_dict() if self.media_info else None,
"media_recognize_fail_count": self.media_recognize_fail_count
"media_recognize_fail_count": self.media_recognize_fail_count,
"resource_source": self.resource_source,
"match_source": self.match_source,
"candidate_recognized": self.candidate_recognized,
"media_info_is_target": self.media_info_is_target,
}

View File

@@ -248,6 +248,14 @@ class Context(BaseModel):
media_info: Optional[Union[MediaInfo, Any]] = None
# 种子信息
torrent_info: Optional[TorrentInfo] = None
# 候选资源来源rss、spider、search、unknown
resource_source: Optional[str] = "unknown"
# 候选匹配来源tmdbid、doubanid、imdbid、title、plugin、unknown
match_source: Optional[str] = "unknown"
# 候选自身是否已经识别出有效媒体 ID
candidate_recognized: Optional[bool] = False
# 当前 media_info 是否为目标媒体回填
media_info_is_target: Optional[bool] = False
class MediaSeason(BaseModel):