Correct exit on connect failure

This commit is contained in:
Clinton Hall 2014-05-07 22:18:08 +09:30
commit d9465de4d8

View file

@ -14,20 +14,20 @@ from nzbtomedia.transcoder import transcoder
class autoProcessTV: class autoProcessTV:
def numMissing(self, url1, params, headers): def numMissing(self, url1, params, headers):
r = None r = None
missing = 0
try: try:
r = requests.get(url1, params=params, headers=headers, stream=True, verify=False) r = requests.get(url1, params=params, headers=headers, stream=True, verify=False)
except requests.ConnectionError: except requests.ConnectionError:
logger.error("Unable to open URL: %s" % (url1), section) logger.error("Unable to open URL: %s" % (url1), section)
missing = 0 return missing
if not r.status_code in [requests.codes.ok, requests.codes.created, requests.codes.accepted]: if not r.status_code in [requests.codes.ok, requests.codes.created, requests.codes.accepted]:
logger.error("Server returned status %s" % (str(r.status_code)), section) logger.error("Server returned status %s" % (str(r.status_code)), section)
missing = 0
else: else:
try: try:
res = json.loads(r.content) res = json.loads(r.content)
missing = int(res['totalRecords']) missing = int(res['totalRecords'])
except: except:
missing = 0 pass
return missing return missing
def processEpisode(self, section, dirName, inputName=None, failed=False, clientAgent = "manual", inputCategory=None): def processEpisode(self, section, dirName, inputName=None, failed=False, clientAgent = "manual", inputCategory=None):