[floatplane] New version of the Floatplane extractor

youtube_dl/extractor/extractors.py

@@ -351,6 +351,7 @@ from .firsttv import FirstTVIE
 from .fivemin import FiveMinIE
 from .fivetv import FiveTVIE
 from .flickr import FlickrIE
+from .floatplane import FloatplaneIE
 from .folketinget import FolketingetIE
 from .footyroom import FootyRoomIE
 from .formula1 import Formula1IE

youtube_dl/extractor/floatplane.py

@@ -0,0 +1,59 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+from ..utils import ExtractorError, urlencode_postdata, try_get, int_or_none
+from ..compat import compat_str
+
+
+class FloatplaneIE(InfoExtractor):
+    _VALID_URL = r'https?://(?:www\.)?floatplane\.com/video/(?P<id>[A-Za-z0-9]+)'
+    _NETRC_MACHINE = 'floatplane'
+
+    def _real_extract(self, url):
+        video_id = self._match_id(url)
+
+        username, password = self._get_login_info()
+        try:
+            request_data = urlencode_postdata({
+                'username': username,
+                'password': password
+            })
+            user_json = self._download_json('https://www.floatplane.com/api/auth/login', video_id, data=request_data, note='Logging in')
+            self.to_screen('Logged in as %s.' % user_json['user']['username'])
+            # What if url_json['user']['needs2FA']? Can we ask for token from ytdl?
+            # Post 'token' to https://www.floatplane.com/api/auth/checkFor2faLogin
+        except ExtractorError as e:
+            # Will cause 401 if password isn't accepted
+            print(e)
+            raise ExtractorError('Floatplane login error! Check your .netrc for correct login/password.', expected=True)
+
+        try:
+            info_json = self._download_json('https://www.floatplane.com/api/video/info?videoGUID=%s' % video_id, video_id, note='Downloading video info')
+        except ExtractorError as e:
+            print(e)
+            raise ExtractorError('Floatplane download error! Please make sure you\'re logged in.', expected=True)
+
+        formats = []
+        # unfortunately it needs to download video url for each quality
+        #   so I have to download 4 pages each time
+        for level in info_json['levels']:
+            video_url = self._download_webpage('https://www.floatplane.com/api/video/url?guid=%s&quality=%s' % (video_id, level['name']), video_id, note='Downloading url for %s' % level['label'])
+            video_url = video_url.strip('"')
+            formats.append({
+                'format_id': level['name'],
+                'url': video_url,
+                'quality': level.get('label'),
+                'width': level.get('width'),
+                'height': level.get('height')
+            })
+
+        return {
+            'id': video_id,
+            'title': info_json['title'],
+            'description': info_json.get('description'),
+            'url': video_url,
+            'thumbnail': try_get(info_json, lambda x: x['thumbnail']['path'], compat_str),
+            'formats': formats,
+            'duration': int_or_none(info_json.get('duration'))
+        }