[wat] Capture and output error message

This commit is contained in:
Sergey M․ 2014-09-23 19:58:35 +07:00
parent f7d159cf95
commit 86916dae4b
1 changed files with 9 additions and 1 deletions

View File

@ -5,7 +5,10 @@ import re
import hashlib
from .common import InfoExtractor
from ..utils import unified_strdate
from ..utils import (
ExtractorError,
unified_strdate,
)
class WatIE(InfoExtractor):
@ -57,6 +60,11 @@ class WatIE(InfoExtractor):
video_info = self.download_video_info(real_id)
error_desc = video_info.get('error_desc')
if error_desc:
raise ExtractorError(
'%s returned error: %s' % (self.IE_NAME, error_desc), expected=True)
geo_list = video_info.get('geoList')
country = geo_list[0] if geo_list else ''