qbittorrent

> updates to assign_category.py - Changed 'sys.argv' to handle input better - Updated logging for the category's - Implemented better error handing if torrent is not a hi-def movie or show - confirm that the category's are assigned. Added better
2023-11-30 00:37:30 -08:00
parent 1032a63d8c
commit b5dca02805
2 changed files with 137 additions and 19 deletions
--- a/qbittorrent/assign_category.py
+++ b/qbittorrent/assign_category.py
@@ -13,49 +13,79 @@
 #
 import re
 import sys
+import logging
 import qbittorrentapi

+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+# Checking for correct number of command-line arguments
+if len(sys.argv) != 3:
+    logging.error('[ACAT] - Incorrect number of arguments. Exiting script.')
+    sys.exit(1)
+
+# Extract torrent name and hash from command line arguments
+torrent_name = sys.argv[1]
+torrent_hash = sys.argv[2]
+
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - *** STARTING ASSIGN CATEGORY SCRIPT ***')
+logging.warning('[ACAT] - -------------')
+logging.warning(f'[ACAT] - raw torrent name: {torrent_name}')
+logging.warning(f'[ACAT] - raw torrent hash: {torrent_hash}')
+logging.warning('[ACAT] - -------------')
+
 # qbittorrent webui login crendentials.
 qbitt_username = '<USERNAME>'
 qbitt_password = '<PASSWORD>'

 # qbittorrent webui address.
-#url = 'http://192.168.1.55:8080'
+#url = 'http://192.168.1.55:8124'
+#url = 'http://localhost:8124'
 url = 'http://<SERVER IP ADDRESS>:<PORT>'

 # The categories to use for matching torrents. Change as needed. 
 category_1080_shows = "1080+ | Shows"
 category_1080_movies = "1080+ | Movies"

-# Connect to the qBittorrent API
-qb = qbittorrentapi.Client(host=url)
-
-# Log in qbittorrent API
-qb.auth_log_in(username=qbitt_username, password=qbitt_password)
-
-# Get information about all torrents in the download queue
-torrents = qb.torrents_info()
-
-# loop through the torrents
-for torrent in torrents:
-   
-    # Get the name of the completed torrent from the command line arguments inside qbittorrent
-    torrent_name = sys.argv[1]  # Name of the added torrent as defined by "%N" in qbittorrent settings.
-    torrent_hash = sys.argv[2]  # Hash of the added torrent as defined by "%I" in qbittorrent settings.
+try:
+    # Connect to the qBittorrent API
+    qb = qbittorrentapi.Client(host=url)
+    qb.auth_log_in(username=qbitt_username, password=qbitt_password)

    # Define the regular expression pattern for 'SxxExx' or 'sxxexx' format.
    pattern = r"[Ss]\d{2}[Ee]\d{2}"

    # Check for 'SxxExx' or 'sxxexx' episode pattern and '1080p' or '2160p'
-    # If so, assign 1080 show category.
    if re.search(pattern, torrent_name) and ("1080p" in torrent_name or "2160p" in torrent_name):
        qb.torrents_set_category(category=category_1080_shows, hashes=torrent_hash)
        qb.torrents_set_auto_management(hashes=torrent_hash, enable=True)
+        logging.warning(f'[ACAT] - Torrent "{torrent_name}" category set to: "{category_1080_shows}".')
+        logging.warning('[ACAT] - -------------')

    # Then check for movies criteria.
    elif ("1080p" in torrent_name or "2160p" in torrent_name) and "YTS.MX" in torrent_name:
        qb.torrents_set_category(category=category_1080_movies, hashes=torrent_hash)
        qb.torrents_set_auto_management(hashes=torrent_hash, enable=True) 
