mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-06 05:01:10 -07:00
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
import requests
|
|
|
|
from core import logger
|
|
|
|
|
|
def command_complete(url, params, headers, section):
|
|
try:
|
|
r = requests.get(url, params=params, headers=headers, stream=True, verify=False, timeout=(30, 60))
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL: {0}".format(url), section)
|
|
return None
|
|
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 None
|
|
else:
|
|
try:
|
|
return r.json()['state']
|
|
except (ValueError, KeyError):
|
|
# ValueError catches simplejson's JSONDecodeError and json's ValueError
|
|
logger.error("{0} did not return expected json data.".format(section), section)
|
|
return None
|
|
|
|
|
|
def completed_download_handling(url2, headers, section="MAIN"):
|
|
try:
|
|
r = requests.get(url2, params={}, headers=headers, stream=True, verify=False, timeout=(30, 60))
|
|
except requests.ConnectionError:
|
|
logger.error("Unable to open URL: {0}".format(url2), section)
|
|
return False
|
|
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 False
|
|
else:
|
|
try:
|
|
return r.json().get("enableCompletedDownloadHandling", False)
|
|
except ValueError:
|
|
# ValueError catches simplejson's JSONDecodeError and json's ValueError
|
|
return False
|