114 lines
2.5 KiB
Python
114 lines
2.5 KiB
Python
"""
|
|
Pydantic models for NFOGuard API
|
|
"""
|
|
from pydantic import BaseModel
|
|
from typing import Optional, Dict, Any, List
|
|
|
|
|
|
class SonarrWebhook(BaseModel):
|
|
"""Sonarr webhook payload model"""
|
|
eventType: str
|
|
series: Optional[Dict[str, Any]] = None
|
|
episodes: Optional[list] = []
|
|
episodeFile: Optional[Dict[str, Any]] = None
|
|
isUpgrade: Optional[bool] = False
|
|
|
|
class Config:
|
|
extra = "allow"
|
|
|
|
|
|
class RadarrWebhook(BaseModel):
|
|
"""Radarr webhook payload model"""
|
|
eventType: str
|
|
movie: Optional[Dict[str, Any]] = None
|
|
movieFile: Optional[Dict[str, Any]] = None
|
|
isUpgrade: Optional[bool] = False
|
|
deletedFiles: Optional[list] = []
|
|
remoteMovie: Optional[Dict[str, Any]] = None
|
|
renamedMovieFiles: Optional[List[Dict[str, Any]]] = None
|
|
|
|
class Config:
|
|
extra = "allow"
|
|
|
|
|
|
class HealthResponse(BaseModel):
|
|
"""Health check response model"""
|
|
status: str
|
|
version: str
|
|
uptime: str
|
|
database_status: str
|
|
radarr_database: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class TVSeasonRequest(BaseModel):
|
|
"""TV season processing request model"""
|
|
series_path: str
|
|
season_name: str
|
|
|
|
|
|
class TVEpisodeRequest(BaseModel):
|
|
"""TV episode processing request model"""
|
|
series_path: str
|
|
season: int
|
|
episode: int
|
|
|
|
|
|
# Web interface models
|
|
class MovieUpdateRequest(BaseModel):
|
|
"""Request to update movie dateadded"""
|
|
imdb_id: str
|
|
dateadded: Optional[str]
|
|
source: str
|
|
|
|
|
|
class EpisodeUpdateRequest(BaseModel):
|
|
"""Request to update episode dateadded"""
|
|
imdb_id: str
|
|
season: int
|
|
episode: int
|
|
dateadded: Optional[str]
|
|
source: str
|
|
|
|
|
|
class BulkUpdateRequest(BaseModel):
|
|
"""Request for bulk source updates"""
|
|
media_type: str # "movies" or "episodes"
|
|
old_source: str
|
|
new_source: str
|
|
|
|
|
|
class MovieResponse(BaseModel):
|
|
"""Movie data response"""
|
|
imdb_id: str
|
|
title: str
|
|
path: str
|
|
released: Optional[str]
|
|
dateadded: Optional[str]
|
|
source: Optional[str]
|
|
has_video_file: bool
|
|
last_updated: str
|
|
|
|
|
|
class SeriesResponse(BaseModel):
|
|
"""TV series data response"""
|
|
imdb_id: str
|
|
title: str
|
|
path: str
|
|
last_updated: str
|
|
total_episodes: int
|
|
episodes_with_dates: int
|
|
episodes_with_video: int
|
|
|
|
|
|
class EpisodeResponse(BaseModel):
|
|
"""TV episode data response"""
|
|
season: int
|
|
episode: int
|
|
aired: Optional[str]
|
|
dateadded: Optional[str]
|
|
source: Optional[str]
|
|
has_video_file: bool
|
|
last_updated: str
|
|
series_path: str
|
|
season_name: str
|
|
episode_name: str |