Raphaël G. Git Repositories - youtubedl/blob - 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': '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         }