浏览代码

[dramafever] Improve geo restriction detection and use geo bypass mechanism

Sergey M․ 8 年之前
父节点
当前提交
d392005a79
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      youtube_dl/extractor/dramafever.py

+ 3 - 2
youtube_dl/extractor/dramafever.py

@@ -116,8 +116,9 @@ class DramaFeverIE(DramaFeverBaseIE):
                 'http://www.dramafever.com/amp/episode/feed.json?guid=%s' % video_id)
                 'http://www.dramafever.com/amp/episode/feed.json?guid=%s' % video_id)
         except ExtractorError as e:
         except ExtractorError as e:
             if isinstance(e.cause, compat_HTTPError):
             if isinstance(e.cause, compat_HTTPError):
-                raise ExtractorError(
-                    'Currently unavailable in your country.', expected=True)
+                self.raise_geo_restricted(
+                    msg='Currently unavailable in your country',
+                    countries=['US', 'CA'])
             raise
             raise
 
 
         series_id, episode_number = video_id.split('.')
         series_id, episode_number = video_id.split('.')