plexflow 0.0.79__py3-none-any.whl → 0.0.81__py3-none-any.whl

plexflow/core/context/partials/candidates.py

@@ -1,6 +1,7 @@
 from plexflow.core.context.partial_context import PartialContext
 from plexflow.core.downloads.candidates.download_candidate import DownloadCandidate
 from typing import List
+from plexflow.core.downloads.candidates.rank.ranking import Ranked
 
 class Candidates(PartialContext):
     def __init__(self, **kwargs) -> None:
@@ -13,3 +14,9 @@ class Candidates(PartialContext):
         if len(candidates) == 0:
             return
         self.set("download/candidates", candidates)
+
+    def update_ranked(self, ranked_candidates: Ranked):
+        self.set('download/ranked/candidates', ranked_candidates)
+    
+    def ranked(self) -> Ranked:
+        return self.get('download/ranked/candidates')

plexflow/core/downloads/candidates/rank/ranking.py

@@ -0,0 +1,61 @@
+from plexflow.core.downloads.candidates.download_candidate import DownloadCandidate
+from typing import List
+from plexflow.core.downloads.candidates.rank.utils import rank_candidate
+
+class RankedCandidate:
+    def __init__(self, rank: float, candidate: DownloadCandidate):
+        self.rank: float = rank
+        self.candidate: DownloadCandidate = candidate
+
+class Ranked:
+    def __init__(self):
+        self.ranked = {
+            'native': {
+                'nl': [],
+                'en': []
+            },
+            'encoder': {
+                'nl': [],
+                'en': []
+            },
+            'nosub': [] 
+        }
+    
+    def append_native(self, candidate: DownloadCandidate, rank: float, language: str):
+        self.ranked['native'][language].append(RankedCandidate(rank, candidate))
+
+    def append_encoder(self, candidate: DownloadCandidate, rank: float, language: str):
+        self.ranked['encoder'][language].append(RankedCandidate(rank, candidate))
+
+    def append_nosub(self, candidate: DownloadCandidate, rank: float):
+        self.ranked['nosub'].append(RankedCandidate(rank, candidate))
+
+    def native_candidates(self, language: str):
+        return sorted(self.ranked['native'][language], key=lambda rc: rc.rank, reverse=True)
+
+    def encoder_candidates(self, language: str):
+        return sorted(self.ranked['encoder'][language], key=lambda rc: rc.rank, reverse=True)
+
+    def nosub_candidates(self):
+        return sorted(self.ranked['nosub'], key=lambda rc: rc.rank, reverse=True)
+
+
+def rank_candidates(candidates: List[DownloadCandidate]):
+    ranked = Ranked()
+
+    for candidate in candidates:
+        seeds = candidate.max_seeds
+        size = candidate.max_size_bytes
+
+        rank = rank_candidate(seeds, size)
+
+        # TODO: support native subtitles
+        if candidate.has_dutch_subtitles:
+            # encoder dutch
+            ranked.append_encoder(candidate=candidate, rank=rank, language='nl')
+        elif candidate.has_english_subtitles:
+            ranked.append_encoder(candidate=candidate, rank=rank, language='en')
+        else:
+            ranked.append_nosub(candidate=candidate, rank=rank)
+    
+    return ranked

plexflow/core/downloads/candidates/rank/utils.py

@@ -0,0 +1,4 @@
+import math
+
+def rank_candidate(seeds, size_bytes):
+    return seeds / math.sqrt(max(1, size_bytes))

plexflow/core/torrents/results/torrent.py

@@ -150,6 +150,14 @@ class Torrent(ABC):
         parts = self.parsed_release_name
         return parts.get("quality")
 
+    @property
+    def has_native_dutch_subtitles(self):
+        return False
+
+    @property
+    def has_native_english_subtitles(self):
+        return False
+
     @property
     def is_bad_quality(self):
         parsed_title = self.parsed_release_name

plexflow/utils/subtitle/search.py

