mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-08-13 18:16:52 -07:00
Refactor parses from utils to utils.parsers
This commit is contained in:
Labrys of Knossos
parent
bd16f11485
commit
2d6e8034e2
2 changed files with 160 additions and 153 deletions
|
|
@ -22,6 +22,10 @@ import subliminal
|
||||||
|
|
||||||
import core
|
import core
|
||||||
from core import extractor, logger, main_db
|
from core import extractor, logger, main_db
|
||||||
|
from core.utils.parsers import (
|
||||||
|
parse_args, parse_deluge, parse_other, parse_qbittorrent, parse_rtorrent, parse_transmission,
|
||||||
|
parse_utorrent, parse_vuze,
|
||||||
|
)
|
||||||
from core.utils.processes import RunningProcess
|
from core.utils.processes import RunningProcess
|
||||||
from core.utils.torrents import create_torrent_class, pause_torrent, remove_torrent, resume_torrent
|
from core.utils.torrents import create_torrent_class, pause_torrent, remove_torrent, resume_torrent
|
||||||
|
|
||||||
|
|
@ -479,159 +483,6 @@ def convert_to_ascii(input_name, dir_name):
|
||||||
return input_name, dir_name
|
return input_name, dir_name
|
||||||
|
|
||||||
|
|
||||||
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
|
|
||||||
|
|
||||||
|
|
||||||
def get_dirs(section, subsection, link='hard'):
|
def get_dirs(section, subsection, link='hard'):
|
||||||
to_return = []
|
to_return = []
|
||||||
|
|
||||||
|
|
|
||||||
156
core/utils/parsers.py
Normal file
156
core/utils/parsers.py
Normal file
|
|
@ -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
|
||||||
Loading…
Add table
Add a link
Reference in a new issue