plexflow 0.0.64__py3-none-any.whl

plexflow/core/metadata/auto/auto_meta.py

@@ -0,0 +1,40 @@
+from plexflow.core.metadata.auto.auto_providers.auto.movie import AutoMovie
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+from plexflow.core.metadata.auto.auto_providers.tmdb.movie import AutoTmdbMovie
+from plexflow.core.metadata.auto.auto_providers.tvdb.movie import AutoTvdbMovie
+from plexflow.core.metadata.auto.auto_providers.moviemeter.movie import AutoMovieMeterMovie
+from plexflow.core.metadata.auto.auto_providers.imdb.movie import AutoImdbMovie
+from plexflow.core.metadata.auto.auto_providers.tmdb.show import AutoTmdbShow
+from plexflow.core.metadata.auto.auto_providers.tvdb.show import AutoTvdbShow
+from plexflow.core.metadata.auto.auto_providers.imdb.show import AutoImdbShow
+from plexflow.core.metadata.auto.auto_providers.plex.movie import AutoPlexMovie
+
+class AutoMeta:
+    @staticmethod
+    def movie(imdb_id: str = None, source: str = 'tmdb', rating_key: str = None) -> AutoMovie:
+        if source == 'tmdb':
+            return AutoTmdbMovie(imdb_id)
+        elif source == 'tvdb':
+            return AutoTvdbMovie(imdb_id)
+        elif source =='moviemeter':
+            return AutoMovieMeterMovie(imdb_id)
+        elif source == 'imdb':
+            return AutoImdbMovie(imdb_id)
+        elif source == "plex":
+            if isinstance(rating_key, str):
+                return AutoPlexMovie(rating_key)
+            else:
+                raise ValueError("if source = 'plex', rating_key must be set")
+        else:
+            raise ValueError(f"Invalid source: {source}")
+
+    @staticmethod
+    def show(imdb_id: str, source: str = 'tmdb') -> AutoShow:
+        if source == 'tmdb':
+            return AutoTmdbShow(imdb_id)
+        elif source == 'tvdb':
+            return AutoTvdbShow(imdb_id)
+        elif source == 'imdb':
+            return AutoImdbShow(imdb_id)
+        else:
+            raise ValueError(f"Invalid source: {source}")

plexflow/core/metadata/auto/auto_providers/auto/episode.py

@@ -0,0 +1,49 @@
+from abc import ABC, abstractmethod
+from datetime import datetime
+
+from plexflow.core.metadata.auto.auto_providers.auto.season import AutoSeason
+
+class AutoEpisode(ABC):
+    def __init__(self, parent: AutoSeason) -> None:
+        self._parent = parent
+
+    @property
+    def source(self) -> str:
+        return self._parent.source
+
+    @property
+    def season(self) -> AutoSeason:
+        return self._parent
+
+    @property
+    @abstractmethod
+    def runtime(self) -> int:
+        pass
+
+    @property
+    def year(self) -> int:
+        return self.release_date.year
+
+    @property
+    @abstractmethod
+    def episode_number(self) -> int:
+        pass
+    
+    @property
+    def season_number(self) -> int:
+        return self._parent.season_number
+    
+    @property
+    @abstractmethod
+    def summary(self) -> str:
+        pass
+    
+    @property
+    @abstractmethod
+    def title(self) -> str:
+        pass
+    
+    @property
+    @abstractmethod
+    def release_date(self) -> datetime:
+        pass

plexflow/core/metadata/auto/auto_providers/auto/item.py

@@ -0,0 +1,55 @@
+from abc import ABC, abstractmethod
+from datetime import datetime
+from typing import Any
+
+class AutoItem(ABC):
+    def __init__(self, imdb_id: str, source: str) -> None:
+        self._imdb_id = imdb_id
+        self._source = source
+    
+    @property
+    @abstractmethod
+    def id(self) -> Any:
+        pass
+    
+    @property
+    @abstractmethod
+    def title(self) -> str:
+        pass
+    
+    @property
+    @abstractmethod
+    def release_date(self) -> datetime:
+        pass
+    
+    @property
+    @abstractmethod
+    def runtime(self) -> int:
+        pass
+    
+    @property
+    def year(self) -> int:
+        return self.release_date.year
+    
+    @property
+    @abstractmethod
+    def titles(self) -> list:
+        pass
+    
+    @property
+    @abstractmethod
+    def summary(self) -> str:
+        pass
+    
+    @property
+    @abstractmethod
+    def language(self) -> str:
+        pass
+    
+    @property
+    def imdb_id(self) -> str:
+        return self._imdb_id
+    
+    @property
+    def source(self) -> str:
+        return self._source

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

