[tvigle] Raise GeoRestrictedError

This commit is contained in:
Sergey M․ 2017-02-25 23:27:14 +07:00
parent 9d0c08a02c
commit b3aec47665
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

View File

@ -17,6 +17,9 @@ class TvigleIE(InfoExtractor):
IE_DESC = 'Интернет-телевидение Tvigle.ru' IE_DESC = 'Интернет-телевидение Tvigle.ru'
_VALID_URL = r'https?://(?:www\.)?(?:tvigle\.ru/(?:[^/]+/)+(?P<display_id>[^/]+)/$|cloud\.tvigle\.ru/video/(?P<id>\d+))' _VALID_URL = r'https?://(?:www\.)?(?:tvigle\.ru/(?:[^/]+/)+(?P<display_id>[^/]+)/$|cloud\.tvigle\.ru/video/(?P<id>\d+))'
_GEO_BYPASS = False
_GEO_COUNTRIES = ['RU']
_TESTS = [ _TESTS = [
{ {
'url': 'http://www.tvigle.ru/video/sokrat/', 'url': 'http://www.tvigle.ru/video/sokrat/',
@ -72,8 +75,13 @@ def _real_extract(self, url):
error_message = item.get('errorMessage') error_message = item.get('errorMessage')
if not videos and error_message: if not videos and error_message:
raise ExtractorError( if item.get('isGeoBlocked') is True:
'%s returned error: %s' % (self.IE_NAME, error_message), expected=True) self.raise_geo_restricted(
msg=error_message, countries=self._GEO_COUNTRIES)
else:
raise ExtractorError(
'%s returned error: %s' % (self.IE_NAME, error_message),
expected=True)
title = item['title'] title = item['title']
description = item.get('description') description = item.get('description')