From aa2fbf5b9503f130ea0879279be3cd80be6d7e69 Mon Sep 17 00:00:00 2001 From: Clinton Hall Date: Mon, 27 Jan 2020 13:04:24 +1300 Subject: [PATCH] Add new SG api #1712 --- core/auto_process/tv.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/core/auto_process/tv.py b/core/auto_process/tv.py index 801f3234..79634401 100644 --- a/core/auto_process/tv.py +++ b/core/auto_process/tv.py @@ -271,7 +271,9 @@ def process(section, dir_name, input_name=None, failed=False, client_agent='manu url = None if section == 'SickBeard': - if apikey: + if apikey and fork == 'SickGear': + url = '{0}{1}:{2}{3}/api/{4}/?cmd=sg.postprocess'.format(protocol, host, port, web_root, apikey) + elif apikey: url = '{0}{1}:{2}{3}/api/{4}/?cmd=postprocess'.format(protocol, host, port, web_root, apikey) elif fork == 'Stheno': url = '{0}{1}:{2}{3}/home/postprocess/process_episode'.format(protocol, host, port, web_root) @@ -300,7 +302,7 @@ def process(section, dir_name, input_name=None, failed=False, client_agent='manu login = '{0}{1}:{2}{3}/login'.format(protocol, host, port, web_root) login_params = {'username': username, 'password': password} r = s.get(login, verify=False, timeout=(30, 60)) - if r.status_code == 401 and r.cookies.get('_xsrf'): + if r.status_code in [401, 403] and r.cookies.get('_xsrf'): login_params['_xsrf'] = r.cookies.get('_xsrf') s.post(login, data=login_params, stream=True, verify=False, timeout=(30, 60)) r = s.get(url, auth=(username, password), params=fork_params, stream=True, verify=False, timeout=(30, 1800))