@@ -0,0 +1,13 @@
+from abc import ABC, abstractmethod
+from datetime import datetime
+from plexflow.core.metadata.providers.imdb.imdb import search_movie_by_imdb
+from plexflow.core.metadata.auto.auto_providers.auto.item import AutoItem
+
+class AutoMovie(AutoItem):
+    def __init__(self, imdb_id: str, source: str) -> None:
+        super().__init__(imdb_id, source)
+        self._imdb_info = search_movie_by_imdb(self.imdb_id)
+    
+    @property
+    def rank(self) -> int:
+        return self._imdb_info.rank

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

@@ -0,0 +1,43 @@
+from typing import TYPE_CHECKING
+from abc import ABC, abstractmethod
+from datetime import datetime
+
+if TYPE_CHECKING:
+    from plexflow.core.metadata.auto.auto_providers.auto.episode import AutoEpisode
+
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+
+class AutoSeason(ABC):
+    def __init__(self, parent: AutoShow) -> None:
+        self._parent = parent
+
+    @property
+    def show(self) -> AutoShow:
+        return self._parent
+
+    @property
+    def source(self) -> str:
+        return self._parent.source
+ 
+    @property
+    def total_episodes(self) -> int:
+        return len(self.episodes)
+    
+    @property
+    @abstractmethod
+    def season_number(self) -> int:
+        pass
+    
+    @property
+    @abstractmethod
+    def episodes(self) -> list['AutoEpisode']:
+        pass
+    
+    @property
+    def year(self):
+        return self.release_date.year
+    
+    @property
+    @abstractmethod
+    def release_date(self) -> datetime:
+        pass

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

@@ -0,0 +1,26 @@
+from typing import TYPE_CHECKING
+from abc import abstractmethod
+
+if TYPE_CHECKING:
+    from plexflow.core.metadata.auto.auto_providers.auto.season import AutoSeason
+
+from plexflow.core.metadata.providers.imdb.imdb import search_show_by_imdb
+from plexflow.core.metadata.auto.auto_providers.auto.item import AutoItem
+
+class AutoShow(AutoItem):
+    def __init__(self, imdb_id: str, source: str) -> None:
+        super().__init__(imdb_id, source)
+        self._imdb_info = search_show_by_imdb(self.imdb_id)
+    
+    @property
+    def rank(self) -> int:
+        return self._imdb_info.rank
+
+    @property
+    def total_seasons(self) -> int:
+        return len(self.seasons)
+        
+    @property
+    @abstractmethod
+    def seasons(self) -> list['AutoSeason']:
+        pass

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

@@ -0,0 +1,36 @@
+from plexflow.core.metadata.auto.auto_providers.auto.movie import AutoMovie
+from plexflow.core.metadata.providers.imdb.imdb import search_movie_by_imdb
+from datetime import datetime
+
+class AutoImdbMovie(AutoMovie):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'imdb')
+        self._movie = search_movie_by_imdb(self.imdb_id)
+    
+    @property
+    def id(self) -> str:
+        return self._movie.imdb_id
+    
+    @property
+    def title(self) -> str:
+        return self._movie.title
+    
+    @property
+    def release_date(self) -> datetime:
+        return self._movie.release_date
+    
+    @property
+    def runtime(self) -> int:
+        return round(self._movie.runtime / 60)
+    
+    @property
+    def titles(self) -> list:
+        return []
+    
+    @property
+    def summary(self) -> str:
+        return None
+    
+    @property
+    def language(self) -> str:
+        return None

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

@@ -0,0 +1,45 @@
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+from plexflow.core.metadata.providers.imdb.imdb import search_show_by_imdb
+from datetime import datetime
+from plexflow.core.metadata.auto.auto_providers.tmdb.season import AutoTmdbSeason, AutoSeason
+
+class AutoImdbShow(AutoShow):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'imdb')
+        self._show = search_show_by_imdb(self.imdb_id)
+    
+    @property
+    def id(self) -> int:
+        return self._show.imdb_id
+    
+    @property
+    def title(self) -> str:
+        return self._show.title
+    
+    @property
+    def release_date(self) -> datetime:
+        return self._show.release_date
+    
+    @property
+    def runtime(self) -> int:
+        return round(self._show.runtime / 60)
+   
+    @property
+    def titles(self) -> list:
+        return []
+    
+    @property
+    def summary(self) -> str:
+        return None
+    
+    @property
+    def language(self) -> str:
+        return None
+
+    @property
+    def seasons(self) -> list[AutoSeason]:
+        return []
+
+    @property
+    def total_seasons(self) -> int:
+        return self._show.seasons

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

