|
| 1 | +from typing import Literal, Optional |
| 2 | + |
1 | 3 | from fastapi import APIRouter, Depends |
2 | 4 | from fastapi.responses import JSONResponse |
3 | 5 | from pydantic import BaseModel |
4 | 6 |
|
| 7 | +from module.conf import settings |
| 8 | +from module.database import Database |
5 | 9 | from module.manager import TorrentManager |
6 | 10 | from module.models import APIResponse, Bangumi, BangumiUpdate |
| 11 | +from module.parser.analyser.offset_detector import ( |
| 12 | + OffsetSuggestion as DetectorSuggestion, |
| 13 | +) |
| 14 | +from module.parser.analyser.offset_detector import detect_offset_mismatch |
| 15 | +from module.parser.analyser.tmdb_parser import tmdb_parser |
7 | 16 | from module.security.api import UNAUTHORIZED, get_current_user |
8 | 17 |
|
9 | 18 | from .response import u_response |
10 | 19 |
|
11 | 20 |
|
12 | 21 | class OffsetSuggestion(BaseModel): |
| 22 | + """Legacy offset suggestion model.""" |
13 | 23 | suggested_offset: int |
14 | 24 | reason: str |
15 | 25 |
|
| 26 | + |
| 27 | +class TMDBSummary(BaseModel): |
| 28 | + """Summary of TMDB data for display.""" |
| 29 | + title: str |
| 30 | + total_seasons: int |
| 31 | + season_episode_counts: dict[int, int] |
| 32 | + status: Optional[str] |
| 33 | + virtual_season_starts: Optional[dict[int, list[int]]] = None # {1: [1, 29], ...} |
| 34 | + |
| 35 | + |
| 36 | +class OffsetSuggestionDetail(BaseModel): |
| 37 | + """Detailed offset suggestion from detector.""" |
| 38 | + season_offset: int |
| 39 | + episode_offset: int |
| 40 | + reason: str |
| 41 | + confidence: Literal["high", "medium", "low"] |
| 42 | + |
| 43 | + |
| 44 | +class DetectOffsetRequest(BaseModel): |
| 45 | + """Request body for detect-offset endpoint.""" |
| 46 | + title: str |
| 47 | + parsed_season: int |
| 48 | + parsed_episode: int |
| 49 | + |
| 50 | + |
| 51 | +class DetectOffsetResponse(BaseModel): |
| 52 | + """Response for detect-offset endpoint.""" |
| 53 | + has_mismatch: bool |
| 54 | + suggestion: Optional[OffsetSuggestionDetail] |
| 55 | + tmdb_info: Optional[TMDBSummary] |
| 56 | + |
16 | 57 | router = APIRouter(prefix="/bangumi", tags=["bangumi"]) |
17 | 58 |
|
18 | 59 |
|
@@ -202,3 +243,99 @@ async def suggest_offset(bangumi_id: int): |
202 | 243 | with TorrentManager() as manager: |
203 | 244 | resp = await manager.suggest_offset(bangumi_id) |
204 | 245 | return resp |
| 246 | + |
| 247 | + |
| 248 | +@router.post( |
| 249 | + path="/detect-offset", |
| 250 | + response_model=DetectOffsetResponse, |
| 251 | + dependencies=[Depends(get_current_user)], |
| 252 | +) |
| 253 | +async def detect_offset(request: DetectOffsetRequest): |
| 254 | + """Detect season/episode mismatch with TMDB data. |
| 255 | +
|
| 256 | + Called by frontend before adding/subscribing to check if offsets are needed. |
| 257 | + """ |
| 258 | + language = settings.rss_parser.language |
| 259 | + tmdb_info = await tmdb_parser(request.title, language) |
| 260 | + |
| 261 | + if not tmdb_info: |
| 262 | + return DetectOffsetResponse( |
| 263 | + has_mismatch=False, |
| 264 | + suggestion=None, |
| 265 | + tmdb_info=None, |
| 266 | + ) |
| 267 | + |
| 268 | + # Detect mismatch |
| 269 | + suggestion = detect_offset_mismatch( |
| 270 | + parsed_season=request.parsed_season, |
| 271 | + parsed_episode=request.parsed_episode, |
| 272 | + tmdb_info=tmdb_info, |
| 273 | + ) |
| 274 | + |
| 275 | + # Build TMDB summary |
| 276 | + tmdb_summary = TMDBSummary( |
| 277 | + title=tmdb_info.title, |
| 278 | + total_seasons=tmdb_info.last_season, |
| 279 | + season_episode_counts=tmdb_info.season_episode_counts or {}, |
| 280 | + status=tmdb_info.series_status, |
| 281 | + virtual_season_starts=tmdb_info.virtual_season_starts, |
| 282 | + ) |
| 283 | + |
| 284 | + if suggestion: |
| 285 | + return DetectOffsetResponse( |
| 286 | + has_mismatch=True, |
| 287 | + suggestion=OffsetSuggestionDetail( |
| 288 | + season_offset=suggestion.season_offset, |
| 289 | + episode_offset=suggestion.episode_offset, |
| 290 | + reason=suggestion.reason, |
| 291 | + confidence=suggestion.confidence, |
| 292 | + ), |
| 293 | + tmdb_info=tmdb_summary, |
| 294 | + ) |
| 295 | + |
| 296 | + return DetectOffsetResponse( |
| 297 | + has_mismatch=False, |
| 298 | + suggestion=None, |
| 299 | + tmdb_info=tmdb_summary, |
| 300 | + ) |
| 301 | + |
| 302 | + |
| 303 | +@router.post( |
| 304 | + path="/dismiss-review/{bangumi_id}", |
| 305 | + response_model=APIResponse, |
| 306 | + dependencies=[Depends(get_current_user)], |
| 307 | +) |
| 308 | +async def dismiss_review(bangumi_id: int): |
| 309 | + """Clear the needs_review flag for a bangumi after user reviews.""" |
| 310 | + with Database() as db: |
| 311 | + success = db.bangumi.clear_needs_review(bangumi_id) |
| 312 | + |
| 313 | + if success: |
| 314 | + return JSONResponse( |
| 315 | + status_code=200, |
| 316 | + content={ |
| 317 | + "status": True, |
| 318 | + "msg_en": "Review dismissed.", |
| 319 | + "msg_zh": "已取消检查标记。", |
| 320 | + }, |
| 321 | + ) |
| 322 | + else: |
| 323 | + return JSONResponse( |
| 324 | + status_code=404, |
| 325 | + content={ |
| 326 | + "status": False, |
| 327 | + "msg_en": f"Bangumi {bangumi_id} not found.", |
| 328 | + "msg_zh": f"未找到番剧 {bangumi_id}。", |
| 329 | + }, |
| 330 | + ) |
| 331 | + |
| 332 | + |
| 333 | +@router.get( |
| 334 | + path="/needs-review", |
| 335 | + response_model=list[Bangumi], |
| 336 | + dependencies=[Depends(get_current_user)], |
| 337 | +) |
| 338 | +async def get_needs_review(): |
| 339 | + """Get all bangumi that need review for offset mismatch.""" |
| 340 | + with Database() as db: |
| 341 | + return db.bangumi.get_needs_review() |
0 commit comments