]> git.bitcoin.ninja Git - youtube-dl/blob - youtube_dl/extractor/cammodels.py
remove unnecessary assignment parenthesis
[youtube-dl] / youtube_dl / extractor / cammodels.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..compat import compat_str
6 from ..utils import (
7     ExtractorError,
8     int_or_none,
9 )
10
11
12 class CamModelsIE(InfoExtractor):
13     _VALID_URL = r'https?://(?:www\.)?cammodels\.com/cam/(?P<id>[^/?#&]+)'
14     _TESTS = [{
15         'url': 'https://www.cammodels.com/cam/AutumnKnight/',
16         'only_matching': True,
17     }]
18
19     def _real_extract(self, url):
20         user_id = self._match_id(url)
21
22         webpage = self._download_webpage(url, user_id)
23
24         manifest_root = self._html_search_regex(
25             r'manifestUrlRoot=([^&\']+)', webpage, 'manifest', default=None)
26
27         if not manifest_root:
28             ERRORS = (
29                 ("I'm offline, but let's stay connected", 'This user is currently offline'),
30                 ('in a private show', 'This user is in a private show'),
31             )
32             for pattern, message in ERRORS:
33                 if pattern in webpage:
34                     error = message
35                     expected = True
36                     break
37             else:
38                 error = 'Unable to find manifest URL root'
39                 expected = False
40             raise ExtractorError(error, expected=expected)
41
42         manifest = self._download_json(
43             '%s%s.json' % (manifest_root, user_id), user_id)
44
45         formats = []
46         for format_id, format_dict in manifest['formats'].items():
47             if not isinstance(format_dict, dict):
48                 continue
49             encodings = format_dict.get('encodings')
50             if not isinstance(encodings, list):
51                 continue
52             vcodec = format_dict.get('videoCodec')
53             acodec = format_dict.get('audioCodec')
54             for media in encodings:
55                 if not isinstance(media, dict):
56                     continue
57                 media_url = media.get('location')
58                 if not media_url or not isinstance(media_url, compat_str):
59                     continue
60
61                 format_id_list = [format_id]
62                 height = int_or_none(media.get('videoHeight'))
63                 if height is not None:
64                     format_id_list.append('%dp' % height)
65                 f = {
66                     'url': media_url,
67                     'format_id': '-'.join(format_id_list),
68                     'width': int_or_none(media.get('videoWidth')),
69                     'height': height,
70                     'vbr': int_or_none(media.get('videoKbps')),
71                     'abr': int_or_none(media.get('audioKbps')),
72                     'fps': int_or_none(media.get('fps')),
73                     'vcodec': vcodec,
74                     'acodec': acodec,
75                 }
76                 if 'rtmp' in format_id:
77                     f['ext'] = 'flv'
78                 elif 'hls' in format_id:
79                     f.update({
80                         'ext': 'mp4',
81                         # hls skips fragments, preferring rtmp
82                         'preference': -1,
83                     })
84                 else:
85                     continue
86                 formats.append(f)
87         self._sort_formats(formats)
88
89         return {
90             'id': user_id,
91             'title': self._live_title(user_id),
92             'is_live': True,
93             'formats': formats,
94         }