plexflow 0.0.64__py3-none-any.whl

plexflow/core/metadata/providers/tmdb/tmdb.py

@@ -0,0 +1,85 @@
+import os
+import tmdbsimple
+from tmdbsimple import Find, Movies, TV, TV_Seasons
+from typing import Union, List
+from plexflow.core.metadata.providers.tmdb.datatypes import TmdbMovie, TmdbShow, movie_from_json, show_from_json, TmdbSeason
+import json
+
+def search_movie_by_imdb(imdb_id: str) -> Union[TmdbMovie, None]:
+    """
+    Search for a movie using its IMDB ID.
+
+    Parameters:
+    imdb_id (str): The IMDB ID of the movie.
+
+    Returns:
+    Movie: A Movie object containing all the details of the movie.
+    None: If the movie is not found or an error occurs.
+
+    Example:
+    >>> search_movie_by_imdb('tt0111161')
+    Movie(id=278, title='The Shawshank Redemption', overview='Framed in the 1940s for the double murder of his wife and her lover, upstanding banker Andy Dufresne begins a new life at the Shawshank prison, where he puts his accounting skills to work for an amoral warden. During his long stretch in prison, Dufresne comes to be admired by the other inmates -- including an older prisoner named Red -- for his integrity and unquenchable sense of hope.', release_date='1994-09-23', ...)
+    """
+    tmdbsimple.API_KEY = os.getenv("TMDB_API_KEY")
+
+    try:
+        find = Find(imdb_id)
+        response = find.info(external_source='imdb_id')
+        if response['movie_results']:
+            movie_id = response['movie_results'][0]['id']
+            movie = Movies(movie_id)
+            info = movie.info(append_to_response="alternative_titles")
+            return movie_from_json(json.dumps(info))
+        else:
+            raise ValueError(f"No movie found with IMDB ID {imdb_id}")
+    except Exception as e:
+        raise e
+
+def search_show_by_imdb(imdb_id: str) -> Union[TmdbShow, None]:
+    """
+    Search for a TV show using its IMDB ID.
+
+    Parameters:
+    imdb_id (str): The IMDB ID of the TV show.
+
+    Returns:
+    Show: A Show object containing all the details of the TV show.
+    None: If the TV show is not found or an error occurs.
+
+    Example:
+    >>> search_show_by_imdb('tt0944947')
+    Show(id=1399, name='Game of Thrones', overview='Seven noble families fight for control of the mythical land of Westeros. Friction between the houses leads to full-scale war. All while a very ancient evil awakens in the farthest north. Amidst the war, a neglected military order of misfits, the Night\'s Watch, is all that stands between the realms of men and icy horrors beyond.', first_air_date='2011-04-17', ...)
+    """
+    tmdbsimple.API_KEY = os.getenv("TMDB_API_KEY")
+
+    try:
+        find = Find(imdb_id)
+        response = find.info(external_source='imdb_id')
+        if response['tv_results']:
+            show_id = response['tv_results'][0]['id']
+            show = TV(show_id)
+            info = show.info(append_to_response="alternative_titles,external_ids")
+            return show_from_json(json.dumps(info))
+        else:
+            raise ValueError(f"No TV show found with IMDB ID {imdb_id}")
+    except Exception as e:
+        raise e
+
+
+def get_season_by_show_id(tmdb_id: int, season: int) -> TmdbSeason:
+    tmdbsimple.API_KEY = os.getenv("TMDB_API_KEY")
+
+    try:
+        tv = TV_Seasons(tv_id=tmdb_id, season_number=season)
+        response = tv.info()
+        return TmdbSeason.from_dict(response)
+    except Exception as e:
+        raise e
+
+def get_all_seasons_by_imdb_id(imdb_id: str) -> List[TmdbSeason]:
+    show = search_show_by_imdb(imdb_id=imdb_id)
+    results = []
+    for season in show.seasons:
+        season_info = get_season_by_show_id(tmdb_id=show.id, season=season.season_number)
+        results.append(season_info)
+    return results

plexflow/core/metadata/providers/tvdb/datatypes.py

