plexflow 0.0.109__py3-none-any.whl → 0.0.111__py3-none-any.whl

plexflow/core/context/partials/movie_assets.py

@@ -1,6 +1,8 @@
 from plexflow.core.context.partial_context import PartialContext
 from typing import List, Tuple
 from qbittorrentapi.torrents import TorrentDictionary
+from plexflow.core.subtitles.utils.plex_internal_subtitle import PlexInternalSubtitle
+from plexflow.core.subtitles.utils.plex_external_subtitle import PlexExternalSubtitle
 
 class MovieAssets(PartialContext):
     def __init__(self, **kwargs) -> None:
@@ -12,6 +14,12 @@ class MovieAssets(PartialContext):
     def movie_path(self) -> str:
         return self.get("assets/movie/path")
 
+    def internal_subtitles(self) -> List[PlexInternalSubtitle]:
+        return self.get("assets/subtitles/internal")
+    
+    def external_subtitles(self) -> List[PlexExternalSubtitle]:
+        return self.get("assets/subtitles/external")
+
     def subtitle_paths(self) -> List[str]:
         return self.get("assets/subtitle/path")
 
@@ -27,6 +35,12 @@ class MovieAssets(PartialContext):
     def update_subtitle_paths(self, paths: List[str]):
         self.set("assets/subtitle/path", paths)
 
+    def update_internal_subtitles(self, subtitles: List[PlexInternalSubtitle]):
+        self.set("assets/subtitles/internal", subtitles)
+
+    def update_external_subtitles(self, subtitles: List[PlexExternalSubtitle]):
+        self.set("assets/subtitles/external", subtitles)
+
     def update_embedded_subtitles(self, subtitles: List[Tuple[int, str]]):
         self.set("assets/embedded_subtitles", subtitles)
 

plexflow/core/context/partials/subtitles.py

@@ -1,16 +1,15 @@
 from plexflow.core.context.partial_context import PartialContext
-from datetime import datetime as dt
-from plexflow.core.subtitles.providers.oss.oss_subtitle import OSSSubtitle
+from plexflow.core.subtitles.utils.plex_external_subtitle import PlexExternalSubtitle
 from typing import List
 
 class Subtitles(PartialContext):
     def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)
 
-    def all(self) -> List[OSSSubtitle]:
+    def all(self) -> List[PlexExternalSubtitle]:
         return self.get("subtitles/oss")
 
-    def update(self, subtitles: List[OSSSubtitle]):
+    def update(self, subtitles: List[PlexExternalSubtitle]):
         if len(subtitles) == 0:
             return
         self.set("subtitles/oss", subtitles)

plexflow/core/downloads/candidates/download_candidate.py

@@ -1,7 +1,7 @@
 import math
 from typing import List, Optional, Set
 from plexflow.core.torrents.results.universal import UniversalTorrent
-from plexflow.core.subtitles.results.subtitle import Subtitle
+from plexflow.core.subtitles.utils.plex_external_subtitle import PlexExternalSubtitle
 from plexflow.utils.imdb.imdb_codes import IMDbCode
 
 class DownloadCandidate:
@@ -9,7 +9,7 @@ class DownloadCandidate:
     Represents a download candidate for a torrent with associated subtitles.
     """
 
-    def __init__(self, torrent: UniversalTorrent, subtitles: List[Subtitle]):
+    def __init__(self, torrent: UniversalTorrent, subtitles: List[PlexExternalSubtitle]):
         """
         Initializes a new instance of the DownloadCandidate class.
 
@@ -197,7 +197,7 @@ class DownloadCandidate:
         """
         return any([
             self.has_native_dutch_subtitles,
-            any(s.language == "nl" for s in self.subtitles),
+            any(s.oss_subtitle.language == "nl" for s in self.subtitles),
         ])
 
     @property
@@ -210,5 +210,5 @@ class DownloadCandidate:
         """
         return any([
             self.has_native_english_subtitles,
-            any(s.language == "en" for s in self.subtitles),
+            any(s.oss_subtitle.language == "en" for s in self.subtitles),
         ])

plexflow/core/downloads/candidates/utils.py

