plexflow 0.0.122__tar.gz → 0.0.123__tar.gz

{plexflow-0.0.122 → plexflow-0.0.123}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: plexflow
-Version: 0.0.122
+Version: 0.0.123
 Summary: A short description of the package.
 License: MIT
 Keywords: keyword1,keyword2,keyword3

plexflow-0.0.123/plexflow/core/context/partials/show.py

@@ -0,0 +1,107 @@
+from plexflow.core.context.partial_context import PartialContext
+from datetime import datetime as dt
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+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.show import AutoPlexShow
+
+class Show(PartialContext):
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+
+    @property
+    def sources(self) -> list:
+        keys = self.get_keys("show/*")
+        # extract the source from the key
+        return [key.split("/")[-1] for key in keys]
+
+    def from_source(self, source: str) -> AutoShow:
+        return self.get(f"show/{source}")
+
+    @property
+    def title(self) -> str:
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.title:
+                return details.title
+    
+    @property
+    def year(self) -> int:
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.year:
+                return details.year
+    
+    @property
+    def release_date(self) -> dt:
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.release_date:
+                return details.release_date
+    
+    @property
+    def rank(self) -> int:
+        return self.plex.rank
+    
+    @property
+    def released(self) -> bool:
+        dates = []
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.release_date:
+                dates.append(details.release_date)
+        
+        sorted_dates = sorted(dates)
+        now = dt.now()
+        return all([date < now for date in sorted_dates])
+    
+    @property
+    def runtime(self) -> int:
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.runtime:
+                return details.runtime
+
+    @property
+    def titles(self) -> set:
+        titles = set()
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.title:
+                titles.add(details.title)
+                titles.update(details.titles)
+        return titles
+
+    @property
+    def summary(self) -> str:
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.summary:
+                return details.summary
+    
+    @property
+    def language(self) -> str:
+        for source in self.sources:
+            details = self.from_source(source)
+            if details and details.language:
+                return details.language
+
+    @property
+    def plex(self) -> AutoPlexShow:
+        return self.from_source("plex")
+        
+    @property
+    def tmdb(self) -> AutoTmdbShow:
+        return self.from_source("tmdb")
+        
+    @property
+    def imdb(self) -> AutoImdbShow:
+        return self.from_source("imdb")
+        
+    @property
+    def tvdb(self) -> AutoTvdbShow:
+        return self.from_source("tvdb")
+    
+    def update(self, movie: AutoShow):
+        self.set(f"show/{movie.source}", movie)

{plexflow-0.0.122 → plexflow-0.0.123}/plexflow/core/metadata/auto/auto_meta.py

@@ -7,6 +7,7 @@ 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.show import AutoPlexShow
 from plexflow.core.metadata.auto.auto_providers.plex.movie import AutoPlexMovie
 
 class AutoMeta:
@@ -29,12 +30,14 @@ class AutoMeta:
             raise ValueError(f"Invalid source: {source}")
 
     @staticmethod
-    def show(imdb_id: str, source: str = 'tmdb') -> AutoShow:
+    def show(imdb_id: str, source: str = 'tmdb', rating_key: str = None) -> AutoShow:
         if source == 'tmdb':
             return AutoTmdbShow(imdb_id)
         elif source == 'tvdb':
             return AutoTvdbShow(imdb_id)
         elif source == 'imdb':
             return AutoImdbShow(imdb_id)
+        elif source == 'plex':
+            return AutoPlexShow(rating_key=rating_key)
         else:
             raise ValueError(f"Invalid source: {source}")

plexflow-0.0.123/plexflow/core/metadata/auto/auto_providers/plex/episode.py

@@ -0,0 +1,29 @@
+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.plex.datatypes import PlexEpisodeMetadata
+
+class AutoPlexEpisode(AutoEpisode):
+    def __init__(self, parent: AutoSeason, data: PlexEpisodeMetadata) -> None:
+        super().__init__(parent=parent)
+        self._episode = data
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._episode.originallyAvailableAt, '%Y-%m-%d') if self._episode.originallyAvailableAt else None
+
+    @property
+    def episode_number(self) -> int:
+        return self._episode.index
+    
+    @property
+    def title(self) -> str:
+        return self._episode.title
+    
+    @property
+    def runtime(self) -> int:
+        return self._episode.duration
+    
+    @property
+    def summary(self) -> str:
+        return self._episode.summary

plexflow-0.0.123/plexflow/core/metadata/auto/auto_providers/plex/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.providers.plex.datatypes import PlexSeasonMetadata
+from plexflow.core.metadata.providers.plex.plex import search_episodes_by_season_rating_key
+from plexflow.core.metadata.auto.auto_providers.plex.episode import AutoPlexEpisode
+
+class AutoPlexSeason(AutoSeason):
+    def __init__(self, parent: AutoShow, data: PlexSeasonMetadata) -> None:
+        super().__init__(parent)
+        self._season = data
+
+    @property
+    def episodes(self) -> list[AutoEpisode]:
+        episodes = search_episodes_by_season_rating_key(key=self._season.ratingKey)
+        return [AutoPlexEpisode(self, episode) for episode in episodes]
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._season.originallyAvailableAt, '%Y-%m-%d')
+
+    @property
+    def season_number(self) -> int:
+        return self._season.index

