]> git.bitcoin.ninja Git - youtube-dl/blob - youtube_dl/extractor/onionstudios.py
[brightcove:new] Add ability to smuggle geo_countries into URL
[youtube-dl] / youtube_dl / extractor / onionstudios.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..utils import (
8     determine_ext,
9     int_or_none,
10     float_or_none,
11     mimetype2ext,
12 )
13
14
15 class OnionStudiosIE(InfoExtractor):
16     _VALID_URL = r'https?://(?:www\.)?onionstudios\.com/(?:videos/[^/]+-|embed\?.*\bid=)(?P<id>\d+)(?!-)'
17
18     _TESTS = [{
19         'url': 'http://www.onionstudios.com/videos/hannibal-charges-forward-stops-for-a-cocktail-2937',
20         'md5': 'e49f947c105b8a78a675a0ee1bddedfe',
21         'info_dict': {
22             'id': '2937',
23             'ext': 'mp4',
24             'title': 'Hannibal charges forward, stops for a cocktail',
25             'thumbnail': r're:^https?://.*\.jpg$',
26             'uploader': 'The A.V. Club',
27             'uploader_id': 'the-av-club',
28         },
29     }, {
30         'url': 'http://www.onionstudios.com/embed?id=2855&autoplay=true',
31         'only_matching': True,
32     }]
33
34     @staticmethod
35     def _extract_url(webpage):
36         mobj = re.search(
37             r'<iframe[^>]+?src=(["\'])(?P<url>(?:https?:)?//(?:www\.)?onionstudios\.com/embed.+?)\1', webpage)
38         if mobj:
39             return mobj.group('url')
40
41     def _real_extract(self, url):
42         video_id = self._match_id(url)
43
44         video_data = self._download_json(
45             'http://www.onionstudios.com/video/%s.json' % video_id, video_id)
46
47         title = video_data['title']
48
49         formats = []
50         for source in video_data.get('sources', []):
51             source_url = source.get('url')
52             if not source_url:
53                 continue
54             ext = mimetype2ext(source.get('content_type')) or determine_ext(source_url)
55             if ext == 'm3u8':
56                 formats.extend(self._extract_m3u8_formats(
57                     source_url, video_id, 'mp4', 'm3u8_native', m3u8_id='hls', fatal=False))
58             else:
59                 tbr = int_or_none(source.get('bitrate'))
60                 formats.append({
61                     'format_id': ext + ('-%d' % tbr if tbr else ''),
62                     'url': source_url,
63                     'width': int_or_none(source.get('width')),
64                     'tbr': tbr,
65                     'ext': ext,
66                 })
67         self._sort_formats(formats)
68
69         return {
70             'id': video_id,
71             'title': title,
72             'thumbnail': video_data.get('poster_url'),
73             'uploader': video_data.get('channel_name'),
74             'uploader_id': video_data.get('channel_slug'),
75             'duration': float_or_none(video_data.get('duration', 1000)),
76             'tags': video_data.get('tags'),
77             'formats': formats,
78         }