@@ -1,16 +1,16 @@
 from typing import Iterable, List
 from plexflow.core.torrents.results.universal import UniversalTorrent
-from plexflow.core.subtitles.results.subtitle import Subtitle
+from plexflow.core.subtitles.utils.plex_external_subtitle import PlexExternalSubtitle
 from plexflow.core.downloads.candidates.download_candidate import DownloadCandidate
 
-def create_download_candidates(torrents: Iterable[UniversalTorrent], subtitles: Iterable[Subtitle]) -> List[DownloadCandidate]:
+def create_download_candidates(torrents: Iterable[UniversalTorrent], subtitles: Iterable[PlexExternalSubtitle]) -> List[DownloadCandidate]:
     """
-    This function creates a list of DownloadCandidate objects from an iterable of UniversalTorrent and an iterable of Subtitle.
+    This function creates a list of DownloadCandidate objects from an iterable of UniversalTorrent and an iterable of PlexExternalSubtitle.
     All subtitles that are compatible with a torrent will become together a DownloadCandidate.
 
     Parameters:
     torrents (Iterable[UniversalTorrent]): An iterable of UniversalTorrent objects.
-    subtitles (Iterable[Subtitle]): An iterable of Subtitle objects.
+    subtitles (Iterable[PlexExternalSubtitle]): An iterable of PlexExternalSubtitle objects.
 
     Returns:
     List[DownloadCandidate]: A list of DownloadCandidate objects.
@@ -23,7 +23,7 @@ def create_download_candidates(torrents: Iterable[UniversalTorrent], subtitles:
     if not isinstance(torrents, Iterable):
         raise ValueError("torrents should be an iterable of UniversalTorrent objects")
     if not isinstance(subtitles, Iterable):
-        raise ValueError("subtitles should be an iterable of Subtitle objects")
+        raise ValueError("subtitles should be an iterable of PlexExternalSubtitle objects")
 
     download_candidates = []
 
@@ -32,7 +32,7 @@ def create_download_candidates(torrents: Iterable[UniversalTorrent], subtitles:
         if not isinstance(torrent, UniversalTorrent):
             raise ValueError("Each torrent should be an instance of UniversalTorrent")
 
-        compatible_subtitles = [subtitle for subtitle in subtitles if torrent.is_compatible_with(subtitle)]
+        compatible_subtitles = [subtitle for subtitle in subtitles if torrent.is_compatible_with(subtitle.oss_subtitle)]
         download_candidate = DownloadCandidate(torrent, compatible_subtitles)
         download_candidates.append(download_candidate)
 

plexflow/core/subtitles/utils/plex_external_subtitle.py

@@ -0,0 +1,11 @@
+from plexflow.core.subtitles.providers.oss.oss_subtitle import OSSSubtitle
+from plexflow.core.subtitles.utils.plex_subtitle import PlexSubtitle
+
+class PlexExternalSubtitle(PlexSubtitle):
+    def __init__(self, path, name, subtitle: OSSSubtitle):
+        super().__init__(path, name)
+        self.subtitle = subtitle
+    
+    @property
+    def oss_subtitle(self) -> OSSSubtitle:
+        return self.subtitle

plexflow/core/subtitles/utils/plex_internal_subtitle.py

@@ -0,0 +1,6 @@
+
+from plexflow.core.subtitles.utils.plex_subtitle import PlexSubtitle
+
+class PlexInternalSubtitle(PlexSubtitle):
+    def __init__(self, path, name):
+        super().__init__(path, name)

plexflow/core/subtitles/utils/plex_subtitle.py

@@ -0,0 +1,21 @@
+
+class PlexSubtitle:
+    def __init__(self, path: str, name: str):
+        self.path = path
+        self.name = name
+        self.detected_language = None
+
+    @property
+    def filepath(self) -> str:
+        return self.path
+
+    @property
+    def filename(self) -> str:
+        return self.name
+    
+    def set_detected_language(self, lang):
+        self.detected_language = lang
+    
+    @property
+    def language(self) -> str:
+        return self.detected_language

{plexflow-0.0.109.dist-info → plexflow-0.0.111.dist-info}/METADATA

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

{plexflow-0.0.109.dist-info → plexflow-0.0.111.dist-info}/RECORD

@@ -30,9 +30,9 @@ plexflow/core/context/partials/completed_downloads.py,sha256=BM5apreKS9zeuFOT4Eq
 plexflow/core/context/partials/context.py,sha256=pFzpAFxwCJdqPuhXJqhs6j4Xyvf0GQtzPRMmL4fcjEA,314
 plexflow/core/context/partials/ids.py,sha256=QoQ6FbX1OIWrE-iuz-G6kSzBlTt1_I1jyfl2JgKge2o,913
 plexflow/core/context/partials/movie.py,sha256=VXQ2SspFgGSRgDefg4VlHrH2fns3KRuKlU72ps6527o,3861
-plexflow/core/context/partials/movie_assets.py,sha256=ZUgNpPDBDw_vjg3DcBP9OI2_zzbOh275GHkTBSuHbOc,1225
+plexflow/core/context/partials/movie_assets.py,sha256=qjZTs-lpPfZkQQSKm6CB4aeECX5_YzOom51PxZzmnts,1913
 plexflow/core/context/partials/reports.py,sha256=0W58RwK3VSsVHbF0rhvMNNlZZr01eutwermyvdeEZIs,810
-plexflow/core/context/partials/subtitles.py,sha256=Eax0rdGeTqEHkt9KNiyv097X3I1Drt9xb19VDalyA4Q,542
+plexflow/core/context/partials/subtitles.py,sha256=0NhKGkP-sArQswuSyA7puRSjjoobF-3Ah7Pd39QkgTU,535
 plexflow/core/context/partials/tgx_batch.py,sha256=TduB09oBOQ8CtmPYsHIeNe7AI-ypKw21zQAX-7qktEs,859
 plexflow/core/context/partials/tgx_context.py,sha256=_FuhOvKsFqi_uynHxgC9_QIR2CfYmz-uJCRFtGFJmXI,1641
 plexflow/core/context/partials/torrents.py,sha256=GtO4rlb7V2N-4QEWt-ODssEsdUlQ2o0lTo_BFI5i3lw,1109
@@ -67,13 +67,13 @@ plexflow/core/downloads/candidates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQe
 plexflow/core/downloads/candidates/__pycache__/__init__.cpython-312.pyc,sha256=rukNX00PSteBonRk-cayzNRk17ypWfE4noOAR3Ibjn0,170
 plexflow/core/downloads/candidates/__pycache__/download_candidate.cpython-312.pyc,sha256=f4h0U6Cue2z5QaU_Ik7zHTqITe0SLVSeNRLvIVwsB3c,9487
 plexflow/core/downloads/candidates/__pycache__/utils.cpython-312.pyc,sha256=Jwoj7R4TFzjoXc3uq_Wps3WitzFE9K_miKp9XuT_l8Y,2079
-plexflow/core/downloads/candidates/download_candidate.py,sha256=CUq56hnl624GvrXb5ET0-z1xnNGfbCdJu2MMPItkYnI,5927
+plexflow/core/downloads/candidates/download_candidate.py,sha256=6lQbtNw8uGr5B8qrkOOCxxvCKGHvFoEq8JjHGNB5FN8,5989
 plexflow/core/downloads/candidates/filtered.py,sha256=HgUY0S3aWAeHASHTjND9hyjCtGNLhUzLOL90G2CDxKg,1842
 plexflow/core/downloads/candidates/rank/__pycache__/ranking.cpython-312.pyc,sha256=Wz8hGAtLZNT0ThAPwloLBek5wflI6DKFDhr5_b7TzPI,4045
 plexflow/core/downloads/candidates/rank/__pycache__/utils.cpython-312.pyc,sha256=5Q6q1kw62WCWlPVFpXgXYt6oqf2gWHK_DHaZ-CBADwk,426
 plexflow/core/downloads/candidates/rank/ranking.py,sha256=p86f3DQmXJNE53gPk9E5RcKA7dgXI4wPQsIh9ae1uxY,2490
 plexflow/core/downloads/candidates/rank/utils.py,sha256=R4Ai4KmZRONo9vd2rf32DdGxVkANHYFwSjrO5WVjX9U,100
-plexflow/core/downloads/candidates/utils.py,sha256=ahI6bvk7CHT_O0BEXsd7FtC180Swlt9Phj6_op1mhYA,1777
+plexflow/core/downloads/candidates/utils.py,sha256=BnuDSRjU9mVYSh9ejKgfvlLi0bWEqc07uLyskC-bYhw,1874
 plexflow/core/downloads/qbit/__pycache__/interface.cpython-312.pyc,sha256=szUoTWpZNpHwJ0OnXer8ZLeAYIvHND-rX1lB2Juv-J4,3335
 plexflow/core/downloads/qbit/__pycache__/schedule.cpython-312.pyc,sha256=ESOLrk6sLpb1mtijuDGNKvAHaYopVQBwSDzdt8dFNow,1350
 plexflow/core/downloads/qbit/interface.py,sha256=26--RLSWYnI8nx4UThF2qu3Lm7hqpOxKw5Sp47qqg2o,2435
@@ -381,6 +381,9 @@ plexflow/core/subtitles/results/__pycache__/__init__.cpython-312.pyc,sha256=pbtC
 plexflow/core/subtitles/results/__pycache__/subtitle.cpython-311.pyc,sha256=CjeZEPM8DrolKN7eIIHM2kEN3gdc90yhdZssUN0ImdY,6400
 plexflow/core/subtitles/results/__pycache__/subtitle.cpython-312.pyc,sha256=kDLEfwKiTgKMvis6GQ19lXs8Ybz09KV8tgMLx6D1BhE,5984
 plexflow/core/subtitles/results/subtitle.py,sha256=1_xeg_l93b-gWRJEKrOSA8k1OlNOfJgotFWWsfgiUvM,3852
+plexflow/core/subtitles/utils/plex_external_subtitle.py,sha256=LOhYDFZNgAhXROdJeqWwI1EeGLGpo5MQZtB0lwgIfbU,407
+plexflow/core/subtitles/utils/plex_internal_subtitle.py,sha256=kFM789VRX7BuTldYmjCUG22E06xWgEej2L3U5Bp4JHI,185
+plexflow/core/subtitles/utils/plex_subtitle.py,sha256=djARWfuKF7GXyeS0tm6yVtEtMATDoPl61_ja1NTWYUY,470
 plexflow/core/torrents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/core/torrents/__pycache__/__init__.cpython-311.pyc,sha256=Ak5TvWroVnHx7z-S-hqYzBrZyP5axDM85dpaSb4yI3o,172
 plexflow/core/torrents/__pycache__/__init__.cpython-312.pyc,sha256=Dh3-EcDVLfB2Sw6dG4fNzbkxJNbbXhby2kX0kE6VCdI,158
@@ -691,7 +694,7 @@ plexflow/utils/video/__pycache__/audio.cpython-312.pyc,sha256=vXBnJwWgTDFdixMBs-
 plexflow/utils/video/__pycache__/subtitle.cpython-312.pyc,sha256=PCjpCLydGXaRsQy6cikhgsEs8WlComfOoYPiLFqfVMA,2515
 plexflow/utils/video/audio.py,sha256=tJ_lNwcjVuBQYD5cYOlXpr__eh8-hnReIgNRgIYOpqo,3380
 plexflow/utils/video/subtitle.py,sha256=qPvvBjlPj0fynJJvGJgGeKt9ey26R-cF6EoLaYt9iXU,1333
-plexflow-0.0.109.dist-info/METADATA,sha256=6F1QRG0TQZ10hajMo70MMBTpcai9eWnCJwbb8l28ugM,3051
-plexflow-0.0.109.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.109.dist-info/entry_points.txt,sha256=SSLjrLcZa2880346VUIsCG9vcLsquf-RTX9pkdNaTO8,1498
-plexflow-0.0.109.dist-info/RECORD,,
+plexflow-0.0.111.dist-info/METADATA,sha256=xdqWRniW9c6cTYuuE-0CQBNfJd2XERt8aEQuPJab8os,3051
+plexflow-0.0.111.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.111.dist-info/entry_points.txt,sha256=SSLjrLcZa2880346VUIsCG9vcLsquf-RTX9pkdNaTO8,1498
+plexflow-0.0.111.dist-info/RECORD,,