Merge 38e08a1147 into 282e19db82

-Includes suggested changes
-Updates test cases

yt_dlp/extractor/_extractors.py

@@ -2490,6 +2490,7 @@ from .xinpianchang import XinpianchangIE
 from .xminus import XMinusIE
 from .xnxx import XNXXIE
 from .xstream import XstreamIE
+from .xumo import XumoIE
 from .xvideos import (
     XVideosIE,
     XVideosQuickiesIE,

yt_dlp/extractor/xumo.py

@@ -0,0 +1,178 @@
+from .common import InfoExtractor
+from ..utils import determine_ext, int_or_none, smuggle_url, traverse_obj, unsmuggle_url, url_or_none
+
+
+class XumoIE(InfoExtractor):
+    _VALID_URL = r'https?://play\.?xumo\.com/[^?#]+/(?P<id>XM[A-Z0-9]{12})'
+    _TESTS = [{
+        # movie
+        'url': 'https://play.xumo.com/free-movies/a-circus-tale-and-a-love-song/XM041I5U497VD3',
+        'params': {
+            'check_formats': True,
+        },
+        'md5': 'eaac858a8db4ee5a67d6d16920c24e15',
+        'info_dict': {
+            'id': 'XM041I5U497VD3',
+            'title': 'A Circus Tale & A Love Song',
+            'ext': 'mp4',
+            'description': 'md5:aa6372f4785c528ff04c94a275f63446',
+            'duration': 6887,
+            'release_year': 2016,
+            'thumbnail': r're:^https?://.*\.jpg$',
+        },
+    }, {
+        # entire series
+        'url': 'https://play.xumo.com/tv-shows/super-mario-world/XM0AN69OG47PRN',
+        'params': {
+            'skip_download': True,
+        },
+        'playlist_count': 10,
+        'info_dict': {
+            'id': 'XM0AN69OG47PRN',
+            'title': 'Super Mario World',
+        },
+    }, {
+        # episode of series
+        'url': 'https://play.xumo.com/tv-shows/99991299/XM02D369HADFRR',
+        'md5': 'ed2f396272b39f2e0fe47f02b9ae34cc',
+        'info_dict': {
+            'id': 'XM02D369HADFRR',
+            'title': 'Fire Sale // Misadventures In Robin Hood Woods',
+            'ext': 'mp4',
+            'series': 'Super Mario World',
+            'season_number': 1,
+            'episode_number': 2,
+            'episode': 'Fire Sale // Misadventures In Robin Hood Woods',
+            'description': 'md5:48134d36781cf4b225ec0ee4f05356d3',
+            'thumbnail': r're:^https?://.*\.jpg$',
+            'season': 'Season 1',
+            'duration': 1368,
+        },
+    }, {
+        # video from network-based alternate URL scheme
+        'url': 'https://play.xumo.com/networks/fakenetworkname/99991299/XM02D369HADFRR',
+        'md5': 'ed2f396272b39f2e0fe47f02b9ae34cc',
+        'info_dict': {
+            'id': 'XM02D369HADFRR',
+            'title': 'Fire Sale // Misadventures In Robin Hood Woods',
+            'ext': 'mp4',
+            'series': 'Super Mario World',
+            'season_number': 1,
+            'episode_number': 2,
+            'episode': 'Fire Sale // Misadventures In Robin Hood Woods',
+            'description': 'md5:48134d36781cf4b225ec0ee4f05356d3',
+            'thumbnail': r're:^https?://.*\.jpg$',
+            'season': 'Season 1',
+            'duration': 1368,
+        },
+    }]
+
+    _INFO_URL = 'https://valencia-app-mds.xumo.com/v2/assets/asset/'
+    _INFO_QUERY_PARAMS = {
+        'f': [
+            'connectorId',
+            'title',
+            'providers',
+            'descriptions',
+            'runtime',
+            'originalReleaseYear',
+            'cuePoints',
+            'ratings',
+            'hasCaptions',
+            'availableSince',
+            'genres',
+            'season',
+            'episode',
+            'seasons',
+            'season:all',
+            'episodes.episodeTitle',
+            'episodes.runtime',
+            'episodes.descriptions',
+            'episodes.hasCaptions',
+            'episodes.ratings',
+        ],
+    }
+
+    def _get_video_links(self, video_id, info_json):
+        formats, subtitles = [], {}
+
+        for source in traverse_obj(info_json, ('providers', ..., 'sources', ...)) or []:
+            fmts, subs = [], {}
+
+            format_url = url_or_none(source.get('uri'))
+            if not format_url:
+                continue
+            ext = determine_ext(format_url)
+            if ext == 'm3u8':
+                fmts, subs = self._extract_m3u8_formats_and_subtitles(
+                    format_url, video_id, m3u8_id='hls', fatal=False)
+            elif ext == 'mpd':
+                fmts, subs = self._extract_mpd_formats_and_subtitles(
+                    format_url, video_id, mpd_id='dash', fatal=False)
+            elif format_url.endswith('.ism/Manifest'):
+                fmts, subs = self._extract_ism_formats_and_subtitles(
+                    format_url, video_id, ism_id='mss', fatal=False)
+
+            if source.get('drm'):
+                for f in fmts:
+                    f['has_drm'] = True
+
+            formats.extend(fmts)
+            subtitles = self._merge_subtitles(subtitles, subs)
+
+        for caption in traverse_obj(info_json, ('providers', ..., 'captions', ...)):
+            subtitles.setdefault(caption.get('lang') or 'und', []).append({
+                'url': caption.get('url'),
+            })
+        return formats, subtitles
+
+    def _real_extract(self, url):
+
+        url, smuggled_data = unsmuggle_url(url)
+
+        media_id = self._match_valid_url(url).group('id')
+        media_metadata = self._download_json(f'{self._INFO_URL}{media_id}.json', media_id, query=self._INFO_QUERY_PARAMS)
+
+        title = media_metadata.get('title')
+        content_type = media_metadata['contentType']
+
+        if content_type == 'SERIES':
+            # series => return set of URLs pointing to individual episodes and smuggle series title to avoid extra API call for each episode
+            return self.playlist_result([
+                self.url_result(
+                    smuggle_url(f'https://play.xumo.com/tv-shows/x/{episode["id"]}', {'series': title}),
+                    XumoIE, episode['id'], episode.get('episodeTitle'))
+                for episode in traverse_obj(media_metadata, ('seasons', ..., 'episodes', ...))
+            ], media_id, title)
+
+        # video => return video info
+
+        season_number = None
+        series_title = None
+
+        is_episode = content_type == 'EPISODIC'
+
+        if is_episode:
+            season_number = int_or_none(media_metadata.get('season'))
+            if smuggled_data:
+                series_title = traverse_obj(smuggled_data, 'series')
+            else:
+                series_data = self._download_json(f'{self._INFO_URL}{media_metadata["connectorId"]}.json', media_id, query=self._INFO_QUERY_PARAMS)
+                series_title = traverse_obj(series_data, 'title')
+
+        formats, subtitles = self._get_video_links(media_id, media_metadata)
+
+        return {
+            'id': media_id,
+            'title': title,
+            'description': traverse_obj(media_metadata, ('descriptions', ('large', 'medium', 'small', 'tiny')), get_all=False),
+            'release_year': media_metadata.get('originalReleaseYear'),
+            'duration': media_metadata.get('runtime'),
+            'thumbnail': f'https://image.xumo.com/v1/assets/asset/{media_id}/1024x576.jpg',
+            'formats': formats,
+            'subtitles': subtitles,
+            'episode_number': media_metadata.get('episode'),
+            'season_number': season_number,
+            'episode': title if is_episode else None,
+            'series': series_title,
+        }