mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-15 01:32:53 -07:00
eb7822b60b
Logging of debug messages is now optional via log_debug option location in autoProcessMedia.cfg Lots of code cleanup has been performed including cleanup log messages and corrections of spelling errors. Improved release lookup code for autoProcessMovie, narrows search results down by making API calls to the download clients to compare results in CouchPotato's database.
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
import nzbtomedia
|
|
from lib import requests
|
|
from nzbtomedia.nzbToMediaUtil import convert_to_ascii
|
|
from nzbtomedia import logger
|
|
|
|
class autoProcessGames:
|
|
def process(self, dirName, nzbName=None, status=0, clientAgent='manual', inputCategory=None):
|
|
if dirName is None:
|
|
logger.error("No directory was given!")
|
|
return 1 # failure
|
|
|
|
# auto-detect correct section
|
|
section = nzbtomedia.CFG.findsection(inputCategory)
|
|
if not section:
|
|
logger.error(
|
|
"We were unable to find a section for category %s, please check your autoProcessMedia.cfg file." % inputCategory)
|
|
return 1
|
|
|
|
status = int(status)
|
|
|
|
host = nzbtomedia.CFG[section][inputCategory]["host"]
|
|
port = nzbtomedia.CFG[section][inputCategory]["port"]
|
|
apikey = nzbtomedia.CFG[section][inputCategory]["apikey"]
|
|
|
|
try:
|
|
ssl = int(nzbtomedia.CFG[section][inputCategory]["ssl"])
|
|
except:
|
|
ssl = 0
|
|
|
|
try:
|
|
web_root = nzbtomedia.CFG[section][inputCategory]["web_root"]
|
|
except:
|
|
web_root = ""
|
|
|
|
if ssl:
|
|
protocol = "https://"
|
|
else:
|
|
protocol = "http://"
|
|
|
|
nzbName, dirName = convert_to_ascii(nzbName, dirName)
|
|
|
|
url = "%s%s:%s%s/api" % (protocol, host, port, web_root)
|
|
|
|
fields = nzbName.split("-")
|
|
|
|
gamezID = fields[0].replace("[","").replace("]","").replace(" ","")
|
|
|
|
downloadStatus = 'Wanted'
|
|
if status == 0:
|
|
downloadStatus = 'Downloaded'
|
|
|
|
params = {}
|
|
params['api_key'] = apikey
|
|
params['mode'] = 'UPDATEREQUESTEDSTATUS'
|
|
params['db_id'] = gamezID
|
|
params['status'] = downloadStatus
|
|
|
|
logger.debug("Opening URL: %s" % (url),section)
|
|
|
|
try:
|
|
r = requests.get(url, params=params)
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL")
|
|
return 1 # failure
|
|
|
|
result = r.json()
|
|
logger.postprocess("%s" % (result),section)
|
|
|
|
if result['success']:
|
|
logger.postprocess("SUCCESS: Status for %s has been set to %s in Gamez" % (gamezID, downloadStatus),section)
|
|
return 0 # Success
|
|
else:
|
|
logger.error("FAILED: Status for %s has NOT been updated in Gamez" % (gamezID),section)
|
|
return 1 # failure
|