Browse Source

[Beatport] Update extractor name and tests

Déstin Reed 9 years ago
parent
commit
ac943d48d3
2 changed files with 6 additions and 6 deletions
  1. 5 5
      youtube_dl/extractor/beatport.py
  2. 1 1
      youtube_dl/extractor/extractors.py

+ 5 - 5
youtube_dl/extractor/beatportpro.py → youtube_dl/extractor/beatport.py

@@ -8,10 +8,10 @@ from ..compat import compat_str
 from ..utils import int_or_none
 from ..utils import int_or_none
 
 
 
 
-class BeatportProIE(InfoExtractor):
-    _VALID_URL = r'https?://pro\.beatport\.com/track/(?P<display_id>[^/]+)/(?P<id>[0-9]+)'
+class BeatportIE(InfoExtractor):
+    _VALID_URL = r'https?://(?:www\.|pro\.)?beatport\.com/track/(?P<display_id>[^/]+)/(?P<id>[0-9]+)'
     _TESTS = [{
     _TESTS = [{
-        'url': 'https://pro.beatport.com/track/synesthesia-original-mix/5379371',
+        'url': 'https://beatport.com/track/synesthesia-original-mix/5379371',
         'md5': 'b3c34d8639a2f6a7f734382358478887',
         'md5': 'b3c34d8639a2f6a7f734382358478887',
         'info_dict': {
         'info_dict': {
             'id': '5379371',
             'id': '5379371',
@@ -20,7 +20,7 @@ class BeatportProIE(InfoExtractor):
             'title': 'Froxic - Synesthesia (Original Mix)',
             'title': 'Froxic - Synesthesia (Original Mix)',
         },
         },
     }, {
     }, {
-        'url': 'https://pro.beatport.com/track/love-and-war-original-mix/3756896',
+        'url': 'https://beatport.com/track/love-and-war-original-mix/3756896',
         'md5': 'e44c3025dfa38c6577fbaeb43da43514',
         'md5': 'e44c3025dfa38c6577fbaeb43da43514',
         'info_dict': {
         'info_dict': {
             'id': '3756896',
             'id': '3756896',
@@ -29,7 +29,7 @@ class BeatportProIE(InfoExtractor):
             'title': 'Wolfgang Gartner - Love & War (Original Mix)',
             'title': 'Wolfgang Gartner - Love & War (Original Mix)',
         },
         },
     }, {
     }, {
-        'url': 'https://pro.beatport.com/track/birds-original-mix/4991738',
+        'url': 'https://beatport.com/track/birds-original-mix/4991738',
         'md5': 'a1fd8e8046de3950fd039304c186c05f',
         'md5': 'a1fd8e8046de3950fd039304c186c05f',
         'info_dict': {
         'info_dict': {
             'id': '4991738',
             'id': '4991738',

+ 1 - 1
youtube_dl/extractor/extractors.py

@@ -94,7 +94,7 @@ from .bbc import (
 from .beeg import BeegIE
 from .beeg import BeegIE
 from .behindkink import BehindKinkIE
 from .behindkink import BehindKinkIE
 from .bellmedia import BellMediaIE
 from .bellmedia import BellMediaIE
-from .beatportpro import BeatportProIE
+from .beatport import BeatportIE
 from .bet import BetIE
 from .bet import BetIE
 from .bigflix import BigflixIE
 from .bigflix import BigflixIE
 from .bild import BildIE
 from .bild import BildIE