mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-06-28 01:18:30 +00:00
[ie/iprima] Fix login support (#12937)
Closes #12387 Authored by: baierjan
This commit is contained in:
parent
586b557b12
commit
a7d9a5eb79
@ -1,3 +1,4 @@
|
|||||||
|
import json
|
||||||
import re
|
import re
|
||||||
import time
|
import time
|
||||||
|
|
||||||
@ -6,9 +7,7 @@
|
|||||||
ExtractorError,
|
ExtractorError,
|
||||||
determine_ext,
|
determine_ext,
|
||||||
js_to_json,
|
js_to_json,
|
||||||
parse_qs,
|
|
||||||
traverse_obj,
|
traverse_obj,
|
||||||
urlencode_postdata,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ -16,7 +15,6 @@ class IPrimaIE(InfoExtractor):
|
|||||||
_VALID_URL = r'https?://(?!cnn)(?:[^/]+)\.iprima\.cz/(?:[^/]+/)*(?P<id>[^/?#&]+)'
|
_VALID_URL = r'https?://(?!cnn)(?:[^/]+)\.iprima\.cz/(?:[^/]+/)*(?P<id>[^/?#&]+)'
|
||||||
_GEO_BYPASS = False
|
_GEO_BYPASS = False
|
||||||
_NETRC_MACHINE = 'iprima'
|
_NETRC_MACHINE = 'iprima'
|
||||||
_AUTH_ROOT = 'https://auth.iprima.cz'
|
|
||||||
access_token = None
|
access_token = None
|
||||||
|
|
||||||
_TESTS = [{
|
_TESTS = [{
|
||||||
@ -86,48 +84,18 @@ def _perform_login(self, username, password):
|
|||||||
if self.access_token:
|
if self.access_token:
|
||||||
return
|
return
|
||||||
|
|
||||||
login_page = self._download_webpage(
|
|
||||||
f'{self._AUTH_ROOT}/oauth2/login', None, note='Downloading login page',
|
|
||||||
errnote='Downloading login page failed')
|
|
||||||
|
|
||||||
login_form = self._hidden_inputs(login_page)
|
|
||||||
|
|
||||||
login_form.update({
|
|
||||||
'_email': username,
|
|
||||||
'_password': password})
|
|
||||||
|
|
||||||
profile_select_html, login_handle = self._download_webpage_handle(
|
|
||||||
f'{self._AUTH_ROOT}/oauth2/login', None, data=urlencode_postdata(login_form),
|
|
||||||
note='Logging in')
|
|
||||||
|
|
||||||
# a profile may need to be selected first, even when there is only a single one
|
|
||||||
if '/profile-select' in login_handle.url:
|
|
||||||
profile_id = self._search_regex(
|
|
||||||
r'data-identifier\s*=\s*["\']?(\w+)', profile_select_html, 'profile id')
|
|
||||||
|
|
||||||
login_handle = self._request_webpage(
|
|
||||||
f'{self._AUTH_ROOT}/user/profile-select-perform/{profile_id}', None,
|
|
||||||
query={'continueUrl': '/user/login?redirect_uri=/user/'}, note='Selecting profile')
|
|
||||||
|
|
||||||
code = traverse_obj(login_handle.url, ({parse_qs}, 'code', 0))
|
|
||||||
if not code:
|
|
||||||
raise ExtractorError('Login failed', expected=True)
|
|
||||||
|
|
||||||
token_request_data = {
|
|
||||||
'scope': 'openid+email+profile+phone+address+offline_access',
|
|
||||||
'client_id': 'prima_sso',
|
|
||||||
'grant_type': 'authorization_code',
|
|
||||||
'code': code,
|
|
||||||
'redirect_uri': f'{self._AUTH_ROOT}/sso/auth-check'}
|
|
||||||
|
|
||||||
token_data = self._download_json(
|
token_data = self._download_json(
|
||||||
f'{self._AUTH_ROOT}/oauth2/token', None,
|
'https://ucet.iprima.cz/api/session/create', None,
|
||||||
note='Downloading token', errnote='Downloading token failed',
|
note='Logging in', errnote='Failed to log in',
|
||||||
data=urlencode_postdata(token_request_data))
|
data=json.dumps({
|
||||||
|
'email': username,
|
||||||
|
'password': password,
|
||||||
|
'deviceName': 'Windows Chrome',
|
||||||
|
}).encode(), headers={'content-type': 'application/json'})
|
||||||
|
|
||||||
self.access_token = token_data.get('access_token')
|
self.access_token = token_data['accessToken']['value']
|
||||||
if self.access_token is None:
|
if not self.access_token:
|
||||||
raise ExtractorError('Getting token failed', expected=True)
|
raise ExtractorError('Failed to fetch access token')
|
||||||
|
|
||||||
def _real_initialize(self):
|
def _real_initialize(self):
|
||||||
if not self.access_token:
|
if not self.access_token:
|
||||||
|
Loading…
Reference in New Issue
Block a user