diff options
| author | Remita Amine <remitamine@gmail.com> | 2018-11-21 09:44:10 +0100 | 
|---|---|---|
| committer | Remita Amine <remitamine@gmail.com> | 2018-11-21 09:46:36 +0100 | 
| commit | 6c882aa8991383e1c39a6457cbde5dcab260bff5 (patch) | |
| tree | c64951112e0b2a2468a85b1096048182c002d02b | |
| parent | 183417a50fd68c0c63b1d0621c6a0b44fbf2ac52 (diff) | |
[loc] relax _VALID_URL regex and improve formats extraction
| -rw-r--r-- | youtube_dl/extractor/libraryofcongress.py | 37 | 
1 files changed, 24 insertions, 13 deletions
| diff --git a/youtube_dl/extractor/libraryofcongress.py b/youtube_dl/extractor/libraryofcongress.py index 40295a30b..1e5c82c66 100644 --- a/youtube_dl/extractor/libraryofcongress.py +++ b/youtube_dl/extractor/libraryofcongress.py @@ -16,7 +16,7 @@ from ..utils import (  class LibraryOfCongressIE(InfoExtractor):      IE_NAME = 'loc'      IE_DESC = 'Library of Congress' -    _VALID_URL = r'https?://(?:www\.)?loc\.gov/(?:item/|today/cyberlc/feature_wdesc\.php\?.*\brec=)(?P<id>[0-9]+)' +    _VALID_URL = r'https?://(?:www\.)?loc\.gov/(?:item/|today/cyberlc/feature_wdesc\.php\?.*\brec=)(?P<id>[0-9a-z_.]+)'      _TESTS = [{          # embedded via <div class="media-player"          'url': 'http://loc.gov/item/90716351/', @@ -57,6 +57,12 @@ class LibraryOfCongressIE(InfoExtractor):          'params': {              'skip_download': True,          }, +    }, { +        'url': 'https://www.loc.gov/item/ihas.200197114/', +        'only_matching': True, +    }, { +        'url': 'https://www.loc.gov/item/afc1981005_afs20503/', +        'only_matching': True,      }]      def _real_extract(self, url): @@ -67,12 +73,13 @@ class LibraryOfCongressIE(InfoExtractor):              (r'id=(["\'])media-player-(?P<id>.+?)\1',               r'<video[^>]+id=(["\'])uuid-(?P<id>.+?)\1',               r'<video[^>]+data-uuid=(["\'])(?P<id>.+?)\1', -             r'mediaObjectId\s*:\s*(["\'])(?P<id>.+?)\1'), +             r'mediaObjectId\s*:\s*(["\'])(?P<id>.+?)\1', +             r'data-tab="share-media-(?P<id>[0-9A-F]{32})"'),              webpage, 'media id', group='id')          data = self._download_json(              'https://media.loc.gov/services/v1/media?id=%s&context=json' % media_id, -            video_id)['mediaObject'] +            media_id)['mediaObject']          derivative = data['derivatives'][0]          media_url = derivative['derivativeUrl'] @@ -89,25 +96,29 @@ class LibraryOfCongressIE(InfoExtractor):          if ext not in ('mp4', 'mp3'):              media_url += '.mp4' if is_video else '.mp3' -        if 'vod/mp4:' in media_url: -            formats = [{ -                'url': media_url.replace('vod/mp4:', 'hls-vod/media/') + '.m3u8', +        formats = [] +        if '/vod/mp4:' in media_url: +            formats.append({ +                'url': media_url.replace('/vod/mp4:', '/hls-vod/media/') + '.m3u8',                  'format_id': 'hls',                  'ext': 'mp4',                  'protocol': 'm3u8_native',                  'quality': 1, -            }] -        elif 'vod/mp3:' in media_url: -            formats = [{ -                'url': media_url.replace('vod/mp3:', ''), -                'vcodec': 'none', -            }] +            }) +        http_format = { +            'url': re.sub(r'(://[^/]+/)(?:[^/]+/)*(?:mp4|mp3):', r'\1', media_url), +            'format_id': 'http', +            'quality': 1, +        } +        if not is_video: +            http_format['vcodec'] = 'none' +        formats.append(http_format)          download_urls = set()          for m in re.finditer(                  r'<option[^>]+value=(["\'])(?P<url>.+?)\1[^>]+data-file-download=[^>]+>\s*(?P<id>.+?)(?:(?: |\s+)\((?P<size>.+?)\))?\s*<', webpage):              format_id = m.group('id').lower() -            if format_id == 'gif': +            if format_id in ('gif', 'jpeg'):                  continue              download_url = m.group('url')              if download_url in download_urls: | 
