mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-10 23:24:49 +00:00
[extractor] Fix some errors being converted to ExtractorError
This commit is contained in:
parent
48f796874d
commit
0db3bae879
2 changed files with 16 additions and 6 deletions
|
@ -74,6 +74,7 @@ from ..utils import (
|
|||
strip_or_none,
|
||||
traverse_obj,
|
||||
unescapeHTML,
|
||||
UnsupportedError,
|
||||
unified_strdate,
|
||||
unified_timestamp,
|
||||
update_Request,
|
||||
|
@ -604,10 +605,19 @@ class InfoExtractor(object):
|
|||
if self.__maybe_fake_ip_and_retry(e.countries):
|
||||
continue
|
||||
raise
|
||||
except UnsupportedError:
|
||||
raise
|
||||
except ExtractorError as e:
|
||||
video_id = e.video_id or self.get_temp_id(url)
|
||||
raise ExtractorError(
|
||||
e.msg, video_id=video_id, ie=self.IE_NAME, tb=e.traceback, expected=e.expected, cause=e.cause)
|
||||
kwargs = {
|
||||
'video_id': e.video_id or self.get_temp_id(url),
|
||||
'ie': self.IE_NAME,
|
||||
'tb': e.traceback,
|
||||
'expected': e.expected,
|
||||
'cause': e.cause
|
||||
}
|
||||
if hasattr(e, 'countries'):
|
||||
kwargs['countries'] = e.countries
|
||||
raise type(e)(e.msg, **kwargs)
|
||||
except compat_http_client.IncompleteRead as e:
|
||||
raise ExtractorError('A network error has occurred.', cause=e, expected=True, video_id=self.get_temp_id(url))
|
||||
except (KeyError, StopIteration) as e:
|
||||
|
|
|
@ -2492,9 +2492,9 @@ class GeoRestrictedError(ExtractorError):
|
|||
geographic location due to geographic restrictions imposed by a website.
|
||||
"""
|
||||
|
||||
def __init__(self, msg, countries=None):
|
||||
super(GeoRestrictedError, self).__init__(msg, expected=True)
|
||||
self.msg = msg
|
||||
def __init__(self, msg, countries=None, **kwargs):
|
||||
kwargs['expected'] = True
|
||||
super(GeoRestrictedError, self).__init__(msg, **kwargs)
|
||||
self.countries = countries
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue