From 65493f64e1d8682f7e548f17b064111c075b3b2b Mon Sep 17 00:00:00 2001 From: Felix S Date: Sat, 9 Jul 2022 07:46:57 +0000 Subject: [PATCH] [extractor/Audiodraft] Add extractors (#4288) Based on https://github.com/yt-dlp/yt-dlp/pull/4259 Closes https://github.com/yt-dlp/yt-dlp/issues/4028 Authored by: fstirlitz, Ashish0804 --- yt_dlp/extractor/_extractors.py | 4 ++ yt_dlp/extractor/audiodraft.py | 93 +++++++++++++++++++++++++++++++++ 2 files changed, 97 insertions(+) create mode 100644 yt_dlp/extractor/audiodraft.py diff --git a/yt_dlp/extractor/_extractors.py b/yt_dlp/extractor/_extractors.py index 2c8f2620e..2a83c2854 100644 --- a/yt_dlp/extractor/_extractors.py +++ b/yt_dlp/extractor/_extractors.py @@ -104,6 +104,10 @@ from .atvat import ATVAtIE from .audimedia import AudiMediaIE from .audioboom import AudioBoomIE +from .audiodraft import ( + AudiodraftCustomIE, + AudiodraftGenericIE, +) from .audiomack import AudiomackIE, AudiomackAlbumIE from .audius import ( AudiusIE, diff --git a/yt_dlp/extractor/audiodraft.py b/yt_dlp/extractor/audiodraft.py new file mode 100644 index 000000000..71e5afd8c --- /dev/null +++ b/yt_dlp/extractor/audiodraft.py @@ -0,0 +1,93 @@ +from .common import InfoExtractor +from ..utils import int_or_none + + +class AudiodraftBaseIE(InfoExtractor): + def _audiodraft_extract_from_id(self, player_entry_id): + data_json = self._download_json( + 'https://www.audiodraft.com/scripts/general/player/getPlayerInfoNew.php', player_entry_id, + headers={ + 'Content-type': 'application/x-www-form-urlencoded; charset=UTF-8', + 'X-Requested-With': 'XMLHttpRequest', + }, data=f'id={player_entry_id}'.encode('utf-8')) + + return { + 'id': str(data_json['entry_id']), + 'title': data_json.get('entry_title'), + 'url': data_json['path'], + 'vcodec': 'none', + 'ext': 'mp3', + 'uploader': data_json.get('designer_name'), + 'uploader_id': data_json.get('designer_id'), + 'webpage_url': data_json.get('entry_url'), + 'like_count': int_or_none(data_json.get('entry_likes')), + 'average_rating': int_or_none(data_json.get('entry_rating')), + } + + +class AudiodraftCustomIE(AudiodraftBaseIE): + IE_NAME = 'Audiodraft:custom' + _VALID_URL = r'https?://(?:[-\w]+)\.audiodraft\.com/entry/(?P\d+)' + + _TESTS = [{ + 'url': 'http://nokiatune.audiodraft.com/entry/5874', + 'info_dict': { + 'id': '9485', + 'ext': 'mp3', + 'title': 'Hula Hula Calls', + 'uploader': 'unclemaki', + 'uploader_id': '13512', + 'average_rating': 5, + 'like_count': int, + }, + }, { + 'url': 'http://vikinggrace.audiodraft.com/entry/501', + 'info_dict': { + 'id': '22241', + 'ext': 'mp3', + 'title': 'MVG Happy', + 'uploader': 'frog', + 'uploader_id': '19142', + 'average_rating': 5, + 'like_count': int, + }, + }, { + 'url': 'http://timferriss.audiodraft.com/entry/765', + 'info_dict': { + 'id': '19710', + 'ext': 'mp3', + 'title': 'ferris03', + 'uploader': 'malex', + 'uploader_id': '17335', + 'average_rating': 5, + 'like_count': int, + }, + }] + + def _real_extract(self, url): + id = self._match_id(url) + webpage = self._download_webpage(url, id) + player_entry_id = self._search_regex(r'playAudio\(\'(player_entry_\d+)\'\);', webpage, id, 'play entry id') + return self._audiodraft_extract_from_id(player_entry_id) + + +class AudiodraftGenericIE(AudiodraftBaseIE): + IE_NAME = 'Audiodraft:generic' + _VALID_URL = r'https?://www\.audiodraft\.com/contests/[^/#]+#entries&eid=(?P\d+)' + + _TESTS = [{ + 'url': 'https://www.audiodraft.com/contests/570-Score-A-Video-Surprise-Us#entries&eid=30138', + 'info_dict': { + 'id': '30138', + 'ext': 'mp3', + 'title': 'DROP in sound_V2', + 'uploader': 'TiagoSilva', + 'uploader_id': '19452', + 'average_rating': 4, + 'like_count': int, + }, + }] + + def _real_extract(self, url): + id = self._match_id(url) + return self._audiodraft_extract_from_id(f'player_entry_{id}')