plexflow 0.0.64__py3-none-any.whl

plexflow/core/metadata/auto/auto_providers/tvdb/movie.py

@@ -0,0 +1,36 @@
+from plexflow.core.metadata.auto.auto_providers.auto.movie import AutoMovie
+from plexflow.core.metadata.providers.tvdb.tvdb import search_movie_by_imdb
+from datetime import datetime
+
+class AutoTvdbMovie(AutoMovie):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'tvdb')
+        self._movie = search_movie_by_imdb(self.imdb_id)
+    
+    @property
+    def id(self) -> int:
+        return self._movie.id
+    
+    @property
+    def title(self) -> str:
+        return self._movie.name
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._movie.first_release.date, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._movie.runtime
+    
+    @property
+    def titles(self) -> list:
+        return [x.name for x in self._movie.aliases]
+    
+    @property
+    def summary(self) -> str:
+        return ""
+    
+    @property
+    def language(self) -> str:
+        return self._movie.originalLanguage

plexflow/core/metadata/auto/auto_providers/tvdb/season.py

@@ -0,0 +1,25 @@
+from datetime import datetime
+from plexflow.core.metadata.auto.auto_providers.auto.season import AutoSeason
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+from plexflow.core.metadata.auto.auto_providers.auto.episode import AutoEpisode
+from plexflow.core.metadata.auto.auto_providers.tvdb.episode import AutoTvdbEpisode
+from plexflow.core.metadata.providers.tvdb.tvdb import get_season
+
+class AutoTvdbSeason(AutoSeason):
+    def __init__(self, parent: AutoShow, season_id: int) -> None:
+        super().__init__(parent)
+        self._season = get_season(season_id)
+
+    @property
+    def episodes(self) -> list[AutoEpisode]:
+        return [AutoTvdbEpisode(self, x) for x in self._season.get('episodes')]
+    
+    @property
+    def release_date(self) -> datetime:
+        episodes = self._season.get('episodes')
+        first_aired = next(filter(lambda x: x, map(lambda x: x.get('aired'), sorted(episodes, key=lambda x: x.get('number')))), None)
+        return datetime.strptime(first_aired, '%Y-%m-%d') if first_aired else None
+
+    @property
+    def season_number(self) -> int:
+        return self._season.get('number')

plexflow/core/metadata/auto/auto_providers/tvdb/show.py

@@ -0,0 +1,41 @@
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+from plexflow.core.metadata.providers.tvdb.tvdb import search_show_by_imdb
+from datetime import datetime
+from plexflow.core.metadata.auto.auto_providers.tvdb.season import AutoTvdbSeason, AutoSeason
+
+class AutoTvdbShow(AutoShow):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'tvdb')
+        self._show = search_show_by_imdb(self.imdb_id)
+    
+    @property
+    def id(self) -> int:
+        return self._show.id
+    
+    @property
+    def title(self) -> str:
+        return self._show.name
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._show.firstAired, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._show.averageRuntime
+   
+    @property
+    def titles(self) -> list:
+        return [x.name for x in self._show.aliases]
+    
+    @property
+    def summary(self) -> str:
+        return self._show.overview
+    
+    @property
+    def language(self) -> str:
+        return self._show.originalLanguage
+
+    @property
+    def seasons(self) -> list[AutoSeason]:
+        return [AutoTvdbSeason(self, x.id) for x in self._show.seasons]

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

@@ -0,0 +1,53 @@
+from datetime import datetime as dt
+from dataclasses import dataclass, field
+from dataclasses_json import dataclass_json, Undefined
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class ImdbMovie:
+    """
+    Represents a movie from IMDb.
+
+    Attributes:
+        imdb_id (str): The IMDb ID of the movie.
+        title (str): The title of the movie.
+        release_date (dt): The release date of the movie.
+        runtime (int): The runtime of the movie in minutes.
+        rating (float): The rating of the movie.
+        votes (int): The number of votes for the movie.
+        rank (int): The rank of the movie.
+    """
+    imdb_id: str
+    title: str
+    release_date: dt
+    runtime: int
+    rating: float
+    votes: int
+    rank: int
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class ImdbShow:
+    """
+    Represents a TV show from IMDb.
+
+    Attributes:
+        imdb_id (str): The IMDb ID of the show.
+        title (str): The title of the show.
+        release_date (dt): The release date of the show.
+        runtime (int): The runtime of each episode in minutes.
+        rating (float): The average rating of the show.
+        votes (int): The number of votes the show has received.
+        rank (int): The IMDb rank of the show.
+        episodes (int): The total number of episodes in the show.
+        seasons (int): The total number of seasons in the show.
+    """
+    imdb_id: str
+    title: str
+    release_date: dt
+    runtime: int
+    rating: float
+    votes: int
+    rank: int
+    episodes: int
+    seasons: int

plexflow/core/metadata/providers/imdb/imdb.py

