mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 09:30:21 -08:00
6ccc4abc18
`content` returns a bytes object, `text` returns a string object. The latter can be splitted by the string `\n`, the former cannot, which leads to an Exception.
100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
# coding=utf-8
|
|
|
|
from __future__ import (
|
|
absolute_import,
|
|
division,
|
|
print_function,
|
|
unicode_literals,
|
|
)
|
|
|
|
import os
|
|
|
|
import requests
|
|
|
|
import core
|
|
from core import logger
|
|
from core.auto_process.common import ProcessResult
|
|
from core.utils import convert_to_ascii, remote_dir, server_responding
|
|
|
|
requests.packages.urllib3.disable_warnings()
|
|
|
|
|
|
def process(section, dir_name, input_name=None, status=0, client_agent='manual', input_category=None):
|
|
apc_version = '2.04'
|
|
comicrn_version = '1.01'
|
|
|
|
cfg = dict(core.CFG[section][input_category])
|
|
|
|
host = cfg['host']
|
|
port = cfg['port']
|
|
apikey = cfg['apikey']
|
|
ssl = int(cfg.get('ssl', 0))
|
|
web_root = cfg.get('web_root', '')
|
|
remote_path = int(cfg.get('remote_path'), 0)
|
|
protocol = 'https://' if ssl else 'http://'
|
|
|
|
url = '{0}{1}:{2}{3}/api'.format(protocol, host, port, web_root)
|
|
if not server_responding(url):
|
|
logger.error('Server did not respond. Exiting', section)
|
|
return ProcessResult(
|
|
message='{0}: Failed to post-process - {0} did not respond.'.format(section),
|
|
status_code=1,
|
|
)
|
|
|
|
input_name, dir_name = convert_to_ascii(input_name, dir_name)
|
|
clean_name, ext = os.path.splitext(input_name)
|
|
if len(ext) == 4: # we assume this was a standard extension.
|
|
input_name = clean_name
|
|
|
|
params = {
|
|
'cmd': 'forceProcess',
|
|
'apikey': apikey,
|
|
'nzb_folder': remote_dir(dir_name) if remote_path else dir_name,
|
|
}
|
|
|
|
if input_name is not None:
|
|
params['nzb_name'] = input_name
|
|
params['failed'] = int(status)
|
|
params['apc_version'] = apc_version
|
|
params['comicrn_version'] = comicrn_version
|
|
|
|
success = False
|
|
|
|
logger.debug('Opening URL: {0}'.format(url), section)
|
|
try:
|
|
r = requests.post(url, params=params, stream=True, verify=False, timeout=(30, 300))
|
|
except requests.ConnectionError:
|
|
logger.error('Unable to open URL', section)
|
|
return ProcessResult(
|
|
message='{0}: Failed to post-process - Unable to connect to {0}'.format(section),
|
|
status_code=1,
|
|
)
|
|
if r.status_code not in [requests.codes.ok, requests.codes.created, requests.codes.accepted]:
|
|
logger.error('Server returned status {0}'.format(r.status_code), section)
|
|
return ProcessResult(
|
|
message='{0}: Failed to post-process - Server returned status {1}'.format(section, r.status_code),
|
|
status_code=1,
|
|
)
|
|
|
|
result = r.text
|
|
if not type(result) == list:
|
|
result = result.split('\n')
|
|
for line in result:
|
|
if line:
|
|
logger.postprocess('{0}'.format(line), section)
|
|
if 'Post Processing SUCCESSFUL' in line:
|
|
success = True
|
|
|
|
if success:
|
|
logger.postprocess('SUCCESS: This issue has been processed successfully', section)
|
|
return ProcessResult(
|
|
message='{0}: Successfully post-processed {1}'.format(section, input_name),
|
|
status_code=0,
|
|
)
|
|
else:
|
|
logger.warning('The issue does not appear to have successfully processed. Please check your Logs', section)
|
|
return ProcessResult(
|
|
message='{0}: Failed to post-process - Returned log from {0} was not as expected.'.format(section),
|
|
status_code=1,
|
|
)
|