[voot] sort formats
[youtube-dl] / youtube_dl / extractor / voot.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..utils import (
6     ExtractorError,
7     int_or_none,
8     try_get,
9     unified_timestamp,
10 )
11
12
13 class VootIE(InfoExtractor):
14     _VALID_URL = r'https?://(?:www\.)?voot\.com/(?:[^/]+/)+(?P<id>\d+)'
15     _GEO_COUNTRIES = ['IN']
16     _TESTS = [{
17         'url': 'https://www.voot.com/shows/ishq-ka-rang-safed/1/360558/is-this-the-end-of-kamini-/441353',
18         'info_dict': {
19             'id': '441353',
20             'ext': 'mp4',
21             'title': 'Ishq Ka Rang Safed - Season 01 - Episode 340',
22             'description': 'md5:06291fbbbc4dcbe21235c40c262507c1',
23             'timestamp': 1472162937,
24             'upload_date': '20160825',
25             'duration': 1146,
26             'series': 'Ishq Ka Rang Safed',
27             'season_number': 1,
28             'episode': 'Is this the end of Kamini?',
29             'episode_number': 340,
30             'view_count': int,
31             'like_count': int,
32         },
33         'params': {
34             'skip_download': True,
35         },
36         'expected_warnings': ['Failed to download m3u8 information'],
37     }, {
38         'url': 'https://www.voot.com/kids/characters/mighty-cat-masked-niyander-e-/400478/school-bag-disappears/440925',
39         'only_matching': True,
40     }, {
41         'url': 'https://www.voot.com/movies/pandavas-5/424627',
42         'only_matching': True,
43     }]
44
45     def _real_extract(self, url):
46         video_id = self._match_id(url)
47
48         media_info = self._download_json(
49             'https://wapi.voot.com/ws/ott/getMediaInfo.json', video_id,
50             query={
51                 'platform': 'Web',
52                 'pId': 2,
53                 'mediaId': video_id,
54             })
55
56         status_code = try_get(media_info, lambda x: x['status']['code'], int)
57         if status_code != 0:
58             raise ExtractorError(media_info['status']['message'], expected=True)
59
60         media = media_info['assets']
61
62         title = media['MediaName']
63         formats = self._extract_m3u8_formats(
64             'https://cdnapisec.kaltura.com/p/1982551/playManifest/pt/https/f/applehttp/t/web/e/' + media['EntryId'],
65             video_id, 'mp4', m3u8_id='hls')
66         self._sort_formats(formats)
67
68         description, series, season_number, episode, episode_number = [None] * 5
69
70         for meta in try_get(media, lambda x: x['Metas'], list) or []:
71             key, value = meta.get('Key'), meta.get('Value')
72             if not key or not value:
73                 continue
74             if key == 'ContentSynopsis':
75                 description = value
76             elif key == 'RefSeriesTitle':
77                 series = value
78             elif key == 'RefSeriesSeason':
79                 season_number = int_or_none(value)
80             elif key == 'EpisodeMainTitle':
81                 episode = value
82             elif key == 'EpisodeNo':
83                 episode_number = int_or_none(value)
84
85         return {
86             'id': video_id,
87             'title': title,
88             'description': description,
89             'series': series,
90             'season_number': season_number,
91             'episode': episode,
92             'episode_number': episode_number,
93             'timestamp': unified_timestamp(media.get('CreationDate')),
94             'duration': int_or_none(media.get('Duration')),
95             'view_count': int_or_none(media.get('ViewCounter')),
96             'like_count': int_or_none(media.get('like_counter')),
97             'formats': formats,
98         }