github/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-12-06 00:14:33 -05:00
Code Issues Projects Releases Packages Wiki Activity

[PromoDJ] Add support for user's best media playlist

Browse source
This commit is contained in:
DmitryScaletta 2024-02-15 12:21:53 +03:00
parent c820715205
commit c837d90e12
No known key found for this signature in database
GPG key ID: 167A65222EDD4C2A
1 changed files with 23 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
yt_dlp/extractor/promodj.py
View file

@ -114,10 +114,6 @@ def _parse_page_content(self, html):
if YoutubeIE.suitable(iframe_url): if YoutubeIE.suitable(iframe_url):
yield self.url_result(iframe_url, YoutubeIE) yield self.url_result(iframe_url, YoutubeIE)
def _get_playlist_page_size(self, url):
is_default_playlist = '/groups/' not in url
return 30 if is_default_playlist else 20
def _get_current_page(self, html): def _get_current_page(self, html):
return int(clean_html(get_element_by_class('NavigatorCurrentPage', html)) or '1') return int(clean_html(get_element_by_class('NavigatorCurrentPage', html)) or '1')
@ -382,6 +378,7 @@ class PromoDJUserPageIE(PromoDJBaseIE):
'blog', 'blog',
'feedback', 'feedback',
'contact', 'contact',
'uenno',
*PromoDJBaseIE._MEDIA_TYPES, *PromoDJBaseIE._MEDIA_TYPES,
] ]
_NOT_USER_PAGE_RE = '|'.join(_USER_PAGES) _NOT_USER_PAGE_RE = '|'.join(_USER_PAGES)
@ -447,8 +444,11 @@ def _real_extract(self, url):
class PromoDJPlaylistIE(PromoDJBaseIE): class PromoDJPlaylistIE(PromoDJBaseIE):
_PLAYLIST_TYPES = ['uenno', *PromoDJBaseIE._MEDIA_TYPES]
_PLAYLIST_TYPES_RE = '|'.join(_PLAYLIST_TYPES)
_VALID_URL = [ _VALID_URL = [
rf'{PromoDJBaseIE._BASE_URL_RE}/(?P<login>{PromoDJBaseIE._LOGIN_RE})/(?P<type>{PromoDJBaseIE._MEDIA_TYPES_RE})$', rf'{PromoDJBaseIE._BASE_URL_RE}/(?P<login>{PromoDJBaseIE._LOGIN_RE})/(?P<type>{_PLAYLIST_TYPES_RE})$',
rf'{PromoDJBaseIE._BASE_URL_RE}/(?P<login>{PromoDJBaseIE._LOGIN_RE})/(?P<type>groups)/(?P<id>\d+)(?:/(?P<slug>\w+))?', rf'{PromoDJBaseIE._BASE_URL_RE}/(?P<login>{PromoDJBaseIE._LOGIN_RE})/(?P<type>groups)/(?P<id>\d+)(?:/(?P<slug>\w+))?',
] ]
_TESTS = [{ _TESTS = [{
@ -507,20 +507,36 @@ class PromoDJPlaylistIE(PromoDJBaseIE):
# 900+ items # 900+ items
'url': 'https://promodj.com/fonarev/groups/17350/Digital_Emotions_Podcast', 'url': 'https://promodj.com/fonarev/groups/17350/Digital_Emotions_Podcast',
'only_matching': True, 'only_matching': True,
}, {
# user's best music and video
'url': 'https://promodj.com/djbaribyn/uenno',
'info_dict': {
'id': 'djbaribyn-uenno',
},
'playlist_count': 15,
'params': {
'playlistend': 15,
}
}] }]
_ALLOWED_MEDIA_CATS = ['music', 'video'] _ALLOWED_MEDIA_CATS = ['music', 'video']
def _get_page_size(self, url):
if '/uenno' in url:
return 15
if '/groups/' in url:
return 30
return 20
def _real_extract(self, url): def _real_extract(self, url):
match = self._match_valid_url(url) match = self._match_valid_url(url)
login = match.group('login') login = match.group('login')
type = match.group('type') type = match.group('type')
playlist_id = f'{login}-{type}' if len(match.groups()) == 2 else f'{login}-{type}-{match.group("id")}' playlist_id = f'{login}-{type}' if len(match.groups()) == 2 else f'{login}-{type}-{match.group("id")}'
page_size = self._get_playlist_page_size(url)
entries = OnDemandPagedList( entries = OnDemandPagedList(
functools.partial(self._fetch_page, url, self._ALLOWED_MEDIA_CATS, playlist_id), functools.partial(self._fetch_page, url, self._ALLOWED_MEDIA_CATS, playlist_id),
page_size) self._get_page_size(url))
return self.playlist_result(entries, playlist_id=playlist_id) return self.playlist_result(entries, playlist_id=playlist_id)