youtube_dlc/extractor/sonyliv.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 import time
   5 import uuid
   6
   7 from .common import InfoExtractor
   8 from ..compat import compat_HTTPError
   9 from ..utils import (
  10     ExtractorError,
  11     int_or_none,
  12 )
  13
  14
  15 class SonyLIVIE(InfoExtractor):
  16     _VALID_URL = r'https?://(?:www\.)?sonyliv\.com/(?:s(?:how|port)s/[^/]+|movies|clip|trailer|music-videos)/[^/?#&]+-(?P<id>\d+)'
  17     _TESTS = [{
  18         'url': 'https://www.sonyliv.com/shows/bachelors-delight-1700000113/achaari-cheese-toast-1000022678?watch=true',
  19         'info_dict': {
  20             'title': 'Bachelors Delight - Achaari Cheese Toast',
  21             'id': '1000022678',
  22             'ext': 'mp4',
  23             'upload_date': '20200411',
  24             'description': 'md5:3957fa31d9309bf336ceb3f37ad5b7cb',
  25             'timestamp': 1586632091,
  26             'duration': 185,
  27             'season_number': 1,
  28             'episode': 'Achaari Cheese Toast',
  29             'episode_number': 1,
  30             'release_year': 2016,
  31         },
  32         'params': {
  33             'skip_download': True,
  34         },
  35     }, {
  36         'url': 'https://www.sonyliv.com/movies/tahalka-1000050121?watch=true',
  37         'only_matching': True,
  38     }, {
  39         'url': 'https://www.sonyliv.com/clip/jigarbaaz-1000098925',
  40         'only_matching': True,
  41     }, {
  42         'url': 'https://www.sonyliv.com/trailer/sandwiched-forever-1000100286?watch=true',
  43         'only_matching': True,
  44     }, {
  45         'url': 'https://www.sonyliv.com/sports/india-tour-of-australia-2020-21-1700000286/cricket-hls-day-3-1st-test-aus-vs-ind-19-dec-2020-1000100959?watch=true',
  46         'only_matching': True,
  47     }, {
  48         'url': 'https://www.sonyliv.com/music-videos/yeh-un-dinon-ki-baat-hai-1000018779',
  49         'only_matching': True,
  50     }]
  51     _GEO_COUNTRIES = ['IN']
  52     _TOKEN = None
  53
  54     def _call_api(self, version, path, video_id):
  55         headers = {}
  56         if self._TOKEN:
  57             headers['security_token'] = self._TOKEN
  58         try:
  59             return self._download_json(
  60                 'https://apiv2.sonyliv.com/AGL/%s/A/ENG/WEB/%s' % (version, path),
  61                 video_id, headers=headers)['resultObj']
  62         except ExtractorError as e:
  63             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
  64                 message = self._parse_json(
  65                     e.cause.read().decode(), video_id)['message']
  66                 if message == 'Geoblocked Country':
  67                     self.raise_geo_restricted(countries=self._GEO_COUNTRIES)
  68                 raise ExtractorError(message)
  69             raise
  70
  71     def _real_initialize(self):
  72         self._TOKEN = self._call_api('1.4', 'ALL/GETTOKEN', None)
  73
  74     def _real_extract(self, url):
  75         video_id = self._match_id(url)
  76         content = self._call_api(
  77             '1.5', 'IN/CONTENT/VIDEOURL/VOD/' + video_id, video_id)
  78         if content.get('isEncrypted'):
  79             raise ExtractorError('This video is DRM protected.', expected=True)
  80         dash_url = content['videoURL']
  81         headers = {
  82             'x-playback-session-id': '%s-%d' % (uuid.uuid4().hex, time.time() * 1000)
  83         }
  84         formats = self._extract_mpd_formats(
  85             dash_url, video_id, mpd_id='dash', headers=headers, fatal=False)
  86         formats.extend(self._extract_m3u8_formats(
  87             dash_url.replace('.mpd', '.m3u8').replace('/DASH/', '/HLS/'),
  88             video_id, 'mp4', m3u8_id='hls', headers=headers, fatal=False))
  89         for f in formats:
  90             f.setdefault('http_headers', {}).update(headers)
  91         self._sort_formats(formats)
  92
  93         metadata = self._call_api(
  94             '1.6', 'IN/DETAIL/' + video_id, video_id)['containers'][0]['metadata']
  95         title = metadata['title']
  96         episode = metadata.get('episodeTitle')
  97         if episode and title != episode:
  98             title += ' - ' + episode
  99
 100         return {
 101             'id': video_id,
 102             'title': title,
 103             'formats': formats,
 104             'thumbnail': content.get('posterURL'),
 105             'description': metadata.get('longDescription') or metadata.get('shortDescription'),
 106             'timestamp': int_or_none(metadata.get('creationDate'), 1000),
 107             'duration': int_or_none(metadata.get('duration')),
 108             'season_number': int_or_none(metadata.get('season')),
 109             'episode': episode,
 110             'episode_number': int_or_none(metadata.get('episodeNumber')),
 111             'release_year': int_or_none(metadata.get('year')),
 112         }