Cool_Tools/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2024-11-24 07:40:24 +01:00
Code Releases Activity

[keep2share] share access token across calls to _real_extract

Browse Source
This commit is contained in:
schn0sch 2021-04-24 14:03:17 +02:00
parent 5f3cc3bbea
commit 14bf10c648
1 changed files with 24 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
youtube_dl/extractor/keep2share.py
View File

@ -30,13 +30,13 @@ class Keep2ShareIE(InfoExtractor):
},
}]
def _real_extract(self, url):
video_id = self._match_id(url)
def _get_app_secret(self, video_id):
""" retrieve REACT_APP_API_CLIENT_SECRET """
# TODO share app_secret across calls to _real_extract
app_secret = None
if getattr(self, '_app_secret', None) is not None:
return self._app_secret
# retrieve app_secret
url = 'https://k2s.cc/file/' + video_id
webpage = self._download_webpage(url, video_id)
scripts = re.finditer(r'<script\s+src="(?P<src>/static/[^"]*)"', webpage)
for mobj in scripts:
@ -47,26 +47,35 @@ class Keep2ShareIE(InfoExtractor):
'string': r'"(?:[^"]|\\")*"' + '|' + r"'(?:[^']|\\')*'",
}, script, 'app secret', group='secret', default=None)
if secret is not None:
app_secret = self._parse_json(secret, video_id, transform_source=js_to_json)
break
else:
self._app_secret = self._parse_json(secret, video_id,
transform_source=js_to_json)
return self._app_secret
raise RegexNotFoundError('Unable to extract app secret')
# retrieve access_token
# TODO share access_token across calls fo _real_extract
def _get_access_token(self, video_id):
""" retrieve access_token """
if getattr(self, '_access_token', None) is not None:
return self._access_token
data = {
'grant_type': 'client_credentials',
'client_id': 'k2s_web_app',
'client_secret': app_secret,
'client_secret': self._get_app_secret(video_id),
}
data = json.dumps(data, separators=(',', ':')).encode('utf-8')
headers = {'Content-Type': 'application/json'}
tokens = self._download_json('https://api.k2s.cc/v1/auth/token',
video_id, data=data, headers=headers)
access_token = tokens['access_token']
self._access_token = tokens['access_token']
# retrieve API page
headers = {'Cookie': 'accessToken=' + access_token}
return self._access_token
def _real_extract(self, url):
video_id = self._match_id(url)
headers = {'Cookie': 'accessToken=' + self._get_access_token(url)}
info = self._download_json('https://api.k2s.cc/v1/files/' + video_id,
video_id, headers=headers)