mirror/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-03-09 12:50:23 -05:00
Code Activity

store device id to class var

Browse source
This commit is contained in:
grqx_wsl 2024-12-24 13:34:48 +13:00
parent ea7d7d85d3
commit 9aee035fdf
1 changed files with 16 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
yt_dlp/extractor/bahamut.py
View file

@ -11,6 +11,7 @@
class BahamutIE(InfoExtractor): class BahamutIE(InfoExtractor):
_VALID_URL = r'https?://ani\.gamer\.com\.tw/animeVideo\.php\?sn=(?P<id>\d+)' _VALID_URL = r'https?://ani\.gamer\.com\.tw/animeVideo\.php\?sn=(?P<id>\d+)'
_DEVICE_ID = None
# see anime_player.js # see anime_player.js
RATING_TO_AGE_LIMIT = { RATING_TO_AGE_LIMIT = {
@ -22,16 +23,19 @@ class BahamutIE(InfoExtractor):
6: 18, # age-gated, needs login 6: 18, # age-gated, needs login
} }
def _download_device_id(self, video_id):
return self._download_json(
'https://ani.gamer.com.tw/ajax/getdeviceid.php', video_id,
'Downloading device ID', 'Failed to download device ID',
impersonate=True, headers=self.geo_verification_headers())['deviceid']
def _real_extract(self, url): def _real_extract(self, url):
url, unsmuggled_data = unsmuggle_url(url, {}) url, unsmuggled_data = unsmuggle_url(url, {})
video_id = self._match_id(url) video_id = self._match_id(url)
device_id = ( if not self._DEVICE_ID:
self._DEVICE_ID = (
self._configuration_arg('device_id', [None], casesense=True)[0] self._configuration_arg('device_id', [None], casesense=True)[0]
or unsmuggled_data.get('device_id') or self._download_device_id(video_id))
or self._download_json(
'https://ani.gamer.com.tw/ajax/getdeviceid.php', video_id,
'Downloading device ID', 'Failed to download device ID',
impersonate=True, headers=self.geo_verification_headers())['deviceid'])
# TODO: extract metadata from webpage # TODO: extract metadata from webpage
metadata = {} metadata = {}
@ -51,10 +55,8 @@ def _real_extract(self, url):
if self._yes_playlist(playlist_id, video_id) and unsmuggled_data.get('extract_playlist') is not False: if self._yes_playlist(playlist_id, video_id) and unsmuggled_data.get('extract_playlist') is not False:
return self.playlist_result( return self.playlist_result(
(self.url_result( (self.url_result(
# it may be better to use self.cache for storing device_id
smuggle_url(f'https://ani.gamer.com.tw/animeVideo.php?sn={ep["videoSn"]}', { smuggle_url(f'https://ani.gamer.com.tw/animeVideo.php?sn={ep["videoSn"]}', {
'extract_playlist': False, 'extract_playlist': False,
'device_id': device_id,
}), ie=BahamutIE, }), ie=BahamutIE,
video_id=ep['videoSn'], thumbnail=ep.get('cover')) for ep in traverse_obj( video_id=ep['videoSn'], thumbnail=ep.get('cover')) for ep in traverse_obj(
api_result, api_result,
@ -76,7 +78,7 @@ def _real_extract(self, url):
'https://ani.gamer.com.tw/ajax/m3u8.php', video_id, 'https://ani.gamer.com.tw/ajax/m3u8.php', video_id,
note='Downloading m3u8 URL', errnote='Failed to download m3u8 URL', query={ note='Downloading m3u8 URL', errnote='Failed to download m3u8 URL', query={
'sn': video_id, 'sn': video_id,
'device': device_id, 'device': self._DEVICE_ID,
}, impersonate=True, headers=self.geo_verification_headers(), expected_status=400) }, impersonate=True, headers=self.geo_verification_headers(), expected_status=400)
formats_fatal = True formats_fatal = True
@ -87,10 +89,10 @@ def _real_extract(self, url):
self.raise_geo_restricted(metadata_available=True) self.raise_geo_restricted(metadata_available=True)
formats_fatal = False formats_fatal = False
elif error_code == 1007: elif error_code == 1007:
if unsmuggled_data.pop('device_id', None) is not None: if self._configuration_arg('device_id', casesense=True):
return self.url_result( # the passed device id may be wrong or expired
smuggle_url(f'https://ani.gamer.com.tw/animeVideo.php?sn={video_id}', self._DEVICE_ID = self._download_device_id(video_id)
unsmuggled_data), ie=BahamutIE, video_id=video_id) return self.url_result(url, ie=BahamutIE, video_id=video_id)
raise ExtractorError('Invalid device id!') raise ExtractorError('Invalid device id!')
elif error_code == 1017: elif error_code == 1017:
self.raise_login_required(metadata_available=True) self.raise_login_required(metadata_available=True)