[SnapchatStory] Add new extractor

youtube_dl/extractor/extractors.py

@@ -1011,6 +1011,7 @@ from .smotri import (
     SmotriUserIE,
     SmotriBroadcastIE,
 )
+from .snapchat import SnapchatStoryIE
 from .snotr import SnotrIE
 from .sohu import SohuIE
 from .sonyliv import SonyLIVIE

youtube_dl/extractor/snapchat.py

@@ -0,0 +1,75 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+from ..compat import compat_urllib_parse_urlencode
+from ..utils import ExtractorError, int_or_none
+
+
+class SnapchatStoryIE(InfoExtractor):
+    _VALID_URL = r'https?://(?:story|play)\.snapchat\.com/(?:s/)?(?P<id>[^/?#&]+)'
+    _TEST = {
+        'url': 'https://story.snapchat.com/s/m:W7_EDlXWTBiXAEEniNoMPwAAYHNdO123IQ281AWXc-zodAWXc-ziUAHanAA',
+        'md5': '3eebf8f327752d100959db3fa0879bfb',
+        'info_dict': {
+            'id': 'W7_EDlXWTBiXAEEniNoMPwAAYHNdO123IQ281AWXc-zodAWXc-ziUAHanAA',
+            'ext': 'mp4',
+            'title': 'Dunn, North Carolina',
+            'thumbnail': 'https://s.sc-cdn.net/N_FHqfPfsuaDG2vf33HsmA4ipcDtMBuW5SM3dBiME38=/default/preview_overlay.jpg',
+        },
+    }
+
+    def _entries(self, story, title, alt_title):
+        for snap in story.get('snaps', []):
+            media = snap.get('media', {})
+
+            if not media.get('type', '').startswith('VIDEO'):
+                continue
+
+            if len(media.keys()) == 0:
+                continue
+
+            formats = []
+
+            for key in ('mediaStreamingUrl', 'mediaUrl'):
+                if key not in media:
+                    continue
+
+                format_info = {'url': media[key], 'format_id': 'direct'}
+
+                if key == 'mediaStreamingUrl':
+                    format_info['format_id'] = 'streaming'
+                    format_info['ext'] = 'mp4'
+                    format_info['protocol'] = 'm3u8'
+
+                formats.append(format_info)
+
+            if len(formats) == 0:
+                continue
+
+            info = {
+                'id': snap['id'],
+                'title': title,
+                'alt_title': alt_title,
+                'thumbnail': media.get('mediaPreviewUrl'),
+                'duration': int_or_none(snap.get('captureTimeSecs')),
+                'formats': formats,
+            }
+
+            yield info
+
+    def _real_extract(self, url):
+        snap_id = self._match_id(url)
+        data_url = 'https://storysharing.snapchat.com/v1/fetch/%s' % snap_id
+        data = self._download_json(data_url,
+                                   snap_id,
+                                   query=dict(request_origin='ORIGIN_WEB_PLAYER'))
+
+        story = data['story']
+        title = story['metadata']['title']
+        alt_title = story['metadata'].get('subTitles')
+
+        return self.playlist_result(self._entries(story, title, alt_title),
+                                    playlist_title=title,
+                                    playlist_id=snap_id,
+                                    playlist_description=alt_title)