Browse Source

[drtv] Add geo countries to GeoRestrictedError

Sergey M․ 8 years ago
parent
commit
96182695e4
1 changed files with 3 additions and 1 deletions
  1. 3 1
      youtube_dl/extractor/drtv.py

+ 3 - 1
youtube_dl/extractor/drtv.py

@@ -15,6 +15,8 @@ from ..utils import (
 
 
 class DRTVIE(InfoExtractor):
 class DRTVIE(InfoExtractor):
     _VALID_URL = r'https?://(?:www\.)?dr\.dk/(?:tv/se|nyheder|radio/ondemand)/(?:[^/]+/)*(?P<id>[\da-z-]+)(?:[/#?]|$)'
     _VALID_URL = r'https?://(?:www\.)?dr\.dk/(?:tv/se|nyheder|radio/ondemand)/(?:[^/]+/)*(?P<id>[\da-z-]+)(?:[/#?]|$)'
+    _GEO_BYPASS = False
+    _GEO_COUNTRIES = ['DK']
     IE_NAME = 'drtv'
     IE_NAME = 'drtv'
     _TESTS = [{
     _TESTS = [{
         'url': 'https://www.dr.dk/tv/se/boern/ultra/klassen-ultra/klassen-darlig-taber-10',
         'url': 'https://www.dr.dk/tv/se/boern/ultra/klassen-ultra/klassen-darlig-taber-10',
@@ -137,7 +139,7 @@ class DRTVIE(InfoExtractor):
         if not formats and restricted_to_denmark:
         if not formats and restricted_to_denmark:
             self.raise_geo_restricted(
             self.raise_geo_restricted(
                 'Unfortunately, DR is not allowed to show this program outside Denmark.',
                 'Unfortunately, DR is not allowed to show this program outside Denmark.',
-                expected=True)
+                countries=self._GEO_COUNTRIES)
 
 
         self._sort_formats(formats)
         self._sort_formats(formats)