[niconico] fix: download video while sending heartbeat (blackjack4494#72

)
insaneracist · Oct 31, 2020 · d3eb3aa · d3eb3aa
1 parent 709fe21
commit d3eb3aa
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/youtube_dlc/extractor/niconico.py b/youtube_dlc/extractor/niconico.py
@@ -254,6 +254,19 @@ def yesno(boolean):
                 }
             }).encode())
 
+        heartbeat_url = '{}/{}?_format=json&_method=PUT'.format(session_api_endpoint['url'], session_response['data']['session']['id'])
+        heartbeat_headers = {
+            'Accept': 'application/json',
+            'Content-Type': 'application/json',
+        }
+        heartbeat = {
+            'data': json.dumps(session_response['data']),
+            'headers': heartbeat_headers,
+            'interval': session_api_data['heartbeat_lifetime'] / 2000,
+            'method': 'POST',
+            'url': heartbeat_url,
+        }
+
         resolution = video_quality.get('resolution', {})
 
         return {
@@ -264,6 +277,7 @@ def yesno(boolean):
             'vbr': float_or_none(video_quality.get('bitrate'), 1000),
             'height': resolution.get('height'),
             'width': resolution.get('width'),
+            'heartbeat': heartbeat,
         }
 
     def _real_extract(self, url):