artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-04-29 03:55:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add _call_api method

Browse Source
This commit is contained in:
sepro 2024-11-15 05:22:17 +01:00
parent 7f88140dbe
commit 9c32f15b7b
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
yt_dlp/extractor/bandlab.py
View File

@ -15,11 +15,17 @@ from ..utils.traversal import traverse_obj, value
class BandlabBaseIE(InfoExtractor): class BandlabBaseIE(InfoExtractor):
_API_HEADERS = { _API_HEADERS = {
'accept': 'application/json',
'referer': 'https://www.bandlab.com/', 'referer': 'https://www.bandlab.com/',
'x-client-id': 'BandLab-Web', 'x-client-id': 'BandLab-Web',
'x-client-version': '10.1.123', 'x-client-version': '10.1.124',
} }
def _call_api(self, endpoint, asset_id, **kwargs):
return self._download_json(
f'https://www.bandlab.com/api/v1.3/{endpoint}/{asset_id}',
asset_id, headers=self._API_HEADERS, **kwargs)
def _parse_revision(self, revision_data, url=None): def _parse_revision(self, revision_data, url=None):
return { return {
'vcodec': 'none', 'vcodec': 'none',
@ -282,9 +288,8 @@ class BandlabIE(BandlabBaseIE):
revision_data = None revision_data = None
if not revision_id: if not revision_id:
post_data = self._download_json( post_data = self._call_api(
f'https://www.bandlab.com/api/v1.3/posts/{display_id}', display_id, 'posts', display_id, note='Downloading post data',
note='Downloading post data', headers=self._API_HEADERS,
query=traverse_obj(qs, {'sharedKey': ('sharedKey', 0)})) query=traverse_obj(qs, {'sharedKey': ('sharedKey', 0)}))
revision_id = traverse_obj(post_data, (('revisionId', ('revision', 'id')), {str}, any)) revision_id = traverse_obj(post_data, (('revisionId', ('revision', 'id')), {str}, any))
@ -299,9 +304,8 @@ class BandlabIE(BandlabBaseIE):
raise ExtractorError('Could not extract data') raise ExtractorError('Could not extract data')
if not revision_data: if not revision_data:
revision_data = self._download_json( revision_data = self._call_api(
f'https://www.bandlab.com/api/v1.3/revisions/{revision_id}', revision_id, 'revisions', revision_id, note='Downloading revision data', query={'edit': 'false'})
note='Downloading revision data', headers=self._API_HEADERS, query={'edit': 'false'})
return self._parse_revision(revision_data, url=url) return self._parse_revision(revision_data, url=url)
@ -405,9 +409,8 @@ class BandlabPlaylistIE(BandlabBaseIE):
'embed': ['collections', 'albums'], 'embed': ['collections', 'albums'],
}.get(playlist_type) }.get(playlist_type)
for endpoint in endpoints: for endpoint in endpoints:
playlist_data = self._download_json( playlist_data = self._call_api(
f'https://www.bandlab.com/api/v1.3/{endpoint}/{playlist_id}', playlist_id, endpoint, playlist_id, note=f'Downloading {endpoint[:-1]} data',
note=f'Downloading {endpoint[:-1]} data', headers=self._API_HEADERS,
fatal=False, expected_status=404) fatal=False, expected_status=404)
if not playlist_data.get('errorCode'): if not playlist_data.get('errorCode'):
playlist_type = endpoint playlist_type = endpoint