@@ -0,0 +1,112 @@
+import requests
+from bs4 import BeautifulSoup
+from datetime import datetime as dt
+from plexflow.core.metadata.providers.imdb.datatypes import ImdbMovie, ImdbShow
+from typing import Union
+import json
+
+def search_movie_by_imdb(imdb_id: str) -> Union[ImdbMovie, None]:
+    """
+    Search for a movie using its IMDB ID.
+
+    Args:
+        imdb_id (str): The IMDB ID of the movie.
+
+    Returns:
+        Union[ImdbMovie, None]: An instance of the ImdbMovie class representing the movie if found, 
+        or None if the movie is not found.
+
+    Raises:
+        RuntimeError: If an HTTP error, connection error, timeout error, or request exception occurs.
+    """
+    url = f"https://www.imdb.com/title/{imdb_id}"
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
+    }
+
+    try:
+        r = requests.get(url=url, headers=headers)
+        r.raise_for_status()
+    except requests.exceptions.RequestException as err:
+        raise RuntimeError("An error occurred during the request.") from err
+
+    soup = BeautifulSoup(r.text, "html.parser")
+    scripts = soup.select("script[type='application/json']")
+
+    for script in scripts:
+        content = script.text
+        data = json.loads(content)
+        imdb_id = data.get("props", {}).get("pageProps", {}).get("tconst")
+        data = data.get("props", {}).get("pageProps", {}).get("aboveTheFoldData")
+
+        if data:
+            title = data.get("originalTitleText", {}).get("text")
+            release_day = data.get("releaseDate", {}).get("day")
+            release_month = data.get("releaseDate", {}).get("month")
+            release_year = data.get("releaseDate", {}).get("year")
+            date_str = f"{release_year:04d}-{release_month:02d}-{release_day:02d}"
+            release_date = dt.strptime(date_str, "%Y-%m-%d")
+
+            runtime = data.get("runtime", {}).get("seconds")
+            rating = data.get("ratingsSummary", {}).get("aggregateRating")
+            votes = data.get("ratingsSummary", {}).get("voteCount")
+            rank = data.get("meterRanking", {}).get("currentRank")
+
+            return ImdbMovie(imdb_id, title, release_date, runtime, rating, votes, rank)
+
+    return None
+
+
+def search_show_by_imdb(imdb_id: str) -> Union[ImdbShow, None]:
+    """
+    Search for a TV show on IMDb using the IMDb ID.
+
+    Args:
+        imdb_id (str): The IMDb ID of the TV show.
+
+    Returns:
+        Union[ImdbShow, None]: An instance of the ImdbShow class representing the TV show if found, 
+        or None if the TV show is not found.
+
+    Raises:
+        RuntimeError: If an HTTP error, connection error, timeout error, or request exception occurs.
+    """
+    url = f"https://www.imdb.com/title/{imdb_id}"
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
+    }
+
+    try:
+        r = requests.get(url=url, headers=headers)
+        r.raise_for_status()
+    except requests.exceptions.RequestException as err:
+        raise RuntimeError("An error occurred during the request.") from err
+
+    soup = BeautifulSoup(r.text, "html.parser")
+    scripts = soup.select("script[type='application/json']")
+
+    for script in scripts:
+        content = script.text
+        data = json.loads(content)
+        imdb_id = data.get("props", {}).get("pageProps", {}).get("tconst")
+        main_data = data.get("props", {}).get("pageProps", {}).get("mainColumnData")
+        data = data.get("props", {}).get("pageProps", {}).get("aboveTheFoldData")
+
+        if data:
+            title = data.get("originalTitleText", {}).get("text")
+            release_day = data.get("releaseDate", {}).get("day")
+            release_month = data.get("releaseDate", {}).get("month")
+            release_year = data.get("releaseDate", {}).get("year")
+            date_str = f"{release_year:04d}-{release_month:02d}-{release_day:02d}"
+            release_date = dt.strptime(date_str, "%Y-%m-%d")
+            episodes = main_data.get("episodes", {}).get("episodes", {}).get("total")
+            seasons = len(set(map(lambda s: s.get("number"), main_data.get("episodes", {}).get("seasons", []))))
+
+            runtime = data.get("runtime", {}).get("seconds")
+            rating = data.get("ratingsSummary", {}).get("aggregateRating")
+            votes = data.get("ratingsSummary", {}).get("voteCount")
+            rank = data.get("meterRanking", {}).get("currentRank")
+
+            return ImdbShow(imdb_id, title, release_date, runtime, rating, votes, rank, episodes, seasons)
+
+    return None

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

