plexflow 0.0.132__tar.gz → 0.0.133__tar.gz

{plexflow-0.0.132 → plexflow-0.0.133}/PKG-INFO

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

plexflow-0.0.133/plexflow/core/plex/library/folders/assets/plex_subtitle_asset.py

@@ -0,0 +1,18 @@
+from pathlib import Path
+from plexflow.core.plex.library.folders.assets.plex_asset import PlexAsset
+
+
+class PlexSubtitleAsset(PlexAsset):
+    def __init__(self, path: str, root: Path, title: str, year: int, lang: str, index: int = None, season: int = None, episode: int = None):
+        super().__init__(path, root, title, year)
+        self.lang = lang
+        self.index = index
+        self.season = season
+        self.episode = episode
+
+    @property
+    def target_path(self) -> Path:
+        if not self.season and not self.episode:
+            return self.root / Path(f"{self.title} ({self.year}){'.' + str(self.index) if self.index else ''}.{self.lang}" + self.ext)
+        else:
+            return self.root / Path(f"S{self.season:02d}E({self.episode:02d}){'.' + str(self.index) if self.index else ''}.{self.lang}" + self.ext)

plexflow-0.0.133/plexflow/core/plex/library/folders/assets/plex_video_asset.py

@@ -0,0 +1,15 @@
+from plexflow.core.plex.library.folders.assets.plex_asset import PlexAsset
+from pathlib import Path
+
+class PlexVideoAsset(PlexAsset):
+    def __init__(self, path, root, title, year, season: int = None, episode: int = None):
+        super().__init__(path, root, title, year)
+        self.season = season
+        self.episode = episode
+
+    @property
+    def target_path(self) -> Path:
+        if not self.season and not self.episode:
+            return self.root / Path(f"{self.title} ({self.year})" + self.ext)
+        else:
+            return self.root / Path(f"S{self.season:02d}E({self.episode:02d})" + self.ext)

plexflow-0.0.133/plexflow/core/plex/library/folders/plex_show_folder.py

@@ -0,0 +1,40 @@
+import shutil as sh
+import logging
+from typing import List
+from plexflow.core.plex.library.folders.plex_folder import PlexFolder
+from plexflow.core.plex.library.folders.assets.plex_asset import PlexAsset
+from plexflow.core.plex.library.folders.assets.plex_video_asset import PlexVideoAsset
+from plexflow.core.plex.library.folders.assets.plex_subtitle_asset import PlexSubtitleAsset
+
+class PlexShowFolder(PlexFolder):
+    def __init__(self, root, title, year, season: int, episode: int):
+        super().__init__(root, title, year)
+        self.assets: List[PlexAsset] = []
+        self.subtitle_index = None
+        self.season = season
+        self.episode = episode
+    
+    def add_video_path(self, path: str):
+        video_asset = PlexVideoAsset(path, root=self.path_from_root(f"Season {self.season:02d}"), title=self.title, year=self.year, season=self.season, episode=self.episode)
+        self.assets.append(video_asset)
+    
+    def add_subtitle_path(self, path: str, lang: str):
+        subtitle_asset = PlexSubtitleAsset(path, root=self.path_from_root(f"Season {self.season:02d}"), title=self.title, year=self.year, lang=lang, index=self.subtitle_index, season=self.season, episode=self.episode)
+        self.assets.append(subtitle_asset)
+        self.subtitle_index = self.subtitle_index + 1 if self.subtitle_index else 1
+    
+    def create(self, dry_run: bool = False):
+        for asset in self.assets:
+            source = asset.source_path
+            target = asset.target_path
+
+            if not dry_run:
+                target.parent.mkdir(parents=True, exist_ok=True)
+
+            logging.info(f"moving {source} to {target}")
+            
+            if not dry_run:
+                if isinstance(asset, PlexVideoAsset):
+                    sh.move(src=str(source), dst=str(target))
+                else:
+                    sh.copyfile(src=str(source), dst=str(target))

{plexflow-0.0.132 → plexflow-0.0.133}/pyproject.toml

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

plexflow-0.0.132/plexflow/core/plex/library/folders/assets/plex_subtitle_asset.py

@@ -1,13 +0,0 @@
-from pathlib import Path
-from plexflow.core.plex.library.folders.assets.plex_asset import PlexAsset
-
-
-class PlexSubtitleAsset(PlexAsset):
-    def __init__(self, path: str, root: Path, title: str, year: int, lang: str, index: int = None):
-        super().__init__(path, root, title, year)
-        self.lang = lang
-        self.index = index
-
-    @property
-    def target_path(self) -> Path:
-        return self.root / Path(f"{self.title} ({self.year}){'.' + str(self.index) if self.index else ''}.{self.lang}" + self.ext)

plexflow-0.0.132/plexflow/core/plex/library/folders/assets/plex_video_asset.py

@@ -1,5 +0,0 @@
-from plexflow.core.plex.library.folders.assets.plex_asset import PlexAsset
-
-class PlexVideoAsset(PlexAsset):
-    def __init__(self, path, root, title, year):
-        super().__init__(path, root, title, year)