plexflow-0.0.123/plexflow/core/metadata/auto/auto_providers/plex/show.py

@@ -0,0 +1,46 @@
+from plexflow.core.metadata.auto.auto_providers.auto.show import AutoShow
+from plexflow.core.metadata.auto.auto_providers.auto.season import AutoSeason
+from plexflow.core.metadata.providers.plex.plex import search_show_by_rating_key, search_seasons_by_show_rating_key
+from datetime import datetime
+from plexflow.utils.imdb.imdb_codes import extract_imdb_code
+from plexflow.core.metadata.auto.auto_providers.plex.season import AutoPlexSeason
+
+class AutoPlexShow(AutoShow):
+    def __init__(self, rating_key: str) -> None:
+        self._show = search_show_by_rating_key(rating_key)
+        imdb_id = next((next(extract_imdb_code(g.id), None) for g in self._show.Guid), None)
+        
+        super().__init__(imdb_id, 'plex')
+
+    @property
+    def id(self) -> str:
+        return self._show.ratingKey
+    
+    @property
+    def title(self) -> str:
+        return self._show.title
+    
+    @property
+    def release_date(self) -> datetime:
+        return datetime.strptime(self._show.originallyAvailableAt, '%Y-%m-%d')
+    
+    @property
+    def runtime(self) -> int:
+        return self._show.duration // 60000 if isinstance(self._show.duration, int) else None
+    
+    @property
+    def titles(self) -> list:
+        return []
+    
+    @property
+    def summary(self) -> str:
+        return self._show.summary
+    
+    @property
+    def language(self) -> str:
+        return None
+
+    @property
+    def seasons(self) -> list[AutoSeason]:
+        seasons = search_seasons_by_show_rating_key(key=self.id)
+        return [AutoPlexSeason(self, season) for season in seasons]

{plexflow-0.0.122 → plexflow-0.0.123}/plexflow/core/metadata/providers/plex/datatypes.py

@@ -368,17 +368,57 @@ class PlexShowMetadata:
         _catchall (dict): A catch-all dictionary for any additional attributes.
     """
 
+    art: Optional[str] = None,
+    guid: Optional[str] = None,
+    key: Optional[str] = None,
+    primaryExtraKey: Optional[str] = None,
+    rating: Optional[float] = None,
+    ratingKey: Optional[str] = None,
+    studio: Optional[str] = None,
+    subtype: Optional[str] = None,
+    summary: Optional[str] = None,
+    tagline: Optional[str] = None,
+    type: Optional[str] = None,
+    thumb: Optional[str] = None,
+    addedAt: Optional[int] = None,
+    duration: Optional[int] = None,
+    publicPagesURL: Optional[str] = None,
+    slug: Optional[str] = None,
+    userState: Optional[bool] = None,
+    title: Optional[str] = None,
+    leafCount: Optional[int] = None,
+    childCount: Optional[int] = None,
+    skipChildren: Optional[bool] = None,
+    isContinuingSeries: Optional[bool] = None,
+    contentRating: Optional[str] = None,
+    originallyAvailableAt: Optional[str] = None,
+    year: Optional[int] = None,
+    ratingImage: Optional[str] = None,
+    imdbRatingCount: Optional[int] = None,
+    source: Optional[str] = None
+    Image: Optional[List[PlexImage]] = None
+    Guid: Optional[List[PlexGuid]] = None
+    Role: Optional[List[PlexRole]] = None
+    Country: Optional[List[PlexCountry]] = None
+    Director: Optional[List[PlexDirector]] = None
+    Writer: Optional[List[PlexWriter]] = None
+    Rating: Optional[List[PlexRating]] = None
+    Studio: Optional[List[PlexStudio]] = None
+    Producer: Optional[List[PlexProducer]] = None
+    Genre: Optional[List[PlexGenre]] = None
+    _catchall: dict = field(default_factory=dict)
+    
     def __post_init__(self):
-        self.Image = [PlexImage(**image) if isinstance(image, dict) else image for image in self.Image]
-        self.Genre = [PlexGenre(**genre) if isinstance(genre, dict) else genre for genre in self.Genre]
-        self.Guid = [PlexGuid(**guid) if isinstance(guid, dict) else guid for guid in self.Guid]
-        self.Country = [PlexCountry(**country) if isinstance(country, dict) else country for country in self.Country]
-        self.Role = [PlexRole(**role) if isinstance(role, dict) else role for role in self.Role]
-        self.Director = [PlexDirector(**director) if isinstance(director, dict) else director for director in self.Director]
-        self.Producer = [PlexProducer(**producer) if isinstance(producer, dict) else producer for producer in self.Producer]
-        self.Writer = [PlexWriter(**writer) if isinstance(writer, dict) else writer for writer in self.Writer]
-        self.Rating = [PlexRating(**rating) if isinstance(rating, dict) else rating for rating in self.Rating]
-        self.Studio = [PlexStudio(**studio) if isinstance(studio, dict) else studio for studio in self.Studio]
+        self.Image = [PlexImage(**image) if isinstance(image, dict) else image for image in self.Image] if self.Image else []
+        self.Genre = [PlexGenre(**genre) if isinstance(genre, dict) else genre for genre in self.Genre] if self.Genre else []
+        self.Guid = [PlexGuid(**guid) if isinstance(guid, dict) else guid for guid in self.Guid] if self.Guid else []
+        self.Country = [PlexCountry(**country) if isinstance(country, dict) else country for country in self.Country] if self.Country else []
+        self.Role = [PlexRole(**role) if isinstance(role, dict) else role for role in self.Role] if self.Role else []
+        self.Director = [PlexDirector(**director) if isinstance(director, dict) else director for director in self.Director] if self.Director else []
+        self.Producer = [PlexProducer(**producer) if isinstance(producer, dict) else producer for producer in self.Producer] if self.Producer else []
+        self.Writer = [PlexWriter(**writer) if isinstance(writer, dict) else writer for writer in self.Writer] if self.Writer else []
+        self.Rating = [PlexRating(**rating) if isinstance(rating, dict) else rating for rating in self.Rating] if self.Rating else []
+        self.Studio = [PlexStudio(**studio) if isinstance(studio, dict) else studio for studio in self.Studio] if self.Studio else []
 
 
 @dataclass_json(undefined=Undefined.EXCLUDE)
@@ -614,7 +654,7 @@ class PlexEpisodeMetadata:
     _catchall: dict = field(default_factory=dict)
 
     def __post_init__(self):
-        self.Image = [PlexImage(**image) if isinstance(image, dict) else image for image in self.Image]
+        self.Image = [PlexImage(**image) if isinstance(image, dict) else image for image in self.Image] if self.Image else []
         self.Guid = [PlexGuid(**guid) if isinstance(guid, dict) else guid for guid in self.Guid]
         self.Role = [PlexRole(**role) if isinstance(role, dict) else role for role in self.Role] if self.Role else []
         self.Director = [PlexDirector(**director) if isinstance(director, dict) else director for director in self.Director] if self.Director else []

{plexflow-0.0.122 → plexflow-0.0.123}/plexflow/utils/video/audio.py

@@ -2,6 +2,9 @@ import subprocess
 from pathlib import Path
 from typing import List, Optional
 import re
+import json
+from typing import Tuple
+
 
 class FFmpegError(Exception):
     """Custom exception for FFmpeg errors."""
@@ -74,6 +77,9 @@ def get_audio_stream_indices(video_path: Path) -> List[int]:
     - FileNotFoundError: If the video file does not exist.
     - FFmpegError: If ffmpeg encounters an error during processing.
     """