@@ -0,0 +1,257 @@
+from dataclasses import dataclass, field
+from typing import List, Optional, Dict
+from dataclasses_json import dataclass_json, Undefined
+import json
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Alias:
+    language: Optional[str] = None
+    name: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Artwork:
+    height: Optional[int] = None
+    id: Optional[int] = None
+    image: Optional[str] = None
+    includesText: Optional[bool] = None
+    language: Optional[str] = None
+    score: Optional[int] = None
+    thumbnail: Optional[str] = None
+    type: Optional[int] = None
+    width: Optional[int] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Award:
+    id: Optional[int] = None
+    name: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Character:
+    aliases: Optional[List[Alias]] = None
+    episode: Optional[dict] = None
+    episodeId: Optional[int] = None
+    id: Optional[int] = None
+    image: Optional[str] = None
+    isFeatured: Optional[bool] = None
+    movieId: Optional[int] = None
+    movie: Optional[dict] = None
+    name: Optional[str] = None
+    nameTranslations: Optional[List[str]] = None
+    overviewTranslations: Optional[List[str]] = None
+    peopleId: Optional[int] = None
+    personImgURL: Optional[str] = None
+    peopleType: Optional[str] = None
+    seriesId: Optional[int] = None
+    series: Optional[dict] = None
+    sort: Optional[int] = None
+    tagOptions: Optional[List[dict]] = None
+    type: Optional[int] = None
+    url: Optional[str] = None
+    personName: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Company:
+    activeDate: Optional[str] = None
+    aliases: Optional[List[Alias]] = None
+    country: Optional[str] = None
+    id: Optional[int] = None
+    inactiveDate: Optional[str] = None
+    name: Optional[str] = None
+    nameTranslations: Optional[List[str]] = None
+    overviewTranslations: Optional[List[str]] = None
+    primaryCompanyType: Optional[int] = None
+    slug: Optional[str] = None
+    parentCompany: Optional[dict] = None
+    tagOptions: Optional[List[dict]] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class ContentRating:
+    id: Optional[int] = None
+    name: Optional[str] = None
+    description: Optional[str] = None
+    country: Optional[str] = None
+    contentType: Optional[str] = None
+    order: Optional[int] = None
+    fullName: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class FirstRelease:
+    country: Optional[str] = None
+    date: Optional[str] = None
+    detail: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Genre:
+    id: Optional[int] = None
+    name: Optional[str] = None
+    slug: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Inspiration:
+    id: Optional[int] = None
+    type: Optional[str] = None
+    type_name: Optional[str] = None
+    url: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class ListData:
+    aliases: Optional[List[Alias]] = None
+    id: Optional[int] = None
+    image: Optional[str] = None
+    imageIsFallback: Optional[bool] = None
+    isOfficial: Optional[bool] = None
+    name: Optional[str] = None
+    nameTranslations: Optional[List[str]] = None
+    overview: Optional[str] = None
+    overviewTranslations: Optional[List[str]] = None
+    remoteIds: Optional[List[dict]] = None
+    tags: Optional[List[dict]] = None
+    score: Optional[int] = None
+    url: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class ProductionCountry:
+    id: Optional[int] = None
+    country: Optional[str] = None
+    name: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Release:
+    country: Optional[str] = None
+    date: Optional[str] = None
+    detail: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class RemoteId:
+    id: Optional[str] = None
+    type: Optional[int] = None
+    sourceName: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Status:
+    id: Optional[int] = None
+    keepUpdated: Optional[bool] = None
+    name: Optional[str] = None
+    recordType: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Studio:
+    id: Optional[int] = None
+    name: Optional[str] = None
+    parentStudio: Optional[int] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class TagOption:
+    helpText: Optional[str] = None
+    id: Optional[int] = None
+    name: Optional[str] = None
+    tag: Optional[int] = None
+    tagName: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Trailer:
+    id: Optional[int] = None
+    language: Optional[str] = None
+    name: Optional[str] = None
+    url: Optional[str] = None
+    runtime: Optional[int] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Translation:
+    aliases: Optional[List[str]] = None
+    isAlias: Optional[bool] = None
+    isPrimary: Optional[bool] = None
+    language: Optional[str] = None
+    name: Optional[str] = None
+    overview: Optional[str] = None
+    tagline: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class SubtitleLanguage:
+    language: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class TvdbMovie:
+    aliases: Optional[List[Alias]] = None
+    artworks: Optional[List[Artwork]] = None
+    audioLanguages: Optional[List[str]] = None
+    awards: Optional[List[Award]] = None
+    boxOffice: Optional[str] = None
+    boxOfficeUS: Optional[str] = None
+    budget: Optional[str] = None
+    characters: Optional[List[Character]] = None
+    companies: Optional[dict] = None
+    contentRatings: Optional[List[ContentRating]] = None
+    first_release: Optional[FirstRelease] = None
+    genres: Optional[List[Genre]] = None
+    id: Optional[int] = None
+    image: Optional[str] = None
+    inspirations: Optional[List[Inspiration]] = None
+    lastUpdated: Optional[str] = None
+    lists: Optional[List[ListData]] = None
+    name: Optional[str] = None
+    nameTranslations: Optional[List[str]] = None
+    originalCountry: Optional[str] = None
+    originalLanguage: Optional[str] = None
+    overviewTranslations: Optional[List[str]] = None
+    production_countries: Optional[List[ProductionCountry]] = None
+    releases: Optional[List[Release]] = None
+    remoteIds: Optional[List[RemoteId]] = None
+    runtime: Optional[int] = None
+    score: Optional[int] = None
+    slug: Optional[str] = None
+    spoken_languages: Optional[List[str]] = None
+    status: Optional[Status] = None
+    studios: Optional[List[Studio]] = None
+    subtitleLanguages: Optional[List[SubtitleLanguage]] = None
+    tagOptions: Optional[List[TagOption]] = None
+    trailers: Optional[List[Trailer]] = None
+    translations: Optional[dict] = None
+    year: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+
+def movie_from_json(json_str: str) -> TvdbMovie:
+    return TvdbMovie.from_json(json_str)
+
+def show_from_json():
+    pass