[iprima] Improve geo restriction detection and disable geo bypass

This commit is contained in:
Sergey M․ 2017-02-20 23:17:19 +07:00
parent 82f662182b
commit da42ff0668
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

View File

@ -13,6 +13,7 @@ from ..utils import (
class IPrimaIE(InfoExtractor): class IPrimaIE(InfoExtractor):
_VALID_URL = r'https?://play\.iprima\.cz/(?:.+/)?(?P<id>[^?#]+)' _VALID_URL = r'https?://play\.iprima\.cz/(?:.+/)?(?P<id>[^?#]+)'
_GEO_BYPASS = False
_TESTS = [{ _TESTS = [{
'url': 'http://play.iprima.cz/gondici-s-r-o-33', 'url': 'http://play.iprima.cz/gondici-s-r-o-33',
@ -28,6 +29,10 @@ class IPrimaIE(InfoExtractor):
}, { }, {
'url': 'http://play.iprima.cz/particka/particka-92', 'url': 'http://play.iprima.cz/particka/particka-92',
'only_matching': True, 'only_matching': True,
}, {
# geo restricted
'url': 'http://play.iprima.cz/closer-nove-pripady/closer-nove-pripady-iv-1',
'only_matching': True,
}] }]
def _real_extract(self, url): def _real_extract(self, url):
@ -83,7 +88,7 @@ class IPrimaIE(InfoExtractor):
extract_formats(src) extract_formats(src)
if not formats and '>GEO_IP_NOT_ALLOWED<' in playerpage: if not formats and '>GEO_IP_NOT_ALLOWED<' in playerpage:
self.raise_geo_restricted() self.raise_geo_restricted(countries=['CZ'])
self._sort_formats(formats) self._sort_formats(formats)