mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-12 16:22:53 -07:00
77 lines
3 KiB
Python
77 lines
3 KiB
Python
# coding=utf-8
|
|
|
|
import os
|
|
import shutil
|
|
|
|
import requests
|
|
|
|
import core
|
|
from core import logger
|
|
from core.utils import convert_to_ascii, server_responding
|
|
|
|
requests.packages.urllib3.disable_warnings()
|
|
|
|
|
|
def process(section, dir_name, input_name=None, status=0, client_agent='manual', input_category=None):
|
|
status = int(status)
|
|
|
|
cfg = dict(core.CFG[section][input_category])
|
|
|
|
host = cfg["host"]
|
|
port = cfg["port"]
|
|
apikey = cfg["apikey"]
|
|
library = cfg.get("library")
|
|
ssl = int(cfg.get("ssl", 0))
|
|
web_root = cfg.get("web_root", "")
|
|
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 [1, "{0}: Failed to post-process - {1} did not respond.".format(section, section)]
|
|
|
|
input_name, dir_name = convert_to_ascii(input_name, dir_name)
|
|
|
|
fields = input_name.split("-")
|
|
|
|
gamez_id = fields[0].replace("[", "").replace("]", "").replace(" ", "")
|
|
|
|
download_status = 'Downloaded' if status == 0 else 'Wanted'
|
|
|
|
params = {
|
|
'api_key': apikey,
|
|
'mode': 'UPDATEREQUESTEDSTATUS',
|
|
'db_id': gamez_id,
|
|
'status': download_status
|
|
}
|
|
|
|
logger.debug("Opening URL: {0}".format(url), section)
|
|
|
|
try:
|
|
r = requests.get(url, params=params, verify=False, timeout=(30, 300))
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return [1, "{0}: Failed to post-process - Unable to connect to {1}".format(section, section)]
|
|
|
|
result = r.json()
|
|
logger.postprocess("{0}".format(result), section)
|
|
if library:
|
|
logger.postprocess("moving files to library: {0}".format(library), section)
|
|
try:
|
|
shutil.move(dir_name, os.path.join(library, input_name))
|
|
except Exception:
|
|
logger.error("Unable to move {0} to {1}".format(dir_name, os.path.join(library, input_name)), section)
|
|
return [1, "{0}: Failed to post-process - Unable to move files".format(section)]
|
|
else:
|
|
logger.error("No library specified to move files to. Please edit your configuration.", section)
|
|
return [1, "{0}: Failed to post-process - No library defined in {1}".format(section, section)]
|
|
|
|
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 [1, "{0}: Failed to post-process - Server returned status {1}".format(section, r.status_code)]
|
|
elif result['success']:
|
|
logger.postprocess("SUCCESS: Status for {0} has been set to {1} in Gamez".format(gamez_id, download_status), section)
|
|
return [0, "{0}: Successfully post-processed {1}".format(section, input_name)]
|
|
else:
|
|
logger.error("FAILED: Status for {0} has NOT been updated in Gamez".format(gamez_id), section)
|
|
return [1, "{0}: Failed to post-process - Returned log from {1} was not as expected.".format(section, section)]
|