about summary refs log tree commit diff
path: root/youtube_dl/extractor/tvplay.py
diff options
context:
space:
mode:
authorSergey M․ <dstftw@gmail.com>2017-09-19 23:59:36 +0700
committerSergey M․ <dstftw@gmail.com>2017-09-20 00:00:04 +0700
commitdc76eef092dc10f7e3f599fa7d85a04de8d84b8f (patch)
tree7f276a28d602ab32d3670c2d4244f73b0f2b66c6 /youtube_dl/extractor/tvplay.py
parent8a1a60d17397721620e75d83f2aad3a353286f15 (diff)
downloadyoutube-dl-dc76eef092dc10f7e3f599fa7d85a04de8d84b8f.tar.gz
youtube-dl-dc76eef092dc10f7e3f599fa7d85a04de8d84b8f.tar.xz
youtube-dl-dc76eef092dc10f7e3f599fa7d85a04de8d84b8f.zip
[tvplay] Bypass geo restriction
Diffstat (limited to 'youtube_dl/extractor/tvplay.py')
-rw-r--r--youtube_dl/extractor/tvplay.py12
1 files changed, 11 insertions, 1 deletions
diff --git a/youtube_dl/extractor/tvplay.py b/youtube_dl/extractor/tvplay.py
index 99ff82a5d..46132eda1 100644
--- a/youtube_dl/extractor/tvplay.py
+++ b/youtube_dl/extractor/tvplay.py
@@ -15,7 +15,9 @@ from ..utils import (
     int_or_none,
     parse_iso8601,
     qualities,
+    smuggle_url,
     try_get,
+    unsmuggle_url,
     update_url_query,
 )
 
@@ -224,6 +226,9 @@ class TVPlayIE(InfoExtractor):
     ]
 
     def _real_extract(self, url):
+        url, smuggled_data = unsmuggle_url(url, {})
+        self._initialize_geo_bypass(smuggled_data.get('geo_countries'))
+
         video_id = self._match_id(url)
         geo_country = self._search_regex(
             r'https?://[^/]+\.([a-z]{2})', url,
@@ -426,4 +431,9 @@ class ViafreeIE(InfoExtractor):
                 r'currentVideo["\']\s*:\s*.+?["\']id["\']\s*:\s*["\'](\d{6,})',
                 webpage, 'video id')
 
-        return self.url_result('mtg:%s' % video_id, TVPlayIE.ie_key())
+        return self.url_result(
+            smuggle_url(
+                'mtg:%s' % video_id,
+                {'geo_countries': [
+                    compat_urlparse.urlparse(url).netloc.rsplit('.', 1)[-1]]}),
+            ie=TVPlayIE.ie_key(), video_id=video_id)