kumi/yt-dlp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[youtube] Fix session index extraction and headers for non-web player clients (#526)

Browse source 
Fixes #522
This commit is contained in:
coletdjnz 2021-07-18 18:23:32 +12:00 committed by GitHub
parent 34917076ad
commit 314ee30548
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
yt_dlp/extractor/youtube.py
View file

@ -416,6 +416,10 @@ class YoutubeBaseInfoExtractor(InfoExtractor):
def _extract_client_name(self, ytcfg, default_client='WEB'): def _extract_client_name(self, ytcfg, default_client='WEB'):
return self._ytcfg_get_safe(ytcfg, lambda x: x['INNERTUBE_CLIENT_NAME'], compat_str, default_client) return self._ytcfg_get_safe(ytcfg, lambda x: x['INNERTUBE_CLIENT_NAME'], compat_str, default_client)
@staticmethod
def _extract_session_index(ytcfg):
return int_or_none(try_get(ytcfg, lambda x: x['SESSION_INDEX']))
def _extract_client_version(self, ytcfg, default_client='WEB'): def _extract_client_version(self, ytcfg, default_client='WEB'):
return self._ytcfg_get_safe(ytcfg, lambda x: x['INNERTUBE_CLIENT_VERSION'], compat_str, default_client) return self._ytcfg_get_safe(ytcfg, lambda x: x['INNERTUBE_CLIENT_VERSION'], compat_str, default_client)
@ -518,7 +522,7 @@ class YoutubeBaseInfoExtractor(InfoExtractor):
default='{}'), video_id, fatal=False) or {} default='{}'), video_id, fatal=False) or {}
def _generate_api_headers(self, ytcfg=None, identity_token=None, account_syncid=None, def _generate_api_headers(self, ytcfg=None, identity_token=None, account_syncid=None,
visitor_data=None, api_hostname=None, client='WEB'): visitor_data=None, api_hostname=None, client='WEB', session_index=None):
origin = 'https://' + (api_hostname if api_hostname else self._get_innertube_host(client)) origin = 'https://' + (api_hostname if api_hostname else self._get_innertube_host(client))
headers = { headers = {
'X-YouTube-Client-Name': compat_str( 'X-YouTube-Client-Name': compat_str(
@ -533,9 +537,10 @@ class YoutubeBaseInfoExtractor(InfoExtractor):
headers['X-Youtube-Identity-Token'] = identity_token headers['X-Youtube-Identity-Token'] = identity_token
if account_syncid: if account_syncid:
headers['X-Goog-PageId'] = account_syncid headers['X-Goog-PageId'] = account_syncid
session_index = try_get(ytcfg, lambda x: x['SESSION_INDEX'], compat_str) if session_index is None and ytcfg:
if account_syncid or session_index: session_index = self._extract_session_index(ytcfg)
headers['X-Goog-AuthUser'] = session_index or 0 if account_syncid or session_index is not None:
headers['X-Goog-AuthUser'] = session_index if session_index is not None else 0
if visitor_data: if visitor_data:
headers['X-Goog-Visitor-Id'] = visitor_data headers['X-Goog-Visitor-Id'] = visitor_data
auth = self._generate_sapisidhash_header(origin) auth = self._generate_sapisidhash_header(origin)
@ -2294,8 +2299,8 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
ytcfg = self._extract_ytcfg(video_id, webpage) or self._get_default_ytcfg() ytcfg = self._extract_ytcfg(video_id, webpage) or self._get_default_ytcfg()
identity_token = self._extract_identity_token(webpage, video_id) identity_token = self._extract_identity_token(webpage, video_id)
syncid = self._extract_account_syncid(ytcfg) syncid = self._extract_account_syncid(ytcfg)
headers = self._generate_api_headers(ytcfg, identity_token, syncid) session_index = self._extract_session_index(ytcfg)
headers = self._generate_api_headers(ytcfg, identity_token, syncid, session_index=session_index)
player_url = self._extract_player_url(ytcfg, webpage) player_url = self._extract_player_url(ytcfg, webpage)
player_client = self._configuration_arg('player_client', [''])[0] player_client = self._configuration_arg('player_client', [''])[0]
@ -2336,7 +2341,7 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
ytm_headers = self._generate_api_headers( ytm_headers = self._generate_api_headers(
ytm_cfg, identity_token, syncid, ytm_cfg, identity_token, syncid,
client=ytm_client) client=ytm_client, session_index=session_index)
ytm_query = {'videoId': video_id} ytm_query = {'videoId': video_id}
ytm_query.update(self._generate_player_context(sts)) ytm_query.update(self._generate_player_context(sts))
@ -2365,7 +2370,8 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
self.report_warning('Falling back to android client for player API.') self.report_warning('Falling back to android client for player API.')
yt_client = 'ANDROID' yt_client = 'ANDROID'
ytpcfg = {} ytpcfg = {}
ytp_headers = self._generate_api_headers(ytpcfg, identity_token, syncid, yt_client) ytp_headers = self._generate_api_headers(ytpcfg, identity_token, syncid,
client=yt_client, session_index=session_index)
yt_query = {'videoId': video_id} yt_query = {'videoId': video_id}
yt_query.update(self._generate_player_context(sts)) yt_query.update(self._generate_player_context(sts))
@ -2418,7 +2424,8 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
ytcfg_age = {} ytcfg_age = {}
ytage_headers = self._generate_api_headers( ytage_headers = self._generate_api_headers(
ytcfg_age, identity_token, syncid, client=yt_client) ytcfg_age, identity_token, syncid,
client=yt_client, session_index=session_index)
yt_age_query = {'videoId': video_id} yt_age_query = {'videoId': video_id}
yt_age_query.update(self._generate_player_context(sts)) yt_age_query.update(self._generate_player_context(sts))
pr = self._extract_response( pr = self._extract_response(