plexflow 0.0.97__py3-none-any.whl → 0.0.99__py3-none-any.whl

plexflow/core/context/partial_context.py

@@ -17,6 +17,16 @@ class PartialContext:
         universal_id = str(ULID())
         store.store_temporarily(key=store.make_key(dag_run_id), obj=universal_id)
         return universal_id
+    
+    @staticmethod
+    def update_custom(context_id: str, key: str, value: Any, ttl: int = None):
+        ctx = PartialContext(
+            context_id=context_id,
+            dag_run_id=f"CUSTOM_{context_id}",
+            default_ttl=ttl,
+        )
+
+        ctx.set(key, value, ttl=ttl)
 
     def __init__(self, context_id: str, dag_run_id: str, default_ttl: int, **kwargs) -> None:
         self.context_id = context_id

plexflow/core/context/partials/__init__.py

@@ -10,3 +10,4 @@ from plexflow.core.context.partials.reports import TorrentAnalysisReports
 from plexflow.core.context.partials.subtitles import Subtitles
 from plexflow.core.context.partials.universal_torrents import UniversalTorrents
 from plexflow.core.context.partials.candidates import Candidates
+from plexflow.core.context.partials.completed_downloads import CompletedDownloads

plexflow/core/context/partials/completed_downloads.py

@@ -0,0 +1,28 @@
+from plexflow.core.context.partial_context import PartialContext
+from typing import List
+from qbittorrentapi.torrents import TorrentDictionary
+from plexflow.core.storage.object.plexflow_storage import PlexflowObjectStore
+
+class CompletedDownloads(PartialContext):
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+
+    def all(self) -> List[TorrentDictionary]:
+        return self.get("download/completed")
+
+    def update(self, downloads: List[TorrentDictionary]):
+        if len(downloads) == 0:
+            return
+        self.set("download/completed", downloads)
+
+    def torrent(self) -> TorrentDictionary:
+        return self.get("download/torrent")
+
+    @staticmethod
+    def update_completed_torrent(universal_id: str, torrent: TorrentDictionary, **kwargs):
+        PartialContext.update_custom(
+            context_id=universal_id,
+            key="download/torrent",
+            value=torrent,
+            **kwargs
+        )

{plexflow-0.0.97.dist-info → plexflow-0.0.99.dist-info}/METADATA

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

{plexflow-0.0.97.dist-info → plexflow-0.0.99.dist-info}/RECORD

@@ -19,13 +19,14 @@ plexflow/core/context/metadata/context.py,sha256=MoKWnb03_9pnjYacRrfzYN1eVNFbLjO
 plexflow/core/context/metadata/tmdb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/core/context/metadata/tmdb/__pycache__/context.cpython-311.pyc,sha256=dcvTi-0PnZfbLfsLu5QuX_jpAeQYmZR7ZbjzQMHQov4,2627
 plexflow/core/context/metadata/tmdb/context.py,sha256=zlI_Zs_1bAdR60xtl4j2_UBhIery1VcsQz2a53l5g20,1521
-plexflow/core/context/partial_context.py,sha256=Ptad0ZLRgE0FPIJlYzPIGCtrOlxU1mJzScaXzENlgY8,2010
-plexflow/core/context/partials/__init__.py,sha256=oLSaTaAGejMIB1DtBg8xaXUbA3qK43zv48-k0HWxhjI,764
+plexflow/core/context/partial_context.py,sha256=684qHyQ-ulkZnXm-CMmhP7ont-9lIi1WEAbLTjyS4kA,2301
+plexflow/core/context/partials/__init__.py,sha256=6Kr73gP33JOtopWEOK2I1brStbD_1qdB25c7hWVJK9I,846
 plexflow/core/context/partials/__pycache__/__init__.cpython-311.pyc,sha256=x_5HUdt-0vrq9LMmc8Vj0usxqQ_ZHD4WZd4HrvhwAZc,346
 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=TnAPeSEolchQ3znT40xKjEQfeCl9qAMLFThWaeT46JY,1033
+plexflow/core/context/partials/completed_downloads.py,sha256=7BE8VB1NoiURP5H6q9l61TKFAVMB6Z81OtGoJMCIFkA,962
 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
@@ -681,7 +682,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.97.dist-info/METADATA,sha256=MJCxYmOTrIQwRi7BD6RZJMl2ollT0y0LAWBITFPIvqw,3007
-plexflow-0.0.97.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.97.dist-info/entry_points.txt,sha256=uZc6ohXod3uudTgfeTqnkXBS4Cb7eajdjeqZc3P0PX4,1456
-plexflow-0.0.97.dist-info/RECORD,,
+plexflow-0.0.99.dist-info/METADATA,sha256=t4q923QYitwULLNi20dLOR1cFJw28tmoZdVDgLCXQ00,3007
+plexflow-0.0.99.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.99.dist-info/entry_points.txt,sha256=uZc6ohXod3uudTgfeTqnkXBS4Cb7eajdjeqZc3P0PX4,1456
+plexflow-0.0.99.dist-info/RECORD,,