+        logging.warning(f'[ACAT] - Torrent "{torrent_name}" category set to: "{category_1080_movies}".')
+        logging.warning('[ACAT] - -------------')
+
+    # Torrent is not a Hi-def movie or show
+    else:
+        logging.warning(f'[ACAT] - No category change for Torrent "{torrent_name}".')
+        logging.warning('[ACAT] - -------------')
+
+    # Disconnect from the qBittorrent API
+    qb.auth_log_out()
+
+except Exception as e:
+    logging.error(f'[ACAT] - Error occurred: {e}')
+    logging.warning('[ACAT] - -------------')
+
+logging.warning('[ACAT] - *** ENDING ASSIGN CATEGORY SCRIPT ***')
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - -------------')
+
+

-# Disconnect from the qBittorrent API
-qb.auth_log_out()
--- a/qbittorrent/assign_category_01.py
+++ b/qbittorrent/assign_category_01.py
@@ -0,0 +1,88 @@
+#!/usr/bin/python3
+#
+#   qBittorrent-api:
+#   https://qbittorrent-api.readthedocs.io/en/v2023.4.47/introduction.html
+#
+#
+#
+#   Make sure to set the folling variables. 
+#       <USERNAME>
+#       <SERVER IP ADDRESS> 
+#       <PORT>
+#
+#
+import re
+import sys
+import logging
+import qbittorrentapi
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - *** STARTING ASSIGN CATEGORY SCRIPT ***')
+logging.warning('[ACAT] - -------------')
+
+# qbittorrent webui login crendentials.
+qbitt_username = '<USERNAME>'
+qbitt_password = '<PASSWORD>'
+
+# qbittorrent webui address.
+#url = 'http://192.168.1.55:8080'
+url = 'http://<SERVER IP ADDRESS>:<PORT>'
+
+# The categories to use for matching torrents. Change as needed. 
+category_1080_shows = "1080+ | Shows"
+category_1080_movies = "1080+ | Movies"
+
+# Connect to the qBittorrent API
+qb = qbittorrentapi.Client(host=url)
+
+# Log in qbittorrent API
+qb.auth_log_in(username=qbitt_username, password=qbitt_password)
+
+# Get information about all torrents in the download queue
+torrents = qb.torrents_info()
+
+# loop through the torrents
+for torrent in torrents:
+   
+    # Get the name of the completed torrent from the command line arguments inside qbittorrent
+    torrent_name = sys.argv[1]  # Name of the added torrent as defined by "%N" in qbittorrent settings.
+    torrent_hash = sys.argv[2]  # Hash of the added torrent as defined by "%I" in qbittorrent settings.
+
+    # Define the regular expression pattern for 'SxxExx' or 'sxxexx' format.
+    pattern = r"[Ss]\d{2}[Ee]\d{2}"
+
+    # Check for 'SxxExx' or 'sxxexx' episode pattern and '1080p' or '2160p'
+    # If so, assign 1080 show category.
+    if re.search(pattern, torrent_name) and ("1080p" in torrent_name or "2160p" in torrent_name):
+        qb.torrents_set_category(category=category_1080_shows, hashes=torrent_hash)
+        qb.torrents_set_auto_management(hashes=torrent_hash, enable=True)
+
+    # Then check for movies criteria.
+    elif ("1080p" in torrent_name or "2160p" in torrent_name) and "YTS.MX" in torrent_name:
+        qb.torrents_set_category(category=category_1080_movies, hashes=torrent_hash)
+        qb.torrents_set_auto_management(hashes=torrent_hash, enable=True) 
+
+    # Logging and printing the raw torrent name and what category it was changed to
+    logging.warning(f'[ACAT] - raw torrent name: {torrent_name}')
+    logging.warning(f'[ACAT] - raw torrent hash: {torrent_hash}')
+    logging.warning('[ACAT] - -------------')
+     
+    logging.warning(f'[ACAT] - Torrent "{torrent_name}" category was changed to: "{category_1080_movies}".')
+
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - *** ENDING ASSIGN CATEGORY SCRIPT ***')
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - -------------')
+logging.warning('[ACAT] - -------------')
+
+# Disconnect from the qBittorrent API
+qb.auth_log_out()
+
+
+
+