@@ -0,0 +1,40 @@
+from plexflow.core.metadata.auto.auto_providers.auto.movie import AutoMovie
+from plexflow.core.metadata.providers.moviemeter.moviemeter import search_movie_by_imdb
+from datetime import datetime
+
+class AutoMovieMeterMovie(AutoMovie):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'moviemeter')
+        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.display_title
+    
+    @property
+    def release_date(self) -> datetime:
+        return None
+    
+    @property
+    def runtime(self) -> int:
+        return self._movie.duration
+    
+    @property
+    def titles(self) -> list:
+        return list(filter(lambda x: x, [self._movie.title, self._movie.alternative_title]))
+    
+    @property
+    def summary(self) -> str:
+        return self._movie.plot
+
+    @property
+    def year(self) -> int:
+        return self._movie.year
+    
+    @property
+    def language(self) -> str:
+        return None

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

@@ -0,0 +1,39 @@
+from plexflow.core.metadata.auto.auto_providers.auto.movie import AutoMovie
+from plexflow.core.metadata.providers.plex.plex import search_movie_by_rating_key
+from datetime import datetime
+from plexflow.utils.imdb.imdb_codes import extract_imdb_code
+
+class AutoPlexMovie(AutoMovie):
+    def __init__(self, rating_key: str) -> None:
+        self._movie = search_movie_by_rating_key(rating_key)
+        imdb_id = next((next(extract_imdb_code(g.id), None) for g in self._movie.Guid), None)
+        
+        super().__init__(imdb_id, 'plex')
+
+    @property
+    def id(self) -> str:
+        return self._movie.ratingKey
+    
+    @property
+    def title(self) -> str:
+        return self._movie.title
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._movie.originallyAvailableAt, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._movie.duration // 60000 if isinstance(self._movie.duration, int) else None
+    
+    @property
+    def titles(self) -> list:
+        return []
+    
+    @property
+    def summary(self) -> str:
+        return self._movie.summary
+    
+    @property
+    def language(self) -> str:
+        return None

plexflow/core/metadata/auto/auto_providers/tmdb/episode.py

@@ -0,0 +1,30 @@
+from datetime import datetime
+from plexflow.core.metadata.auto.auto_providers.auto.episode import AutoEpisode
+from plexflow.core.metadata.auto.auto_providers.auto.season import AutoSeason
+from plexflow.core.metadata.providers.tmdb.tmdb import get_season_by_show_id
+from plexflow.core.metadata.providers.tmdb.datatypes import SeasonEpisode
+
+class AutoTmdbEpisode(AutoEpisode):
+    def __init__(self, parent: AutoSeason, episode: SeasonEpisode) -> None:
+        super().__init__(parent=parent)
+        self._episode = episode
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._episode.air_date, '%Y-%m-%d') if self._episode.air_date else None
+
+    @property
+    def episode_number(self) -> int:
+        return self._episode.episode_number
+    
+    @property
+    def title(self) -> str:
+        return self._episode.name
+    
+    @property
+    def runtime(self) -> int:
+        return self._episode.runtime
+    
+    @property
+    def summary(self) -> str:
+        return self._episode.overview

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

@@ -0,0 +1,36 @@
+from plexflow.core.metadata.auto.auto_providers.auto.movie import AutoMovie
+from plexflow.core.metadata.providers.tmdb.tmdb import search_movie_by_imdb, search_show_by_imdb, get_all_seasons_by_imdb_id
+from datetime import datetime
+
+class AutoTmdbMovie(AutoMovie):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'tmdb')
+        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.title
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._movie.release_date, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._movie.runtime
+    
+    @property
+    def titles(self) -> list:
+        return [x.title for x in self._movie.alternative_titles.titles]
+    
+    @property
+    def summary(self) -> str:
+        return self._movie.overview
+    
+    @property
+    def language(self) -> str:
+        return self._movie.original_language

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

@@ -0,0 +1,23 @@
+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.tmdb.episode import AutoTmdbEpisode
+from plexflow.core.metadata.providers.tmdb.tmdb import get_season_by_show_id
+
+class AutoTmdbSeason(AutoSeason):
+    def __init__(self, parent: AutoShow, season_number: int) -> None:
+        super().__init__(parent)
+        self._season = get_season_by_show_id(self._parent.id, season=season_number)
+
+    @property
+    def episodes(self) -> list[AutoEpisode]:
+        return [AutoTmdbEpisode(self, x) for x in self._season.episodes]
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._season.air_date, '%Y-%m-%d')
+
+    @property
+    def season_number(self) -> int:
+        return self._season.season_number

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

