0
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-26 02:55:17 +00:00

[7plus] Add cookie based authentication (#1202)

Closes #1103
Authored by: nyuszika7h
This commit is contained in:
nyuszika7h 2021-10-18 03:34:56 +02:00 committed by GitHub
parent 693ec74401
commit e69585f8c6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,6 +1,7 @@
# coding: utf-8 # coding: utf-8
from __future__ import unicode_literals from __future__ import unicode_literals
import json
import re import re
from .brightcove import BrightcoveNewIE from .brightcove import BrightcoveNewIE
@ -42,9 +43,52 @@ class SevenPlusIE(BrightcoveNewIE):
'only_matching': True, 'only_matching': True,
}] }]
def _real_initialize(self):
self.token = None
cookies = self._get_cookies('https://7plus.com.au')
api_key = next((x for x in cookies if x.startswith('glt_')), '')[4:]
if not api_key: # Cookies are signed out, skip login
return
login_resp = self._download_json(
'https://login.7plus.com.au/accounts.getJWT', None, 'Logging in', fatal=False,
query={
'APIKey': api_key,
'sdk': 'js_latest',
'login_token': cookies[f'glt_{api_key}'].value,
'authMode': 'cookie',
'pageURL': 'https://7plus.com.au/',
'sdkBuild': '12471',
'format': 'json',
}) or {}
if 'errorMessage' in login_resp:
self.report_warning(f'Unable to login: 7plus said: {login_resp["errorMessage"]}')
return
id_token = login_resp.get('id_token')
if not id_token:
self.report_warning('Unable to login: Could not extract id token')
return
token_resp = self._download_json(
'https://7plus.com.au/auth/token', None, 'Getting auth token', fatal=False,
headers={'Content-Type': 'application/json'}, data=json.dumps({
'idToken': id_token,
'platformId': 'web',
'regSource': '7plus',
}).encode('utf-8')) or {}
self.token = token_resp.get('token')
if not self.token:
self.report_warning('Unable to log in: Could not extract auth token')
def _real_extract(self, url): def _real_extract(self, url):
path, episode_id = self._match_valid_url(url).groups() path, episode_id = self._match_valid_url(url).groups()
headers = {}
if self.token:
headers['Authorization'] = f'Bearer {self.token}'
try: try:
media = self._download_json( media = self._download_json(
'https://videoservice.swm.digital/playback', episode_id, query={ 'https://videoservice.swm.digital/playback', episode_id, query={
@ -55,7 +99,7 @@ def _real_extract(self, url):
'referenceId': 'ref:' + episode_id, 'referenceId': 'ref:' + episode_id,
'deliveryId': 'csai', 'deliveryId': 'csai',
'videoType': 'vod', 'videoType': 'vod',
})['media'] }, headers=headers)['media']
except ExtractorError as e: except ExtractorError as e:
if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403: if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
raise ExtractorError(self._parse_json( raise ExtractorError(self._parse_json(