mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-27 02:42:30 +00:00
86e5f3ed2e
Using https://github.com/asottile/pyupgrade 1. `__future__` imports and `coding: utf-8` were removed 2. Files were rewritten with `pyupgrade --py36-plus --keep-percent-format` 3. f-strings were cherry-picked from `pyupgrade --py36-plus` Extractors are left untouched (except removing header) to avoid unnecessary merge conflicts
135 lines
4.8 KiB
Python
135 lines
4.8 KiB
Python
import json
|
|
import re
|
|
|
|
from .brightcove import BrightcoveNewIE
|
|
from ..compat import (
|
|
compat_HTTPError,
|
|
compat_str,
|
|
)
|
|
from ..utils import (
|
|
ExtractorError,
|
|
try_get,
|
|
update_url_query,
|
|
)
|
|
|
|
|
|
class SevenPlusIE(BrightcoveNewIE):
|
|
IE_NAME = '7plus'
|
|
_VALID_URL = r'https?://(?:www\.)?7plus\.com\.au/(?P<path>[^?]+\?.*?\bepisode-id=(?P<id>[^&#]+))'
|
|
_TESTS = [{
|
|
'url': 'https://7plus.com.au/MTYS?episode-id=MTYS7-003',
|
|
'info_dict': {
|
|
'id': 'MTYS7-003',
|
|
'ext': 'mp4',
|
|
'title': 'S7 E3 - Wind Surf',
|
|
'description': 'md5:29c6a69f21accda7601278f81b46483d',
|
|
'uploader_id': '5303576322001',
|
|
'upload_date': '20171201',
|
|
'timestamp': 1512106377,
|
|
'series': 'Mighty Ships',
|
|
'season_number': 7,
|
|
'episode_number': 3,
|
|
'episode': 'Wind Surf',
|
|
},
|
|
'params': {
|
|
'skip_download': True,
|
|
}
|
|
}, {
|
|
'url': 'https://7plus.com.au/UUUU?episode-id=AUMS43-001',
|
|
'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):
|
|
path, episode_id = self._match_valid_url(url).groups()
|
|
|
|
headers = {}
|
|
if self.token:
|
|
headers['Authorization'] = f'Bearer {self.token}'
|
|
|
|
try:
|
|
media = self._download_json(
|
|
'https://videoservice.swm.digital/playback', episode_id, query={
|
|
'appId': '7plus',
|
|
'deviceType': 'web',
|
|
'platformType': 'web',
|
|
'accountId': 5303576322001,
|
|
'referenceId': 'ref:' + episode_id,
|
|
'deliveryId': 'csai',
|
|
'videoType': 'vod',
|
|
}, headers=headers)['media']
|
|
except ExtractorError as e:
|
|
if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
|
|
raise ExtractorError(self._parse_json(
|
|
e.cause.read().decode(), episode_id)[0]['error_code'], expected=True)
|
|
raise
|
|
|
|
for source in media.get('sources', {}):
|
|
src = source.get('src')
|
|
if not src:
|
|
continue
|
|
source['src'] = update_url_query(src, {'rule': ''})
|
|
|
|
info = self._parse_brightcove_metadata(media, episode_id)
|
|
|
|
content = self._download_json(
|
|
'https://component-cdn.swm.digital/content/' + path,
|
|
episode_id, headers={
|
|
'market-id': 4,
|
|
}, fatal=False) or {}
|
|
for item in content.get('items', {}):
|
|
if item.get('componentData', {}).get('componentType') == 'infoPanel':
|
|
for src_key, dst_key in [('title', 'title'), ('shortSynopsis', 'description')]:
|
|
value = item.get(src_key)
|
|
if value:
|
|
info[dst_key] = value
|
|
info['series'] = try_get(
|
|
item, lambda x: x['seriesLogo']['name'], compat_str)
|
|
mobj = re.search(r'^S(\d+)\s+E(\d+)\s+-\s+(.+)$', info['title'])
|
|
if mobj:
|
|
info.update({
|
|
'season_number': int(mobj.group(1)),
|
|
'episode_number': int(mobj.group(2)),
|
|
'episode': mobj.group(3),
|
|
})
|
|
|
|
return info
|