Refactor parses from utils to utils.parsers

core/utils/parsers.py

@@ -0,0 +1,156 @@
+import os
+
+import core
+
+
+def parse_other(args):
+    return os.path.normpath(args[1]), '', '', '', ''
+
+
+def parse_rtorrent(args):
+    # rtorrent usage: system.method.set_key = event.download.finished,TorrentToMedia,
+    # 'execute={/path/to/nzbToMedia/TorrentToMedia.py,\'$d.get_base_path=\',\'$d.get_name=\',\'$d.get_custom1=\',\'$d.get_hash=\'}'
+    input_directory = os.path.normpath(args[1])
+    try:
+        input_name = args[2]
+    except Exception:
+        input_name = ''
+    try:
+        input_category = args[3]
+    except Exception:
+        input_category = ''
+    try:
+        input_hash = args[4]
+    except Exception:
+        input_hash = ''
+    try:
+        input_id = args[4]
+    except Exception:
+        input_id = ''
+
+    return input_directory, input_name, input_category, input_hash, input_id
+
+
+def parse_utorrent(args):
+    # uTorrent usage: call TorrentToMedia.py '%D' '%N' '%L' '%I'
+    input_directory = os.path.normpath(args[1])
+    input_name = args[2]
+    try:
+        input_category = args[3]
+    except Exception:
+        input_category = ''
+    try:
+        input_hash = args[4]
+    except Exception:
+        input_hash = ''
+    try:
+        input_id = args[4]
+    except Exception:
+        input_id = ''
+
+    return input_directory, input_name, input_category, input_hash, input_id
+
+
+def parse_deluge(args):
+    # Deluge usage: call TorrentToMedia.py TORRENT_ID TORRENT_NAME TORRENT_DIR
+    input_directory = os.path.normpath(args[3])
+    input_name = args[2]
+    input_hash = args[1]
+    input_id = args[1]
+    try:
+        input_category = core.TORRENT_CLASS.core.get_torrent_status(input_id, ['label']).get()['label']
+    except Exception:
+        input_category = ''
+    return input_directory, input_name, input_category, input_hash, input_id
+
+
+def parse_transmission(args):
+    # Transmission usage: call TorrenToMedia.py (%TR_TORRENT_DIR% %TR_TORRENT_NAME% is passed on as environmental variables)
+    input_directory = os.path.normpath(os.getenv('TR_TORRENT_DIR'))
+    input_name = os.getenv('TR_TORRENT_NAME')
+    input_category = ''  # We dont have a category yet
+    input_hash = os.getenv('TR_TORRENT_HASH')
+    input_id = os.getenv('TR_TORRENT_ID')
+    return input_directory, input_name, input_category, input_hash, input_id
+
+
+def parse_vuze(args):
+    # vuze usage: C:\full\path\to\nzbToMedia\TorrentToMedia.py '%D%N%L%I%K%F'
+    try:
+        cur_input = args[1].split(',')
+    except Exception:
+        cur_input = []
+    try:
+        input_directory = os.path.normpath(cur_input[0])
+    except Exception:
+        input_directory = ''
+    try:
+        input_name = cur_input[1]
+    except Exception:
+        input_name = ''
+    try:
+        input_category = cur_input[2]
+    except Exception:
+        input_category = ''
+    try:
+        input_hash = cur_input[3]
+    except Exception:
+        input_hash = ''
+    try:
+        input_id = cur_input[3]
+    except Exception:
+        input_id = ''
+    try:
+        if cur_input[4] == 'single':
+            input_name = cur_input[5]
+    except Exception:
+        pass
+
+    return input_directory, input_name, input_category, input_hash, input_id
+
+
+def parse_qbittorrent(args):
+    # qbittorrent usage: C:\full\path\to\nzbToMedia\TorrentToMedia.py '%D|%N|%L|%I'
+    try:
+        cur_input = args[1].split('|')
+    except Exception:
+        cur_input = []
+    try:
+        input_directory = os.path.normpath(cur_input[0].replace('\'', ''))
+    except Exception:
+        input_directory = ''
+    try:
+        input_name = cur_input[1].replace('\'', '')
+    except Exception:
+        input_name = ''
+    try:
+        input_category = cur_input[2].replace('\'', '')
+    except Exception:
+        input_category = ''
+    try:
+        input_hash = cur_input[3].replace('\'', '')
+    except Exception:
+        input_hash = ''
+    try:
+        input_id = cur_input[3].replace('\'', '')
+    except Exception:
+        input_id = ''
+
+    return input_directory, input_name, input_category, input_hash, input_id
+
+
+def parse_args(client_agent, args):
+    clients = {
+        'other': parse_other,
+        'rtorrent': parse_rtorrent,
+        'utorrent': parse_utorrent,
+        'deluge': parse_deluge,
+        'transmission': parse_transmission,
+        'qbittorrent': parse_qbittorrent,
+        'vuze': parse_vuze,
+    }
+
+    try:
+        return clients[client_agent](args)
+    except Exception:
+        return None, None, None, None, None