mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-10-31 23:02:40 +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
248 lines
11 KiB
Python
248 lines
11 KiB
Python
import json
|
|
|
|
from .common import InfoExtractor
|
|
from ..compat import compat_str
|
|
from ..utils import (
|
|
ExtractorError,
|
|
int_or_none,
|
|
parse_age_limit,
|
|
str_or_none,
|
|
try_get,
|
|
unified_strdate,
|
|
unified_timestamp,
|
|
url_or_none,
|
|
)
|
|
|
|
|
|
class Zee5IE(InfoExtractor):
|
|
_VALID_URL = r'''(?x)
|
|
(?:
|
|
zee5:|
|
|
https?://(?:www\.)?zee5\.com/(?:[^#?]+/)?
|
|
(?:
|
|
(?:tv-shows|kids|web-series|zee5originals)(?:/[^#/?]+){3}
|
|
|movies/[^#/?]+
|
|
)/(?P<display_id>[^#/?]+)/
|
|
)
|
|
(?P<id>[^#/?]+)/?(?:$|[?#])
|
|
'''
|
|
_TESTS = [{
|
|
'url': 'https://www.zee5.com/movies/details/krishna-the-birth/0-0-63098',
|
|
'info_dict': {
|
|
'id': '0-0-63098',
|
|
'ext': 'mp4',
|
|
'display_id': 'krishna-the-birth',
|
|
'title': 'Krishna - The Birth',
|
|
'duration': 4368,
|
|
'description': compat_str,
|
|
'alt_title': 'Krishna - The Birth',
|
|
'uploader': 'Zee Entertainment Enterprises Ltd',
|
|
'release_date': '20060101',
|
|
'upload_date': '20060101',
|
|
'timestamp': 1136073600,
|
|
'thumbnail': r're:^https?://.*\.jpg$',
|
|
'episode_number': 0,
|
|
'episode': 'Episode 0',
|
|
'tags': list
|
|
},
|
|
'params': {
|
|
'format': 'bv',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/kids/kids-shows/bandbudh-aur-budbak/0-6-1899/yoga-se-hoga-bandbudh-aur-budbak/0-1-239839',
|
|
'info_dict': {
|
|
'id': '0-1-239839',
|
|
'ext': 'mp4',
|
|
'display_id': 'yoga-se-hoga-bandbudh-aur-budbak',
|
|
'title': 'Yoga Se Hoga-Bandbudh aur Budbak',
|
|
'duration': 659,
|
|
'description': compat_str,
|
|
'alt_title': 'Yoga Se Hoga-Bandbudh aur Budbak',
|
|
'uploader': 'Zee Entertainment Enterprises Ltd',
|
|
'release_date': '20150101',
|
|
'upload_date': '20150101',
|
|
'timestamp': 1420070400,
|
|
'thumbnail': r're:^https?://.*\.jpg$',
|
|
'series': 'Bandbudh Aur Budbak',
|
|
'season_number': 1,
|
|
'episode_number': 1,
|
|
'episode': 'Episode 1',
|
|
'season': 'Season 1',
|
|
'tags': list,
|
|
},
|
|
'params': {
|
|
'format': 'bv',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/hi/tv-shows/details/kundali-bhagya/0-6-366/kundali-bhagya-march-08-2021/0-1-manual_7g9jv1os7730?country=IN',
|
|
'only_matching': True
|
|
}, {
|
|
'url': 'https://www.zee5.com/global/hi/tv-shows/details/kundali-bhagya/0-6-366/kundali-bhagya-march-08-2021/0-1-manual_7g9jv1os7730',
|
|
'only_matching': True
|
|
}, {
|
|
'url': 'https://www.zee5.com/web-series/details/mithya/0-6-4z587408/maine-dekhi-hai-uski-mrityu/0-1-6z587412',
|
|
'only_matching': True
|
|
}]
|
|
_DETAIL_API_URL = 'https://spapi.zee5.com/singlePlayback/getDetails/secure?content_id={}&device_id={}&platform_name=desktop_web&country=IN&check_parental_control=false'
|
|
_DEVICE_ID = 'TszZPYPuY9Pq2cJizV0U000000000000'
|
|
_USER_TOKEN = None
|
|
_LOGIN_HINT = 'Use "--username <mobile_number>" to login using otp or "--username token" and "--password <user_token>" to login using user token.'
|
|
_NETRC_MACHINE = 'zee5'
|
|
_GEO_COUNTRIES = ['IN']
|
|
|
|
def _perform_login(self, username, password):
|
|
if len(username) == 10 and username.isdigit() and self._USER_TOKEN is None:
|
|
self.report_login()
|
|
otp_request_json = self._download_json(f'https://b2bapi.zee5.com/device/sendotp_v1.php?phoneno=91{username}',
|
|
None, note='Sending OTP')
|
|
if otp_request_json['code'] == 0:
|
|
self.to_screen(otp_request_json['message'])
|
|
else:
|
|
raise ExtractorError(otp_request_json['message'], expected=True)
|
|
otp_code = self._get_tfa_info('OTP')
|
|
otp_verify_json = self._download_json(f'https://b2bapi.zee5.com/device/verifyotp_v1.php?phoneno=91{username}&otp={otp_code}&guest_token={self._DEVICE_ID}&platform=web',
|
|
None, note='Verifying OTP', fatal=False)
|
|
if not otp_verify_json:
|
|
raise ExtractorError('Unable to verify OTP.', expected=True)
|
|
self._USER_TOKEN = otp_verify_json.get('token')
|
|
if not self._USER_TOKEN:
|
|
raise ExtractorError(otp_request_json['message'], expected=True)
|
|
elif username.lower() == 'token' and len(password) > 1198:
|
|
self._USER_TOKEN = password
|
|
else:
|
|
raise ExtractorError(self._LOGIN_HINT, expected=True)
|
|
|
|
def _real_extract(self, url):
|
|
video_id, display_id = self._match_valid_url(url).group('id', 'display_id')
|
|
access_token_request = self._download_json(
|
|
'https://useraction.zee5.com/token/platform_tokens.php?platform_name=web_app',
|
|
video_id, note='Downloading access token')
|
|
data = {
|
|
'x-access-token': access_token_request['token']
|
|
}
|
|
if self._USER_TOKEN:
|
|
data['Authorization'] = 'bearer %s' % self._USER_TOKEN
|
|
else:
|
|
data['X-Z5-Guest-Token'] = self._DEVICE_ID
|
|
|
|
json_data = self._download_json(
|
|
self._DETAIL_API_URL.format(video_id, self._DEVICE_ID),
|
|
video_id, headers={'content-type': 'application/json'}, data=json.dumps(data).encode('utf-8'))
|
|
asset_data = json_data['assetDetails']
|
|
show_data = json_data.get('showDetails', {})
|
|
if 'premium' in asset_data['business_type']:
|
|
raise ExtractorError('Premium content is DRM protected.', expected=True)
|
|
if not asset_data.get('hls_url'):
|
|
self.raise_login_required(self._LOGIN_HINT, metadata_available=True, method=None)
|
|
formats, m3u8_subs = self._extract_m3u8_formats_and_subtitles(asset_data['hls_url'], video_id, 'mp4', fatal=False)
|
|
self._sort_formats(formats)
|
|
|
|
subtitles = {}
|
|
for sub in asset_data.get('subtitle_url', []):
|
|
sub_url = sub.get('url')
|
|
if not sub_url:
|
|
continue
|
|
subtitles.setdefault(sub.get('language', 'en'), []).append({
|
|
'url': self._proto_relative_url(sub_url),
|
|
})
|
|
subtitles = self._merge_subtitles(subtitles, m3u8_subs)
|
|
return {
|
|
'id': video_id,
|
|
'display_id': display_id,
|
|
'title': asset_data['title'],
|
|
'formats': formats,
|
|
'subtitles': subtitles,
|
|
'duration': int_or_none(asset_data.get('duration')),
|
|
'description': str_or_none(asset_data.get('description')),
|
|
'alt_title': str_or_none(asset_data.get('original_title')),
|
|
'uploader': str_or_none(asset_data.get('content_owner')),
|
|
'age_limit': parse_age_limit(asset_data.get('age_rating')),
|
|
'release_date': unified_strdate(asset_data.get('release_date')),
|
|
'timestamp': unified_timestamp(asset_data.get('release_date')),
|
|
'thumbnail': url_or_none(asset_data.get('image_url')),
|
|
'series': str_or_none(asset_data.get('tvshow_name')),
|
|
'season': try_get(show_data, lambda x: x['seasons']['title'], str),
|
|
'season_number': int_or_none(try_get(show_data, lambda x: x['seasons'][0]['orderid'])),
|
|
'episode_number': int_or_none(try_get(asset_data, lambda x: x['orderid'])),
|
|
'tags': try_get(asset_data, lambda x: x['tags'], list)
|
|
}
|
|
|
|
|
|
class Zee5SeriesIE(InfoExtractor):
|
|
IE_NAME = 'zee5:series'
|
|
_VALID_URL = r'''(?x)
|
|
(?:
|
|
zee5:series:|
|
|
https?://(?:www\.)?zee5\.com/(?:[^#?]+/)?
|
|
(?:tv-shows|web-series|kids|zee5originals)(?:/[^#/?]+){2}/
|
|
)
|
|
(?P<id>[^#/?]+)(?:/episodes)?/?(?:$|[?#])
|
|
'''
|
|
_TESTS = [{
|
|
'url': 'https://www.zee5.com/kids/kids-shows/bandbudh-aur-budbak/0-6-1899',
|
|
'playlist_mincount': 156,
|
|
'info_dict': {
|
|
'id': '0-6-1899',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/tv-shows/details/bhabi-ji-ghar-par-hai/0-6-199',
|
|
'playlist_mincount': 1500,
|
|
'info_dict': {
|
|
'id': '0-6-199',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/tv-shows/details/agent-raghav-crime-branch/0-6-965',
|
|
'playlist_mincount': 24,
|
|
'info_dict': {
|
|
'id': '0-6-965',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/ta/tv-shows/details/nagabhairavi/0-6-3201',
|
|
'playlist_mincount': 3,
|
|
'info_dict': {
|
|
'id': '0-6-3201',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/global/hi/tv-shows/details/khwaabon-ki-zamin-par/0-6-270',
|
|
'playlist_mincount': 150,
|
|
'info_dict': {
|
|
'id': '0-6-270',
|
|
},
|
|
}, {
|
|
'url': 'https://www.zee5.com/tv-shows/details/chala-hawa-yeu-dya-ladies-zindabaad/0-6-2943/episodes',
|
|
'only_matching': True,
|
|
}, {
|
|
'url': 'https://www.zee5.com/web-series/details/mithya/0-6-4z587408',
|
|
'only_matching': True,
|
|
}]
|
|
|
|
def _entries(self, show_id):
|
|
access_token_request = self._download_json(
|
|
'https://useraction.zee5.com/token/platform_tokens.php?platform_name=web_app',
|
|
show_id, note='Downloading access token')
|
|
headers = {
|
|
'X-Access-Token': access_token_request['token'],
|
|
'Referer': 'https://www.zee5.com/',
|
|
}
|
|
show_url = f'https://gwapi.zee5.com/content/tvshow/{show_id}?translation=en&country=IN'
|
|
|
|
page_num = 0
|
|
show_json = self._download_json(show_url, video_id=show_id, headers=headers)
|
|
for season in show_json.get('seasons') or []:
|
|
season_id = try_get(season, lambda x: x['id'], compat_str)
|
|
next_url = f'https://gwapi.zee5.com/content/tvshow/?season_id={season_id}&type=episode&translation=en&country=IN&on_air=false&asset_subtype=tvshow&page=1&limit=100'
|
|
while next_url:
|
|
page_num += 1
|
|
episodes_json = self._download_json(
|
|
next_url, video_id=show_id, headers=headers,
|
|
note='Downloading JSON metadata page %d' % page_num)
|
|
for episode in try_get(episodes_json, lambda x: x['episode'], list) or []:
|
|
video_id = episode.get('id')
|
|
yield self.url_result(
|
|
'zee5:%s' % video_id,
|
|
ie=Zee5IE.ie_key(), video_id=video_id)
|
|
next_url = url_or_none(episodes_json.get('next_episode_api'))
|
|
|
|
def _real_extract(self, url):
|
|
show_id = self._match_id(url)
|
|
return self.playlist_result(self._entries(show_id), playlist_id=show_id)
|