@@ -48,7 +48,7 @@ class SubtitleSearcher:
             pattern : re.Pattern
                 a compiled regular expression pattern
         """
-        pattern = re.compile(rf'\b(text|subtitles?)\b(.*?)\b({hint_word})\b', re.IGNORECASE | re.DOTALL | re.UNICODE)
+        pattern = re.compile(rf'\b(text|subs|subtitles?)\b(.*?)\b({hint_word})\.?\b', re.IGNORECASE | re.DOTALL | re.UNICODE)
         return pattern
 
     def search_subtitles(self, text):

{plexflow-0.0.79.dist-info → plexflow-0.0.81.dist-info}/METADATA

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

{plexflow-0.0.79.dist-info → plexflow-0.0.81.dist-info}/RECORD

@@ -25,7 +25,7 @@ plexflow/core/context/partials/__pycache__/__init__.cpython-311.pyc,sha256=x_5HU
 plexflow/core/context/partials/__pycache__/ids.cpython-311.pyc,sha256=wV7r4FAwonAshns-NGcowqGamtrM_Q_2NbiHQhOLkjg,2638
 plexflow/core/context/partials/__pycache__/watchlist.cpython-311.pyc,sha256=k-0BM-AvH-zObpwY0XR8CYrT21kPEa0fmeBmf0wscE0,1466
 plexflow/core/context/partials/cache.py,sha256=1QQ6yP69wqJiacuc2idhHsXpyUUCqavqgLk25wQB_tI,508
-plexflow/core/context/partials/candidates.py,sha256=0YiMUxITmJ9c37EtlLua9E5ghqrHH2lIPFyVFIcJ15s,543
+plexflow/core/context/partials/candidates.py,sha256=oouE7whR-3trH5xKDxu4ozkf2Kkzkrgda28uwEQTk-M,823
 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
@@ -67,6 +67,10 @@ plexflow/core/downloads/candidates/__pycache__/download_candidate.cpython-312.py
 plexflow/core/downloads/candidates/__pycache__/utils.cpython-312.pyc,sha256=Jwoj7R4TFzjoXc3uq_Wps3WitzFE9K_miKp9XuT_l8Y,2079
 plexflow/core/downloads/candidates/download_candidate.py,sha256=KlW4GhAbp2hhDuoAphuFj3uweJE4LjwRvUNe-R7h-es,5819
 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=NJPpNxZ1z-VT5LkDCaamV0Zqj4IKlbkrZwEojMOI0lc,2181
+plexflow/core/downloads/candidates/rank/utils.py,sha256=R4Ai4KmZRONo9vd2rf32DdGxVkANHYFwSjrO5WVjX9U,100
 plexflow/core/downloads/candidates/utils.py,sha256=ahI6bvk7CHT_O0BEXsd7FtC180Swlt9Phj6_op1mhYA,1777
 plexflow/core/env/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/core/env/env.py,sha256=8cK49WiKD_H3B9K0QiMu7pkXTld40ojRB3sK7noQM44,953
@@ -494,7 +498,7 @@ plexflow/core/torrents/providers/tpb/__pycache__/__init__.cpython-312.pyc,sha256
 plexflow/core/torrents/providers/tpb/__pycache__/tpb.cpython-311.pyc,sha256=e_38p1LN_QMIlb92ffVan46SkaPTaMVYFpys7BzPEaY,1688
 plexflow/core/torrents/providers/tpb/__pycache__/tpb.cpython-312.pyc,sha256=MAp1JprHh9oeFLKRNGOqyytN12g0ZdUSdF6TwJ9MDCs,1504
 plexflow/core/torrents/providers/tpb/__pycache__/utils.cpython-311.pyc,sha256=YwB71cMUDX6MxUduWoFXRQXsP_LIRqK4rAVrJMX4v38,6704
-plexflow/core/torrents/providers/tpb/__pycache__/utils.cpython-312.pyc,sha256=1Ar8w_HhNvx-qrkSNP6KRNKo0rv9MBz5wIkRysb7XE8,6380
+plexflow/core/torrents/providers/tpb/__pycache__/utils.cpython-312.pyc,sha256=cJSN6zMizHkzh3_WWJ2uF4bZ6_svJSQnN296iP8YJv4,6378
 plexflow/core/torrents/providers/tpb/tpb.py,sha256=YB-a3kZ4DUzXkI4TJJszVvoJiX4L0QUJuCo-9iqC6JQ,681
 plexflow/core/torrents/providers/tpb/utils.py,sha256=xZEU9tbRnG8nR2lspPenIA0UqmqxMIH_uv1sMtW79qM,3621
 plexflow/core/torrents/providers/yts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -511,10 +515,10 @@ plexflow/core/torrents/results/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 plexflow/core/torrents/results/__pycache__/__init__.cpython-311.pyc,sha256=FJHx4wYMLA-6GPhdMPY5TUfw7JicV0HMgxCwlpbba54,180
 plexflow/core/torrents/results/__pycache__/__init__.cpython-312.pyc,sha256=ZUXrXdVtbKDOsGXTY2h9h1GTk4atgBsFSsm-ffcVUUc,168
 plexflow/core/torrents/results/__pycache__/torrent.cpython-311.pyc,sha256=SxfylbSxF9gpzXvNey4Fv9XdKLjwm0a0LYlGFhg7zAY,7655
-plexflow/core/torrents/results/__pycache__/torrent.cpython-312.pyc,sha256=6-XcS5D4UMNkJEJ-tNn8DwSROZcLYA_jm3pR8Eu-hbY,7420
-plexflow/core/torrents/results/__pycache__/universal.cpython-312.pyc,sha256=HlAiFLzyDC-7NpW1Nyzj6HPdAefkUTBgUtiMIg6QKWg,14162
+plexflow/core/torrents/results/__pycache__/torrent.cpython-312.pyc,sha256=gVBo7bTDY_6m9MmdgYy6fcnnoVQkXZiRI2iDANyG5RA,7776
+plexflow/core/torrents/results/__pycache__/universal.cpython-312.pyc,sha256=fWAmvqL76rp3K0_ydGNEHFJe-05duOvvo76ZpYjZQYE,14162
 plexflow/core/torrents/results/__pycache__/utils.cpython-312.pyc,sha256=vqEt3jQLzZ-K_E9WV0kKz4K79AScvcg2xyOUTlAcu70,1136
-plexflow/core/torrents/results/torrent.py,sha256=xsFFOG8Wh5YslQ4d5lhrhi2I5LHGBTCdC30CU06GAKs,4770
+plexflow/core/torrents/results/torrent.py,sha256=vsQNeG0PBMwIiuPhnbgtgnl1q7oGiOCN64ClxQDqPz4,4928
 plexflow/core/torrents/results/universal.py,sha256=ycprOMT1Px8IthmB82YhmnxeNt7P3GAcrOlEkpKoaQk,7706
 plexflow/core/torrents/results/utils.py,sha256=abiiO_QQYDpA5aMyO8WFPxnGu5sL5xfACezE5bwrnJU,691
 plexflow/events/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -535,7 +539,7 @@ plexflow/spiders/quiet_logger.py,sha256=adLBggbjrYo4VcJE61ho7TUisSRxeI5VeNCZnSzd
 plexflow/spiders/tgx/__pycache__/settings.cpython-312.pyc,sha256=YH3GVoQ3qDQrnxpdXwfk0yxzSMqx--tEUL0thav9DV0,1181
 plexflow/spiders/tgx/__pycache__/spider.cpython-312.pyc,sha256=YIHL9xFvLKZQ1UpYdTBxohij1mw_Lf0-kBTUA_kxIQE,4849
 plexflow/spiders/tgx/pipelines/__pycache__/dump_json_pipeline.cpython-312.pyc,sha256=luLi4Dnm86iH0LhbojlH-C6tj9m_WDlka5LyIWuOzPA,1784
-plexflow/spiders/tgx/pipelines/__pycache__/meta_pipeline.cpython-312.pyc,sha256=ykgObro_ZUKwGbtiy33Eq2eawz83Tag8NWduagiy7YU,980
+plexflow/spiders/tgx/pipelines/__pycache__/meta_pipeline.cpython-312.pyc,sha256=JG9aZY6NJTGGnkcpvgsgmVTkmC6EcjtpQPrKPzTnKfQ,973
 plexflow/spiders/tgx/pipelines/__pycache__/publish_pipeline.cpython-312.pyc,sha256=Z2_N2cauzcwCsN2CD4IAQBEVBV9GBlfyzxB6c1zbagU,833
 plexflow/spiders/tgx/pipelines/__pycache__/torrent_info_pipeline.cpython-312.pyc,sha256=GOPgySoxsQeIerM99ybAz_fZkQy0HBe1cpZI82nuRvE,876
 plexflow/spiders/tgx/pipelines/__pycache__/validation_pipeline.cpython-312.pyc,sha256=lkUxJGiBxEbapkfSiVHK11fMLme5xeDMGjIKI2SfvhU,1034
@@ -626,7 +630,7 @@ plexflow/utils/strings/json_extract.py,sha256=590oY1LMnbe9COm3fD6z74OtOzxeXd3Hrf
 plexflow/utils/strings/language.py,sha256=J9-wqmCdxf9Ws5_X1tV4vX4d7AGkKci0eaBE4Lit0j0,269
 plexflow/utils/strings/sanitize.py,sha256=V-58iC4-0xIb-dgK595JbPNyyB9jZbY1xHMazXnQyuI,274
 plexflow/utils/subtitle/__pycache__/search.cpython-312.pyc,sha256=VFulOf14P0m9ncFRxXwENemR8wNolA8xZykpfdAq4eg,3542
-plexflow/utils/subtitle/search.py,sha256=MbLJFeU9LDdFnXZXAI86YroBfD4X4Jp0YynLsnEvQ-c,2433
+plexflow/utils/subtitle/search.py,sha256=HW_JOrj-cpRrVnxWh56XjRWhb98E73V1V5QHCq2y7CM,2441
 plexflow/utils/tasks/__pycache__/decorators.cpython-311.pyc,sha256=8zuAsEfYLEBQQ4DGKqkx-JFCjd73kGhAiPblBNFa2cQ,1675
 plexflow/utils/tasks/__pycache__/tasks.cpython-311.pyc,sha256=8RRMTHAy-t6yW0rlDX9wpZyO-9Cg4SQoD-NVasL94SQ,1670
 plexflow/utils/tasks/decorators.py,sha256=1pF3bVFO2YCJd2gFNVYdmaQpbV3nRP1RxJ_OhLU-g7E,3051
@@ -647,7 +651,7 @@ plexflow/utils/torrent/extract/__pycache__/common.cpython-312.pyc,sha256=WTgyNFy
 plexflow/utils/torrent/extract/__pycache__/ext.cpython-312.pyc,sha256=qRVyjKJFb5P5wGETuI05oqUxUaKvNMzBpk8jB7sHk1E,272185
 plexflow/utils/torrent/extract/__pycache__/extratorrent.cpython-312.pyc,sha256=a08Mno5zwRLr5VQTzF5OFskW5cHt5kKdhJbxQRqWzF4,3448
 plexflow/utils/torrent/extract/__pycache__/piratesparadise.cpython-312.pyc,sha256=waiUKJht8KLl40FvbF1Sgspqx1dxckV7Sa4Ft4sQfgw,2621
-plexflow/utils/torrent/extract/__pycache__/tgx.cpython-312.pyc,sha256=KCow3ShVA0F98hmM_ESTANVSoQt2h9L_8-K5rljAMNs,4661
+plexflow/utils/torrent/extract/__pycache__/tgx.cpython-312.pyc,sha256=rH61J6FQJZ97FC8ZwEDKNwU6lYOMenU1QvDYAD-bEwM,4659
 plexflow/utils/torrent/extract/__pycache__/therarbg.cpython-312.pyc,sha256=cPovyLOodqJbjr14ROsBBhRQzRjp9KT0dC_WHb3tZCU,7889
 plexflow/utils/torrent/extract/__pycache__/torrentquest.cpython-312.pyc,sha256=hOpCw3fSf5GPYxZnOv9VEW1RzsnFzTfHH5wK5ubbp3Y,7880
 plexflow/utils/torrent/extract/common.py,sha256=DtEe7aUfE96pwWnVl0aUOlRBq2ps36tEfealFZp70hg,1216
@@ -671,7 +675,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=LOGONGxs_RzmqtGP-DBKreOzS1eUFEKo75Q6AfnavW0,1290
-plexflow-0.0.79.dist-info/METADATA,sha256=jLTyHWvmN5DEQHd9Vr8t-hauw_wbEfTIMbmG972F1eY,2954
-plexflow-0.0.79.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.79.dist-info/entry_points.txt,sha256=aEqDHlozu_zjWrl2sibtrqtQHMgU8kSJZrE782CP47g,1362
-plexflow-0.0.79.dist-info/RECORD,,
+plexflow-0.0.81.dist-info/METADATA,sha256=eZjH3OuFgLv2-r6cT50GyuM5by2PmDhrqzFI4BO5S_E,2954
+plexflow-0.0.81.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.81.dist-info/entry_points.txt,sha256=aEqDHlozu_zjWrl2sibtrqtQHMgU8kSJZrE782CP47g,1362
+plexflow-0.0.81.dist-info/RECORD,,