@@ -0,0 +1,111 @@
+from dataclasses import dataclass, field
+from typing import List, Optional, Dict
+from dataclasses_json import dataclass_json, Undefined
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Actor:
+    """
+    Represents an actor in a movie.
+
+    Attributes:
+        name (Optional[str]): The name of the actor.
+        voice (Optional[bool]): Indicates if the actor provided a voice-over for the movie.
+        _catchall (Dict[str, str]): A catch-all dictionary for any additional attributes.
+    """
+    name: Optional[str] = None
+    voice: Optional[bool] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Director:
+    """
+    Represents a movie director.
+
+    Attributes:
+        name (Optional[str]): The name of the director.
+        id (Optional[str]): The ID of the director.
+        _catchall (Dict[str, str]): A catch-all dictionary for any additional attributes.
+    """
+    name: Optional[str] = None
+    id: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class Posters:
+    """
+    Represents the posters of a movie.
+
+    Attributes:
+        thumb (Optional[str]): The URL of the thumbnail-sized poster.
+        small (Optional[str]): The URL of the small-sized poster.
+        regular (Optional[str]): The URL of the regular-sized poster.
+        large (Optional[str]): The URL of the large-sized poster.
+        _catchall (Dict[str, str]): A catch-all dictionary for any additional poster URLs.
+    """
+    thumb: Optional[str] = None
+    small: Optional[str] = None
+    regular: Optional[str] = None
+    large: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+@dataclass_json(undefined=Undefined.EXCLUDE)
+@dataclass
+class MovieMeterMovie:
+    """
+    Represents a movie from MovieMeter.
+
+    Attributes:
+        id (Optional[int]): The ID of the movie.
+        objectID (Optional[int]): The object ID of the movie.
+        url (Optional[str]): The URL of the movie.
+        year (Optional[int]): The year of the movie.
+        imdb (Optional[str]): The IMDb ID of the movie.
+        title (Optional[str]): The title of the movie.
+        display_title (Optional[str]): The display title of the movie.
+        alternative_title (Optional[str]): The alternative title of the movie.
+        plot (Optional[str]): The plot of the movie.
+        duration (Optional[int]): The duration of the movie.
+        votes_count (Optional[int]): The count of votes for the movie.
+        average (Optional[float]): The average rating of the movie.
+        average_x2 (Optional[float]): The average rating multiplied by 2.
+        average_x10 (Optional[float]): The average rating multiplied by 10.
+        posters (Optional[Posters]): The posters of the movie.
+        countries (Optional[List[str]]): The countries of the movie.
+        genres (Optional[List[str]]): The genres of the movie.
+        type (Optional[List[str]]): The type of the movie.
+        actors (Optional[List[Actor]]): The actors in the movie.
+        directors (Optional[List[Director]]): The directors of the movie.
+        user_vote (Optional[str]): The user's vote for the movie.
+        _catchall (Dict[str, str]): A catch-all dictionary for additional attributes.
+    """
+
+    id: Optional[int] = None
+    objectID: Optional[int] = None
+    url: Optional[str] = None
+    year: Optional[int] = None
+    imdb: Optional[str] = None
+    title: Optional[str] = None
+    display_title: Optional[str] = None
+    alternative_title: Optional[str] = None
+    plot: Optional[str] = None
+    duration: Optional[int] = None
+    votes_count: Optional[int] = None
+    average: Optional[float] = None
+    average_x2: Optional[float] = None
+    average_x10: Optional[float] = None
+    posters: Optional[Posters] = None
+    countries: Optional[List[str]] = None
+    genres: Optional[List[str]] = None
+    type: Optional[List[str]] = None
+    actors: Optional[List[Actor]] = None
+    directors: Optional[List[Director]] = None
+    user_vote: Optional[str] = None
+    _catchall: Dict[str, str] = field(default_factory=dict)
+
+    def __post_init__(self):
+        self.actors = [Actor(**actor) if isinstance(actor, dict) else actor for actor in self.actors]
+        self.directors = [Director(**director) if isinstance(director, dict) else director for director in self.directors]
+        self.posters = Posters(**self.posters) if isinstance(self.posters, dict) else self.posters

plexflow/core/metadata/providers/moviemeter/moviemeter.py

@@ -0,0 +1,42 @@
+import requests
+import os
+from typing import Union
+from plexflow.core.metadata.providers.moviemeter.datatypes import MovieMeterMovie
+
+def search_movie_by_imdb(imdb_id: str) -> Union[MovieMeterMovie, None]:
+    """
+    Search for a movie using its IMDB ID.
+
+    Parameters:
+    - imdb_id (str): The IMDB ID of the movie.
+
+    Returns:
+    - MovieMeterMovie: A MovieMeterMovie 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')
+    MovieMeterMovie(title='The Shawshank Redemption', release_date=datetime.datetime(1994, 9, 23, 0, 0), runtime=8520, rating=9.3, votes=2400000, rank=1)
+    """
+    url = f"https://www.moviemeter.nl/api/film/{imdb_id}"
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
+    }
+    params = {
+        "api_key": os.getenv("MOVIEMETER_API_KEY")
+    }
+
+    try:
+        r = requests.get(url=url, headers=headers, params=params)
+        r.raise_for_status()
+    except requests.exceptions.HTTPError as errh:
+        raise RuntimeError("HTTP Error occurred") from errh
+    except requests.exceptions.ConnectionError as errc:
+        raise RuntimeError("Connection Error occurred") from errc
+    except requests.exceptions.Timeout as errt:
+        raise RuntimeError("Timeout Error occurred") from errt
+    except requests.exceptions.RequestException as err:
+        raise RuntimeError("Request Exception occurred") from err
+
+    data = r.json()
+    return MovieMeterMovie.from_dict(data)