mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-05-25 07:26:52 +00:00
19f52d6217
ImdbSource - 使用 Pydantic 重构 IMDb API ClashRuleProvider - fix: 过早实例化系统 Scheduler - fix: 缺少 PyYAML - 配置使用 Pydantic
113 lines
3.7 KiB
Python
113 lines
3.7 KiB
Python
from enum import Enum
|
|
from typing import Optional, List, Tuple, Union
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
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, parent: ImdbApiTitle, **kwargs):
|
|
return cls(**parent.dict(by_alias=True), **kwargs)
|
|
|
|
|
|
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
|
|
|
|
class Config:
|
|
frozen = True
|
|
allow_mutation = 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)
|