plexflow 0.0.92__tar.gz → 0.0.94__tar.gz

{plexflow-0.0.92 → plexflow-0.0.94}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: plexflow
-Version: 0.0.92
+Version: 0.0.94
 Summary: A short description of the package.
 License: MIT
 Keywords: keyword1,keyword2,keyword3
@@ -51,6 +51,7 @@ Requires-Dist: pyngrok (>=7.1.6,<8.0.0)
 Requires-Dist: pysrt (>=1.1.2,<2.0.0)
 Requires-Dist: python-ffmpeg (>=2.0.10,<3.0.0)
 Requires-Dist: python-ulid (>=2.7.0,<3.0.0)
+Requires-Dist: qbittorrent-api (>=2025.2.0,<2026.0.0)
 Requires-Dist: ratelimit (>=2.2.1,<3.0.0)
 Requires-Dist: redis (>=4.1.3,<5.0.0)
 Requires-Dist: requests (>=2.25.1,<3.0.0)

{plexflow-0.0.92 → plexflow-0.0.94}/plexflow/core/downloads/candidates/download_candidate.py

@@ -183,6 +183,10 @@ class DownloadCandidate:
             len(self.subtitles) > 0
         ])
 
+    @property
+    def magnets(self) -> List[str]:
+        return [t.magnet for t in self.torrent.torrents]
+
     @property
     def has_dutch_subtitles(self) -> bool:
         """

plexflow-0.0.94/plexflow/core/downloads/qbit/schedule.py

@@ -0,0 +1,24 @@
+import qbittorrentapi
+from plexflow.core.downloads.candidates.download_candidate import DownloadCandidate
+from typing import List
+
+
+def schedule_download(magnet: str, category: str = None, tags: List[str] = ()):
+    # instantiate a Client using the appropriate WebUI configuration
+    conn_info = dict(
+        host="192.168.1.50",
+        port=8081,
+        username="admin",
+        password="adminadmin",
+    )
+    qbt_client = qbittorrentapi.Client(**conn_info)
+
+    # or use a context manager:
+    with qbittorrentapi.Client(**conn_info) as qbt_client:
+        if qbt_client.torrents_add(
+            urls=magnet, 
+            save_path='/media/cloud2/mediacloud/tmp/torrents',
+            content_layout='Subfolder',
+            tags=tags,
+            category=category) != "Ok.":
+            raise Exception("Failed to add torrent.")

plexflow-0.0.94/plexflow/utils/google/pubsub/consume.py

@@ -0,0 +1,37 @@
+from google.cloud import pubsub_v1
+import json
+
+def consume_message(topic_id: str, project_id: str, as_json: bool = True, timeout: int = 5):
+    subscriber = pubsub_v1.SubscriberClient()
+    # The `subscription_path` method creates a fully qualified identifier
+    # in the form `projects/{project_id}/subscriptions/{subscription_id}`
+    subscription_path = subscriber.subscription_path(project_id, topic_id)
+
+    data = None
+
+    def callback(message: pubsub_v1.subscriber.message.Message) -> None:
+        nonlocal data
+
+        print(f"Received {message}.")
+        print(message.data.decode())
+        if as_json:
+            data = json.loads(message.data.decode())
+        else:
+            data = message.data.decode()
+
+        message.ack()
+
+    streaming_pull_future = subscriber.subscribe(subscription_path, callback=callback)
+    print(f"Listening for messages on {subscription_path}..\n")
+
+    # Wrap subscriber in a 'with' block to automatically call close() when done.
+    with subscriber:
+        try:
+            # When `timeout` is not set, result() will block indefinitely,
+            # unless an exception is encountered first.
+            streaming_pull_future.result(timeout=timeout)
+        except TimeoutError:
+            streaming_pull_future.cancel()  # Trigger the shutdown.
+            streaming_pull_future.result()  # Block until the shutdown is complete.
+    
+    return data

{plexflow-0.0.92 → plexflow-0.0.94}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "plexflow"
-version = "0.0.92"
+version = "0.0.94"
 description = "A short description of the package."
 authors = ["Your Name <you@example.com>"]
 license = "MIT"
@@ -75,6 +75,7 @@ iso639-lang = "^2.2.3"
 cloudinary = "^1.41.0"
 mistralai = "^1.2.5"
 google-cloud-pubsub = "^2.28.0"
+qbittorrent-api = "^2025.2.0"
 [tool.poetry.dev-dependencies]
 pytest = "^8.2.1"
 
@@ -111,4 +112,5 @@ auto_torrents = 'scripts.torrents.auto_torrents:main'
 candidate_matching = 'scripts.torrents.candidate_matching:main'
 audio_extract = 'scripts.video.audio_extract:main'
 groq_transcribe = 'scripts.transcribe.groq_transcribe:main'
-mistral = 'scripts.mistral.mistral:main'
+mistral = 'scripts.mistral.mistral:main'
+schedule = 'scripts.downloads.qbit.schedule:main'