Raphaël G. Git Repositories - youtubedl/blob - youtube_dl/extractor/atvat.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 from .common import InfoExtractor
   5 from ..utils import (
   6     determine_ext,
   7     int_or_none,
   8     unescapeHTML,
   9 )
  10
  11
  12 class ATVAtIE(InfoExtractor):
  13     _VALID_URL = r'https?://(?:www\.)?atv\.at/(?:[^/]+/){2}(?P<id>[dv]\d+)'
  14     _TESTS = [{
  15         'url': 'http://atv.at/aktuell/di-210317-2005-uhr/v1698449/',
  16         'md5': 'c3b6b975fb3150fc628572939df205f2',
  17         'info_dict': {
  18             'id': '1698447',
  19             'ext': 'mp4',
  20             'title': 'DI, 21.03.17 | 20:05 Uhr 1/1',
  21         }
  22     }, {
  23         'url': 'http://atv.at/aktuell/meinrad-knapp/d8416/',
  24         'only_matching': True,
  25     }]
  26
  27     def _real_extract(self, url):
  28         display_id = self._match_id(url)
  29         webpage = self._download_webpage(url, display_id)
  30         video_data = self._parse_json(unescapeHTML(self._search_regex(
  31             [r'flashPlayerOptions\s*=\s*(["\'])(?P<json>(?:(?!\1).)+)\1',
  32              r'class="[^"]*jsb_video/FlashPlayer[^"]*"[^>]+data-jsb="(?P<json>[^"]+)"'],
  33             webpage, 'player data', group='json')),
  34             display_id)['config']['initial_video']
  35
  36         video_id = video_data['id']
  37         video_title = video_data['title']
  38
  39         parts = []
  40         for part in video_data.get('parts', []):
  41             part_id = part['id']
  42             part_title = part['title']
  43
  44             formats = []
  45             for source in part.get('sources', []):
  46                 source_url = source.get('src')
  47                 if not source_url:
  48                     continue
  49                 ext = determine_ext(source_url)
  50                 if ext == 'm3u8':
  51                     formats.extend(self._extract_m3u8_formats(
  52                         source_url, part_id, 'mp4', 'm3u8_native',
  53                         m3u8_id='hls', fatal=False))
  54                 else:
  55                     formats.append({
  56                         'format_id': source.get('delivery'),
  57                         'url': source_url,
  58                     })
  59             self._sort_formats(formats)
  60
  61             parts.append({
  62                 'id': part_id,
  63                 'title': part_title,
  64                 'thumbnail': part.get('preview_image_url'),
  65                 'duration': int_or_none(part.get('duration')),
  66                 'is_live': part.get('is_livestream'),
  67                 'formats': formats,
  68             })
  69
  70         return {
  71             '_type': 'multi_video',
  72             'id': video_id,
  73             'title': video_title,
  74             'entries': parts,
  75         }