From 8219ef642760c1f2db66c95c764879c5ef5f63c1 Mon Sep 17 00:00:00 2001 From: Unknown Date: Tue, 29 Sep 2020 01:42:25 +0200 Subject: [PATCH] [tiktok] add referer - required to download from cdn --- youtube_dlc/extractor/tiktok.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/youtube_dlc/extractor/tiktok.py b/youtube_dlc/extractor/tiktok.py index 075a2cdf9..52e5f4f1f 100644 --- a/youtube_dlc/extractor/tiktok.py +++ b/youtube_dlc/extractor/tiktok.py @@ -12,7 +12,7 @@ class TikTokBaseIE(InfoExtractor): - def _extract_aweme(self, video_data, webpage): + def _extract_aweme(self, video_data, webpage, url): video_info = try_get( video_data, lambda x: x['videoData']['itemInfos'], dict) author_info = try_get( @@ -60,7 +60,10 @@ def _extract_aweme(self, video_data, webpage): 'webpage_url': self._og_search_url(webpage), 'description': str_or_none(video_info.get('text')) or str_or_none(share_info.get('desc')), 'ext': 'mp4', - 'formats': formats + 'formats': formats, + 'http_headers': { + 'Referer': url, + } } @@ -131,6 +134,6 @@ def _real_extract(self, url): # Chech statusCode for success if video_data.get('statusCode') == 0: - return self._extract_aweme(video_data, webpage) + return self._extract_aweme(video_data, webpage, url) raise ExtractorError('Video not available', video_id=video_id)