pepepper/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl synced 2025-01-31 07:40:09 +09:00
Code Issues Projects Releases Wiki Activity

Added Cookie Functionality to Hotstar

Browse Source
This commit is contained in:
Dev Bali 2021-06-22 18:41:47 -07:00
parent 379f52a495
commit de796eaae9
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
youtube_dl/extractor/hotstar.py
View File

@ -20,22 +20,27 @@ from ..utils import (
str_or_none, str_or_none,
try_get, try_get,
url_or_none, url_or_none,
urlencode_postdata,
sanitized_Request
) )
class HotStarBaseIE(InfoExtractor): class HotStarBaseIE(InfoExtractor):
_AKAMAI_ENCRYPTION_KEY = b'\x05\xfc\x1a\x01\xca\xc9\x4b\xc4\x12\xfc\x53\x12\x07\x75\xf9\xee' _AKAMAI_ENCRYPTION_KEY = b'\x05\xfc\x1a\x01\xca\xc9\x4b\xc4\x12\xfc\x53\x12\x07\x75\xf9\xee'
def _call_api_impl(self, path, video_id, headers, query, data=None): def _call_api_impl(self, path, video_id, headers, query, data=None,method=None):
st = int(time.time()) st = int(time.time())
exp = st + 6000 exp = st + 6000
auth = 'st=%d~exp=%d~acl=/*' % (st, exp) auth = 'st=%d~exp=%d~acl=/*' % (st, exp)
auth += '~hmac=' + hmac.new(self._AKAMAI_ENCRYPTION_KEY, auth.encode(), hashlib.sha256).hexdigest() auth += '~hmac=' + hmac.new(self._AKAMAI_ENCRYPTION_KEY, auth.encode(), hashlib.sha256).hexdigest()
h = {'hotstarauth': auth} h = {'hotstarauth': auth}
h.update(headers) h.update(headers)
req = sanitized_Request('https://api.hotstar.com/' + path, data=data)
if method is not None: req.get_method = lambda: method
return self._download_json( return self._download_json(
'https://api.hotstar.com/' + path, req,video_id, headers=h, query=query, data=data)
video_id, headers=h, query=query, data=data)
def _call_api(self, path, video_id, query_name='contentId'): def _call_api(self, path, video_id, query_name='contentId'):
response = self._call_api_impl(path, video_id, { response = self._call_api_impl(path, video_id, {
@ -53,13 +58,15 @@ class HotStarBaseIE(InfoExtractor):
def _call_api_v2(self, path, video_id, headers, query=None, data=None): def _call_api_v2(self, path, video_id, headers, query=None, data=None):
h = {'X-Request-Id': compat_str(uuid.uuid4())} h = {'X-Request-Id': compat_str(uuid.uuid4())}
h.update(headers) h.update(headers)
try: try:
return self._call_api_impl( return self._call_api_impl(
path, video_id, h, query, data) path, video_id, h, query, data)
except ExtractorError as e: except ExtractorError as e:
if isinstance(e.cause, compat_HTTPError): if isinstance(e.cause, compat_HTTPError):
if e.cause.code == 402: if e.cause.code == 402:
self.raise_login_required() raise ExtractorError('This video is only available for registered users. You may want to use --cookies.', expected=True)
message = self._parse_json(e.cause.read().decode(), video_id)['message'] message = self._parse_json(e.cause.read().decode(), video_id)['message']
if message in ('Content not available in region', 'Country is not supported'): if message in ('Content not available in region', 'Country is not supported'):
raise self.raise_geo_restricted(message) raise self.raise_geo_restricted(message)
@ -135,6 +142,13 @@ class HotStarIE(HotStarBaseIE):
formats = [] formats = []
geo_restricted = False geo_restricted = False
for cookie in self._downloader.cookiejar:
if "hotstar" in cookie.domain:
if cookie.name == "userUP":
self._USER_TOKEN = cookie.value
elif cookie.name == "device-id":
self._DEVICE_ID = cookie.value
if not self._USER_TOKEN: if not self._USER_TOKEN:
self._DEVICE_ID = compat_str(uuid.uuid4()) self._DEVICE_ID = compat_str(uuid.uuid4())
self._USER_TOKEN = self._call_api_v2('um/v3/users', video_id, { self._USER_TOKEN = self._call_api_v2('um/v3/users', video_id, {