plexflow 0.0.132__py3-none-any.whl → 0.0.133__py3-none-any.whl

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

@@ -3,11 +3,16 @@ 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):
+    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:
-        return self.root / Path(f"{self.title} ({self.year}){'.' + str(self.index) if self.index else ''}.{self.lang}" + self.ext)
+        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/core/plex/library/folders/assets/plex_video_asset.py

@@ -1,5 +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):
+    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/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.dist-info → plexflow-0.0.133.dist-info}/METADATA

@@ -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.132.dist-info → plexflow-0.0.133.dist-info}/RECORD

@@ -290,10 +290,11 @@ plexflow/core/plex/library/__pycache__/library.cpython-311.pyc,sha256=yL69reegO6
 plexflow/core/plex/library/__pycache__/library.cpython-312.pyc,sha256=6-TJWeb3YFArriMpzfMswXsBrlp3antN7QvUxr4LBaM,4518
 plexflow/core/plex/library/__pycache__/show.cpython-312.pyc,sha256=p0h8b6DtlcQak-HGZZeL31ToMsifkbNqQtbfDBT4tVM,3656
 plexflow/core/plex/library/folders/assets/plex_asset.py,sha256=awY8vhg3KoU2uCB18y5MleE3aAUdVDTnN1OaTmm2oJc,530
-plexflow/core/plex/library/folders/assets/plex_subtitle_asset.py,sha256=51itoIUbHco2HW0Mh9mGNddqwMrZA2ajwuyhVzPcXkI,521
-plexflow/core/plex/library/folders/assets/plex_video_asset.py,sha256=F_7RtdtblYQJc_gp9nUVxIfS4BeHl0eZDgYmTMg2zos,208
+plexflow/core/plex/library/folders/assets/plex_subtitle_asset.py,sha256=nbbpDo8dbNkysDmVqbY8G4Jcc_lbo72oSXgj0KNeIxc,837
+plexflow/core/plex/library/folders/assets/plex_video_asset.py,sha256=8gM44WGehvTrPPh3HMcNmE4HkZ9eNLJByqfer2HOoBo,616
 plexflow/core/plex/library/folders/plex_folder.py,sha256=GZnyQO6bzcire-2mxPjIPJrq8J0S_4uVsOfVCiHo2E0,387
 plexflow/core/plex/library/folders/plex_movie_folder.py,sha256=ns-3yf5VgikXc5IBeN8xPTtB9LHkPIQOZ4CsvcBZNU8,1672
+plexflow/core/plex/library/folders/plex_show_folder.py,sha256=rPBNxBuVt4cfappqTbaHaa6faGDW_Ho1Tw6xjLXv6wU,1892
 plexflow/core/plex/library/library.py,sha256=taLi0e7iY0usZPDHXwPfLmxQI5vyMyo8eJtOSHNKtww,3644
 plexflow/core/plex/library/show.py,sha256=jAZfrjxoF_E1v0sdTpLmyx3NGWC3Uzil0XjJ8NkbUvQ,2648
 plexflow/core/plex/token/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -714,7 +715,7 @@ plexflow/utils/video/__pycache__/audio.cpython-312.pyc,sha256=kmzGDCHSC1hWyHwRut
 plexflow/utils/video/__pycache__/subtitle.cpython-312.pyc,sha256=PCjpCLydGXaRsQy6cikhgsEs8WlComfOoYPiLFqfVMA,2515
 plexflow/utils/video/audio.py,sha256=Pd8OuQHX2QN-lc5iYkB0Vo1OEHmTcvDYH-uKud1f1q4,5262
 plexflow/utils/video/subtitle.py,sha256=qPvvBjlPj0fynJJvGJgGeKt9ey26R-cF6EoLaYt9iXU,1333
-plexflow-0.0.132.dist-info/METADATA,sha256=tMNMebSsdsldEkn03tNHttSwAt8ewZ-c4wXHB_At2Ik,2971
-plexflow-0.0.132.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.132.dist-info/entry_points.txt,sha256=9RJC3ikOQORHNOn573EdwJOBUnFU_4EGHbtNUM5pjjY,1557
-plexflow-0.0.132.dist-info/RECORD,,
+plexflow-0.0.133.dist-info/METADATA,sha256=i5Oj1D2JVpMhfeA7f5vv0EJ1cIujuiYpq1xY8prGbVw,2971
+plexflow-0.0.133.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.133.dist-info/entry_points.txt,sha256=9RJC3ikOQORHNOn573EdwJOBUnFU_4EGHbtNUM5pjjY,1557
+plexflow-0.0.133.dist-info/RECORD,,