xZero707/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-30 07:28:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

simplify _call_api

Browse Source
This commit is contained in:
Mozi 2024-10-18 16:03:03 +00:00
parent 4e649c6a28
commit bb98db29a1
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
yt_dlp/extractor/fifa.py
View File

@ -52,17 +52,19 @@ def _real_initialize(self):
}).encode()) }).encode())
self._HEADERS['x-chili-device-id'] = device_info['id'] self._HEADERS['x-chili-device-id'] = device_info['id']
def _call_api(self, path, video_id, note=None, **kwargs): def _call_api(self, path, video_id, note=None, headers=None, query=None, data=None):
return self._download_json( return self._download_json(
f'https://www.plus.fifa.com/flux-capacitor/api/v1//{path}', video_id, note, **kwargs) f'https://www.plus.fifa.com/flux-capacitor/api/v1//{path}', video_id, note, headers={
**self._HEADERS,
**(headers or {}),
}, query=query, data=data)
def _real_extract(self, url): def _real_extract(self, url):
urlh = self._request_webpage(url, self._match_id(url)) urlh = self._request_webpage(url, self._match_id(url))
video_id, display_id, locale = self._match_valid_url(urlh.url).group('id', 'display_id', 'locale') video_id, display_id, locale = self._match_valid_url(urlh.url).group('id', 'display_id', 'locale')
video_info = self._call_api( video_info = self._call_api(
'videoasset', video_id, 'Downloading video asset', 'videoasset', video_id, 'Downloading video asset', query={'catalog': video_id})[0]
headers=self._HEADERS, query={'catalog': video_id})[0]
formats = [] formats = []
subtitles = {} subtitles = {}
@ -73,12 +75,12 @@ def _real_extract(self, url):
]: ]:
session_info = self._call_api( session_info = self._call_api(
'streaming/session', video_id, 'Getting streaming session', 'streaming/session', video_id, 'Getting streaming session',
headers={**self._HEADERS, 'x-chili-accept-stream': stream_type}, headers={'x-chili-accept-stream': stream_type},
data=json.dumps({'videoAssetId': video_info['id'], 'autoPlay': False}).encode()) data=json.dumps({'videoAssetId': video_info['id'], 'autoPlay': False}).encode())
streams_info = self._call_api( streams_info = self._call_api(
'streaming/urls', video_id, 'Getting streaming urls', 'streaming/urls', video_id, 'Getting streaming urls',
headers={**self._HEADERS, 'x-chili-streaming-session': session_info['id']}) headers={'x-chili-streaming-session': session_info['id']})
for playlist_url in traverse_obj(streams_info, (..., 'url')): for playlist_url in traverse_obj(streams_info, (..., 'url')):
ext = determine_ext(playlist_url) ext = determine_ext(playlist_url)
@ -115,8 +117,9 @@ def _preconnect_link(self):
r'<link\b[^>]+\brel\s*=\s*"preconnect"[^>]+href\s*=\s*"([^"]+)"', r'<link\b[^>]+\brel\s*=\s*"preconnect"[^>]+href\s*=\s*"([^"]+)"',
self._download_webpage('https://fifa.com/', None), 'Preconnect Link') self._download_webpage('https://fifa.com/', None), 'Preconnect Link')
def _call_api(self, path, video_id, note=None, **kwargs): def _call_api(self, path, video_id, note=None, query=None):
return self._download_json(f'{self._preconnect_link}/{path}', video_id, note, **kwargs) return self._download_json(
f'{self._preconnect_link}/{path}', video_id, note, query=query)
class FifaIE(FifaBaseIE): class FifaIE(FifaBaseIE):