@@ -0,0 +1,41 @@
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+from plexflow.core.metadata.providers.tmdb.tmdb import search_show_by_imdb, get_all_seasons_by_imdb_id
+from datetime import datetime
+from plexflow.core.metadata.auto.auto_providers.tmdb.season import AutoTmdbSeason, AutoSeason
+
+class AutoTmdbShow(AutoShow):
+    def __init__(self, imdb_id: str) -> None:
+        super().__init__(imdb_id, 'tmdb')
+        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.original_name
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._show.first_air_date, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._show.last_episode_to_air.runtime if self._show.last_episode_to_air else next(iter(self._show.episode_run_time), None)
+   
+    @property
+    def titles(self) -> list:
+        return [x.title for x in self._show.alternative_titles.results]
+    
+    @property
+    def summary(self) -> str:
+        return self._show.overview
+    
+    @property
+    def language(self) -> str:
+        return self._show.original_language
+
+    @property
+    def seasons(self) -> list[AutoSeason]:
+        return [AutoTmdbSeason(self, x.season_number) for x in self._show.seasons]

plexflow/core/metadata/auto/auto_providers/tmdb.py

@@ -0,0 +1,92 @@
+from plexflow.core.metadata.providers.tmdb.tmdb import search_movie_by_imdb, search_show_by_imdb, get_all_seasons_by_imdb_id
+from datetime import datetime
+
+class AutoTmdbMovie:
+    """
+    Represents a movie retrieved from the TMDb (The Movie Database) API.
+
+    Attributes:
+        imdb_id (str): The IMDb ID of the movie.
+
+    Properties:
+        title (str): The title of the movie.
+        release_date (datetime): The release date of the movie.
+        runtime (int): The runtime of the movie in minutes.
+        year (int): The year of the movie's release.
+        titles (list): A list of alternative titles for the movie.
+        summary (str): The summary or overview of the movie.
+        language (str): The original language of the movie.
+    """
+
+    def __init__(self, imdb_id: str) -> None:
+        self.imdb_id = imdb_id
+        self._movie = search_movie_by_imdb(self.imdb_id)
+    
+    @property
+    def title(self) -> str:
+        return self._movie.title
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._movie.release_date, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._movie.runtime
+    
+    @property
+    def year(self) -> int:
+        return self.release_date.year
+    
+    @property
+    def titles(self) -> list:
+        return [x.title for x in self._movie.alternative_titles]
+    
+    @property
+    def summary(self) -> str:
+        return self._movie.overview
+    
+    @property
+    def language(self) -> str:
+        return self._movie.original_language
+
+class AutoTmdbShow:
+    def __init__(self, imdb_id: str) -> None:
+        self.imdb_id = imdb_id
+        self._show = search_show_by_imdb(self.imdb_id)
+    
+    @property
+    def title(self) -> str:
+        return self._show.original_name
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._show.first_air_date, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._show.episode_run_time
+    
+    @property
+    def year(self) -> int:
+        return self.release_date.year
+    
+    @property
+    def titles(self) -> list:
+        return [x.title for x in self._show.alternative_titles.results]
+    
+    @property
+    def summary(self) -> str:
+        return self._show.overview
+    
+    @property
+    def language(self) -> str:
+        return self._show.original_language
+    
+    @property
+    def total_seasons(self) -> int:
+        return self._show.number_of_seasons
+    
+    @property
+    def seasons(self) -> list:
+        return get_all_seasons_by_imdb_id(self.imdb_id)

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

@@ -0,0 +1,28 @@
+from datetime import datetime
+from plexflow.core.metadata.auto.auto_providers.auto.episode import AutoEpisode
+from plexflow.core.metadata.auto.auto_providers.auto.season import AutoSeason
+
+class AutoTvdbEpisode(AutoEpisode):
+    def __init__(self, parent: AutoSeason, episode: dict) -> None:
+        super().__init__(parent=parent)
+        self._episode = episode
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._episode.get('aired'), '%Y-%m-%d') if self._episode.get('aired') else None
+
+    @property
+    def episode_number(self) -> int:
+        return self._episode.get('number')
+    
+    @property
+    def title(self) -> str:
+        return self._episode.get('name')
+    
+    @property
+    def runtime(self) -> int:
+        return self._episode.get('runtime')
+    
+    @property
+    def summary(self) -> str:
+        return self._episode.get('overview')