+    if isinstance(video_path, str):
+        video_path = Path(video_path)
+
     if not video_path.exists():
         raise FileNotFoundError(f"The video file {video_path} does not exist.")
     
@@ -95,4 +101,48 @@ def get_audio_stream_indices(video_path: Path) -> List[int]:
                 stream_index = int(match.group(1))
                 audio_indices.append(stream_index)
     
-    return audio_indices
+    return audio_indices
+
+def get_audio_stream_info(video_path: Path) -> List[Tuple[int, Optional[str]]]:
+    """
+    Retrieves the indices and language tags of audio streams in a video file using ffprobe.
+
+    Parameters:
+    - video_path (Path): Path to the input video file.
+
+    Returns:
+    - List[Tuple[int, Optional[str]]]: A list of tuples, where each tuple contains
+      (stream_index, language_code). language_code will be None if not found.
+
+    Raises:
+    - FileNotFoundError: If the video file does not exist.
+    - FFmpegError: If ffprobe encounters an error during processing or JSON parsing.
+    """
+    if isinstance(video_path, str):
+        video_path = Path(video_path)
+        
+    if not video_path.exists():
+        raise FileNotFoundError(f"The video file {video_path} does not exist.")
+    
+    command = [
+        'ffprobe',
+        '-v', 'error',                 # Suppress verbose output
+        '-select_streams', 'a',        # Select only audio streams
+        '-show_entries', 'stream=index:stream_tags=language', # Show index and language tag
+        '-of', 'json',                 # Output in JSON format
+        str(video_path)
+    ]
+    
+    try:
+        result = subprocess.run(command, check=True, capture_output=True, text=True)
+        data = json.loads(result.stdout)
+        
+        audio_stream_info = []
+        for s in data.get('streams', []):
+            index = s.get('index')
+            language = s.get('tags', {}).get('language')
+            if index is not None:
+                audio_stream_info.append((index, language))
+        return audio_stream_info
+    except (subprocess.CalledProcessError, json.JSONDecodeError) as e:
+        raise FFmpegError(f"FFprobe error: {e.stderr if isinstance(e, subprocess.CalledProcessError) else e}")

{plexflow-0.0.122 → plexflow-0.0.123}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "plexflow"
-version = "0.0.122"
+version = "0.0.123"
 description = "A short description of the package."
 authors = ["Your Name <you@example.com>"]
 license = "MIT"