mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-05-19 07:26:51 +00:00
125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
from enum import Enum
|
|
from typing import Optional, List, Tuple, Union
|
|
|
|
from pydantic import BaseModel, Field, ConfigDict
|
|
|
|
from .imdbapi import ImdbApiTitle, ImdbApiEpisode, ImdbApiCredit
|
|
from .imdbtypes import ImdbTitle, ImdbName, ImdbImage, ImdbVideo, AkasNode, TitleEdge
|
|
|
|
|
|
class ErrorType(Enum):
|
|
PERSISTED_QUERY_NOT_FOUND = 'PERSISTED_QUERY_NOT_FOUND'
|
|
|
|
|
|
class StaffPickEntry(BaseModel):
|
|
name: str
|
|
ttconst: str
|
|
rmconst: str
|
|
detail: Optional[str] = ""
|
|
description: Optional[str] = ""
|
|
relatedconst: List[str] = Field(default_factory=list)
|
|
viconst: Optional[str] = None
|
|
|
|
|
|
class VerticalList(BaseModel):
|
|
titles: List[ImdbTitle] = Field(default_factory=list)
|
|
names: List[ImdbName] = Field(default_factory=list)
|
|
videos: List[ImdbVideo] = Field(default_factory=list)
|
|
images: List[ImdbImage] = Field(default_factory=list)
|
|
|
|
|
|
class StaffPickApiResponse(BaseModel):
|
|
updated_at: Optional[str]
|
|
entries: List[StaffPickEntry] = Field(default_factory=list)
|
|
imdb_items: VerticalList
|
|
|
|
|
|
class ImdbMediaInfo(ImdbApiTitle):
|
|
akas: List[AkasNode] = Field(default_factory=list)
|
|
episodes: List[ImdbApiEpisode] = Field(default_factory=list)
|
|
credits: List[ImdbApiCredit] = Field(default_factory=list)
|
|
|
|
@classmethod
|
|
def from_title(
|
|
cls,
|
|
title: ImdbApiTitle,
|
|
akas: Optional[List[AkasNode]] = None,
|
|
episodes: Optional[List[ImdbApiEpisode]] = None,
|
|
api_credits: Optional[List[ImdbApiCredit]] = None,
|
|
) -> "ImdbMediaInfo":
|
|
return cls(
|
|
**title.model_dump(exclude_none=True, by_alias=True),
|
|
akas=akas if akas is not None else [],
|
|
episodes=episodes if episodes is not None else [],
|
|
credits=api_credits if api_credits is not None else []
|
|
)
|
|
|
|
|
|
class ImdbApiHash(BaseModel):
|
|
advanced_title_search: str = Field(alias="AdvancedTitleSearch")
|
|
|
|
|
|
class PageInfo(BaseModel):
|
|
has_previous_page: Optional[bool] = Field(None, alias="hasPreviousPage")
|
|
has_next_page: Optional[bool] = Field(None, alias="hasNextPage")
|
|
start_cursor: Optional[str] = Field(None, alias="startCursor")
|
|
end_cursor: Optional[str] = Field(None, alias="endCursor")
|
|
|
|
|
|
class FilterInfo(BaseModel):
|
|
filter_id: Optional[str] = Field(default=None, alias='filterId')
|
|
text: Optional[str] = Field(default=None, alias='text')
|
|
total: Optional[int] = Field(default=None, alias='total')
|
|
|
|
|
|
class SearchState(BaseModel):
|
|
total: int = 0
|
|
page_info: PageInfo = Field(default_factory=PageInfo, alias="pageInfo")
|
|
genres: List[FilterInfo] = Field(default_factory=list)
|
|
keywords: List[FilterInfo] = Field(default_factory=list)
|
|
title_types: List[FilterInfo] = Field(default_factory=list, alias='titleTypes')
|
|
|
|
|
|
class AdvancedTitleSearch(SearchState):
|
|
edges: List[TitleEdge] = Field(default_factory=list)
|
|
|
|
|
|
class AdvancedTitleSearchResponse(BaseModel):
|
|
advanced_title_search: AdvancedTitleSearch = Field(default_factory=AdvancedTitleSearch, alias="advancedTitleSearch")
|
|
|
|
|
|
class SearchParams(BaseModel):
|
|
title_types: Optional[Tuple[str, ...]] = None
|
|
genres: Optional[Tuple[str, ...]] = None
|
|
sort_by: str = 'POPULARITY'
|
|
sort_order: str = 'ASC'
|
|
rating_min: Optional[float] = None
|
|
rating_max: Optional[float] = None
|
|
countries: Optional[Tuple[str, ...]] = None
|
|
languages: Optional[Tuple[str, ...]] = None
|
|
release_date_end: Optional[str] = None
|
|
release_date_start: Optional[str] = None
|
|
award_constraint: Optional[Tuple[str, ...]] = None
|
|
ranked: Optional[Tuple[str, ...]] = None
|
|
interests: Optional[Tuple[str, ...]] = None
|
|
|
|
model_config = ConfigDict(
|
|
frozen=True,
|
|
validate_assignment=False
|
|
)
|
|
|
|
|
|
class ErrorExtension(BaseModel):
|
|
code: Union[ErrorType, str]
|
|
error_type: str = Field('CLIENT', alias='errorType')
|
|
is_retryable: bool = Field(False, alias='isRetryable')
|
|
|
|
|
|
class ErrorValue(BaseModel):
|
|
message: Optional[str] = Field(default=None, alias='message')
|
|
extensions: Optional[ErrorExtension]
|
|
|
|
|
|
class ErrorResponse(BaseModel):
|
|
errors: Optional[List[ErrorValue]] = Field(default_factory=list)
|