commit 89966a5aeacb70cc19b0a87a0514be824d2409af
parent 8e3df9dfeef8503e9a8c01fcf42008d376d8d64d
Author: Yen Chi Hsuan <yan12125@gmail.com>
Date: Sat, 2 May 2015 00:32:46 +0800
[viki] Enhance error message handling (#3774)
Diffstat:
1 file changed, 10 insertions(+), 4 deletions(-)
diff --git a/youtube_dl/extractor/viki.py b/youtube_dl/extractor/viki.py
@@ -11,6 +11,7 @@ from ..utils import (
unescapeHTML,
unified_strdate,
US_RATINGS,
+ clean_html,
)
from .common import InfoExtractor
@@ -71,10 +72,15 @@ class VikiIE(InfoExtractor):
req.add_header('User-Agent', self._USER_AGENT)
info_webpage = self._download_webpage(
req, video_id, note='Downloading info page')
- if re.match(r'\s*<div\s+class="video-error', info_webpage):
- raise ExtractorError(
- 'Video %s is blocked from your location.' % video_id,
- expected=True)
+ err_msg = self._html_search_regex(r'<div[^>]+class="video-error[^>]+>(.+)</div>', info_webpage, 'error message', default=None)
+ if err_msg:
+ err_msg = clean_html(err_msg)
+ if 'not available in your region' in err_msg:
+ raise ExtractorError(
+ 'Video %s is blocked from your location.' % video_id,
+ expected=True)
+ else:
+ raise ExtractorError('Viki said: ' + err_msg)
video_url = self._html_search_regex(
r'<source[^>]+src="([^"]+)"', info_webpage, 'video URL')