xZero707/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-27 06:10:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

[PromoDJ] Rename media_types to allowed_media_cats

Browse Source
This commit is contained in:
DmitryScaletta 2024-02-14 04:40:18 +03:00
parent ca0be3f1c1
commit 13f116fce6
No known key found for this signature in database
GPG Key ID: 167A65222EDD4C2A
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
yt_dlp/extractor/promodj.py
View File

@ -71,7 +71,7 @@ def _set_url_page(self, url, page):
qs['page'] = page
return parsed_url._replace(query=urllib.parse.urlencode(qs, doseq=True)).geturl()
def _fetch_page(self, url, media_types, playlist_id, page):
def _fetch_page(self, url, allowed_media_cats, playlist_id, page):
page_url = self._set_url_page(url, page + 1)
html = self._download_webpage(page_url, f'{playlist_id}-page-{page + 1}')
current_page = int(clean_html(get_element_by_class('NavigatorCurrentPage', html)) or '1')
@ -84,7 +84,7 @@ def _fetch_page(self, url, media_types, playlist_id, page):
continue
url = url.replace('?play=1', '')
is_video = '/videos/' in url
if is_video and 'video' in media_types or not is_video and 'music' in media_types:
if is_video and 'video' in allowed_media_cats or not is_video and 'music' in allowed_media_cats:
yield self.url_result(url, PromoDJIE)
def _parse_playlist_links(self, html):
@ -345,7 +345,7 @@ class PromoDJPlaylistIE(PromoDJBaseIE):
'only_matching': True,
}]
_MEDIA_TYPES = ['music', 'video']
_ALLOWED_MEDIA_CATS = ['music', 'video']
def _real_extract(self, url):
match = self._match_valid_url(url)
@ -355,17 +355,17 @@ def _real_extract(self, url):
page_size = self._get_playlist_page_size(url)
entries = OnDemandPagedList(
functools.partial(self._fetch_page, url, self._MEDIA_TYPES, playlist_id),
functools.partial(self._fetch_page, url, self._ALLOWED_MEDIA_CATS, playlist_id),
page_size)
return self.playlist_result(entries, playlist_id=playlist_id)
class PromoDJMusicPlaylistIE(PromoDJPlaylistIE):
_MEDIA_TYPES = ['music']
_ALLOWED_MEDIA_CATS = ['music']
class PromoDJVideoPlaylistIE(PromoDJPlaylistIE):
_MEDIA_TYPES = ['video']
_ALLOWED_MEDIA_CATS = ['video']
class PromoDJIE(PromoDJBaseIE):