Improve error handling logic, follow code conventions

This commit is contained in:
NobleKangaroo 2021-01-27 17:01:25 -05:00
parent 4ecbc49a15
commit fd311fb1e3

View File

@ -110,24 +110,18 @@ class FunimationIE(InfoExtractor):
headers = {}
if self._TOKEN:
headers['Authorization'] = 'Token %s' % self._TOKEN
try:
sources = self._download_json(
'https://www.funimation.com/api/showexperience/%s/' % video_id,
video_id, headers=headers, query={
'pinst_id': ''.join([random.choice(string.digits + string.ascii_letters) for _ in range(8)]),
})
sources = sources['items']
except KeyError:
if 'errors' in sources:
errors = sources['errors']
if len(errors) > 0:
error = errors[0]
if 'detail' in error:
detail = error['detail']
raise ExtractorError('%s said: %s' % (
self.IE_NAME, detail), expected=True)
else:
raise ExtractorError(error, expected=True)
meta = self._download_json(
'https://www.funimation.com/api/showexperience/%s/' % video_id,
video_id, headers=headers, query={
'pinst_id': ''.join([random.choice(string.digits + string.ascii_letters) for _ in range(8)]),
})
sources = meta.get('items') or []
errors = meta.get('errors')
if errors:
if isinstance(errors, list):
raise ExtractorError('\nERROR: '.join([error.get('detail') or error.get('title') or str(error) for error in errors]), expected=True)
else:
raise ExtractorError(errors, expected=True)
except ExtractorError as e:
if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
error = self._parse_json(e.cause.read(), video_id)['errors'][0]