mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-08-19 21:03:14 -07:00
Use dict literal or comprehension for dict creation
This commit is contained in:
parent
c99b497bd8
commit
6d0d2d3f7e
4 changed files with 35 additions and 24 deletions
|
@ -63,7 +63,6 @@ SABNZB_NO_OF_ARGUMENTS = 8
|
||||||
SABNZB_0717_NO_OF_ARGUMENTS = 9
|
SABNZB_0717_NO_OF_ARGUMENTS = 9
|
||||||
|
|
||||||
# sickbeard fork/branch constants
|
# sickbeard fork/branch constants
|
||||||
FORKS = {}
|
|
||||||
FORK_DEFAULT = 'default'
|
FORK_DEFAULT = 'default'
|
||||||
FORK_FAILED = 'failed'
|
FORK_FAILED = 'failed'
|
||||||
FORK_FAILED_TORRENT = 'failed-torrent'
|
FORK_FAILED_TORRENT = 'failed-torrent'
|
||||||
|
@ -73,15 +72,18 @@ FORK_SICKBEARD_API = 'SickBeard-api'
|
||||||
FORK_MEDUSA = 'Medusa'
|
FORK_MEDUSA = 'Medusa'
|
||||||
FORK_SICKGEAR = 'SickGear'
|
FORK_SICKGEAR = 'SickGear'
|
||||||
FORK_STHENO = 'Stheno'
|
FORK_STHENO = 'Stheno'
|
||||||
FORKS[FORK_DEFAULT] = {'dir': None}
|
|
||||||
FORKS[FORK_FAILED] = {'dirName': None, 'failed': None}
|
FORKS = {
|
||||||
FORKS[FORK_FAILED_TORRENT] = {'dir': None, 'failed': None, 'process_method': None}
|
FORK_DEFAULT: {'dir': None},
|
||||||
FORKS[FORK_SICKRAGE] = {'proc_dir': None, 'failed': None, 'process_method': None, 'force': None, 'delete_on': None}
|
FORK_FAILED: {'dirName': None, 'failed': None},
|
||||||
FORKS[FORK_SICKCHILL] = {'proc_dir': None, 'failed': None, 'process_method': None, 'force': None, 'delete_on': None, 'force_next': None}
|
FORK_FAILED_TORRENT: {'dir': None, 'failed': None, 'process_method': None},
|
||||||
FORKS[FORK_SICKBEARD_API] = {'path': None, 'failed': None, 'process_method': None, 'force_replace': None, 'return_data': None, 'type': None, 'delete': None, 'force_next': None}
|
FORK_SICKRAGE: {'proc_dir': None, 'failed': None, 'process_method': None, 'force': None, 'delete_on': None},
|
||||||
FORKS[FORK_MEDUSA] = {'proc_dir': None, 'failed': None, 'process_method': None, 'force': None, 'delete_on': None, 'ignore_subs': None}
|
FORK_SICKCHILL: {'proc_dir': None, 'failed': None, 'process_method': None, 'force': None, 'delete_on': None, 'force_next': None},
|
||||||
FORKS[FORK_SICKGEAR] = {'dir': None, 'failed': None, 'process_method': None, 'force': None}
|
FORK_SICKBEARD_API: {'path': None, 'failed': None, 'process_method': None, 'force_replace': None, 'return_data': None, 'type': None, 'delete': None, 'force_next': None},
|
||||||
FORKS[FORK_STHENO] = {"proc_dir": None, "failed": None, "process_method": None, "force": None, "delete_on": None, "ignore_subs": None}
|
FORK_MEDUSA: {'proc_dir': None, 'failed': None, 'process_method': None, 'force': None, 'delete_on': None, 'ignore_subs': None},
|
||||||
|
FORK_SICKGEAR: {'dir': None, 'failed': None, 'process_method': None, 'force': None},
|
||||||
|
FORK_STHENO: {"proc_dir": None, "failed": None, "process_method": None, "force": None, "delete_on": None, "ignore_subs": None}
|
||||||
|
}
|
||||||
ALL_FORKS = {k: None for k in set(list(itertools.chain.from_iterable([FORKS[x].keys() for x in FORKS.keys()])))}
|
ALL_FORKS = {k: None for k in set(list(itertools.chain.from_iterable([FORKS[x].keys() for x in FORKS.keys()])))}
|
||||||
|
|
||||||
# NZBGet Exit Codes
|
# NZBGet Exit Codes
|
||||||
|
|
|
@ -163,17 +163,18 @@ def process(section, dir_name, input_name=None, status=0, client_agent='manual',
|
||||||
status_code=0,
|
status_code=0,
|
||||||
)
|
)
|
||||||
|
|
||||||
params = {}
|
params = {
|
||||||
|
'media_folder': remote_dir(dir_name) if remote_path else dir_name,
|
||||||
|
}
|
||||||
|
|
||||||
if download_id and release_id:
|
if download_id and release_id:
|
||||||
params['downloader'] = downloader or client_agent
|
params['downloader'] = downloader or client_agent
|
||||||
params['download_id'] = download_id
|
params['download_id'] = download_id
|
||||||
|
|
||||||
params['media_folder'] = remote_dir(dir_name) if remote_path else dir_name
|
|
||||||
|
|
||||||
if section == 'CouchPotato':
|
if section == 'CouchPotato':
|
||||||
if method == 'manage':
|
if method == 'manage':
|
||||||
command = 'manage.update'
|
command = 'manage.update'
|
||||||
params = {}
|
params.clear()
|
||||||
else:
|
else:
|
||||||
command = 'renamer.scan'
|
command = 'renamer.scan'
|
||||||
|
|
||||||
|
|
|
@ -20,8 +20,16 @@ def auto_fork(section, input_category):
|
||||||
apikey = cfg.get('apikey')
|
apikey = cfg.get('apikey')
|
||||||
ssl = int(cfg.get('ssl', 0))
|
ssl = int(cfg.get('ssl', 0))
|
||||||
web_root = cfg.get('web_root', '')
|
web_root = cfg.get('web_root', '')
|
||||||
replace = {'sickrage': 'SickRage', 'sickchill': 'SickChill', 'sickgear': 'SickGear', 'medusa': 'Medusa', 'sickbeard-api': 'SickBeard-api', 'stheno': 'Stheno'}
|
replace = {
|
||||||
f1 = replace[cfg.get('fork', 'auto')] if cfg.get('fork', 'auto') in replace else cfg.get('fork', 'auto')
|
'medusa': 'Medusa',
|
||||||
|
'sickbeard-api': 'SickBeard-api',
|
||||||
|
'sickgear': 'SickGear',
|
||||||
|
'sickchill': 'SickChill',
|
||||||
|
'sickrage': 'SickRage',
|
||||||
|
'stheno': 'Stheno',
|
||||||
|
}
|
||||||
|
_val = cfg.get('fork', 'auto')
|
||||||
|
f1 = replace.get(_val, _val)
|
||||||
try:
|
try:
|
||||||
fork = f1, core.FORKS[f1]
|
fork = f1, core.FORKS[f1]
|
||||||
except KeyError:
|
except KeyError:
|
||||||
|
|
|
@ -202,17 +202,17 @@ class DBConnection(object):
|
||||||
def table_info(self, table_name):
|
def table_info(self, table_name):
|
||||||
# FIXME ? binding is not supported here, but I cannot find a way to escape a string manually
|
# FIXME ? binding is not supported here, but I cannot find a way to escape a string manually
|
||||||
cursor = self.connection.execute('PRAGMA table_info({0})'.format(table_name))
|
cursor = self.connection.execute('PRAGMA table_info({0})'.format(table_name))
|
||||||
columns = {}
|
return {
|
||||||
for column in cursor:
|
column['name']: {'type': column['type']}
|
||||||
columns[column['name']] = {'type': column['type']}
|
for column in cursor
|
||||||
return columns
|
}
|
||||||
|
|
||||||
# http://stackoverflow.com/questions/3300464/how-can-i-get-dict-from-sqlite-query
|
# http://stackoverflow.com/questions/3300464/how-can-i-get-dict-from-sqlite-query
|
||||||
def _dict_factory(self, cursor, row):
|
def _dict_factory(self, cursor, row):
|
||||||
d = {}
|
return {
|
||||||
for idx, col in enumerate(cursor.description):
|
col[0]: row[idx]
|
||||||
d[col[0]] = row[idx]
|
for idx, col in enumerate(cursor.description)
|
||||||
return d
|
}
|
||||||
|
|
||||||
|
|
||||||
def sanity_check_database(connection, sanity_check):
|
def sanity_check_database(connection, sanity_check):
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue