| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- from enum import Enum
- from typing import Optional
- YOUTUBE_VIDEO_URL = "https://www.youtube.com/watch?v="
- IMDB_VIDEO_URL = "https://www.imdb.com/title/tt"
- class VideoType(Enum):
- UNKNOWN = "U"
- TV_EPISODE = "E"
- MOVIE = "M"
- SKATE_VIDEO = "S"
- YOUTUBE = "Y"
- @classmethod
- def as_choices(cls) -> tuple:
- return tuple((i.name, i.value) for i in cls)
- class VideoMetadata:
- title: str
- video_type: VideoType = VideoType.UNKNOWN
- base_run_time_seconds: int = (
- 60 # Silly default, but things break if this is 0 or null
- )
- imdb_id: Optional[str]
- tmdb_id: Optional[str]
- youtube_id: Optional[str]
- # IMDB specific
- episode_number: Optional[str]
- season_number: Optional[str]
- next_imdb_id: Optional[str]
- year: Optional[int]
- tv_series_id: Optional[int]
- plot: Optional[str]
- imdb_rating: Optional[str]
- tmdb_rating: Optional[str]
- cover_url: Optional[str]
- overview: Optional[str]
- # YouTube specific
- channel_id: Optional[int]
- # General
- cover_url: Optional[str]
- genres: list[str]
- def __init__(
- self,
- imdb_id: Optional[str] = "",
- youtube_id: Optional[str] = "",
- base_run_time_seconds: int = 900,
- ):
- self.imdb_id = imdb_id
- self.youtube_id = youtube_id
- self.base_run_time_seconds = base_run_time_seconds
- def as_dict_with_cover_and_genres(self) -> tuple:
- video_dict = vars(self)
- series_id = ""
- cover = None
- if "cover_url" in video_dict.keys():
- cover = video_dict.pop("cover_url", "")
- genres = video_dict.pop("genres", [])
- if "tv_series_imdb_id" in video_dict.keys():
- series_id = video_dict.pop("tv_series_imdb_id")
- return video_dict, series_id, cover, genres
|
