l1ving_youtube-dl/youtube_dl/extractor/heise.py

# coding: utf-8
from __future__ import unicode_literals

from .common import InfoExtractor
from ..utils import (
    determine_ext,
    int_or_none,
    parse_iso8601,
    xpath_text,
)

import re


class HeiseIE(InfoExtractor):
    _VALID_URL = r'https?://(?:www\.)?heise\.de/(?:[^/]+/)+[^/]+-(?P<id>[0-9]+)\.html'
    _TESTS = [{
        'url': 'http://www.heise.de/video/artikel/Podcast-c-t-uplink-3-3-Owncloud-Tastaturen-Peilsender-Smartphone-2404147.html',
        'md5': 'ffed432483e922e88545ad9f2f15d30e',
        'info_dict': {
            'id': '2404147',
            'ext': 'mp4',
            'title': "Podcast: c't uplink 3.3 – Owncloud / Tastaturen / Peilsender Smartphone",
            'format_id': 'mp4_720p',
            'timestamp': 1411812600,
            'upload_date': '20140927',
            'description': 'md5:c934cbfb326c669c2bcabcbe3d3fcd20',
            'thumbnail': r're:^https?://.*/gallery/$',
        }
    }, {
        'url': 'https://www.heise.de/ct/artikel/c-t-uplink-18-5-Android-Oreo-Nokia-Galaxy-Note-8-AMD-Ryzen-Threadripper-3812972.html',# noqa
        'info_dict': {
            'id': '3812972',
            'ext': 'mp4',
            'title': "c't uplink 18.5: Android Oreo, Nokia, Galaxy Note 8, AMD Ryzen Threadripper",# noqa
            'description': 'md5:0601ade34ae5c4f5058d378327928348'
        }
    }, {
        'url': 'http://www.heise.de/ct/artikel/c-t-uplink-3-3-Owncloud-Tastaturen-Peilsender-Smartphone-2403911.html',
        'only_matching': True,
    }, {
        'url': 'http://www.heise.de/newsticker/meldung/c-t-uplink-Owncloud-Tastaturen-Peilsender-Smartphone-2404251.html?wt_mc=rss.ho.beitrag.atom',
        'only_matching': True,
    }, {
        'url': 'http://www.heise.de/ct/ausgabe/2016-12-Spiele-3214137.html',
        'only_matching': True,
    }]

    def _real_extract(self, url):
        video_id = self._match_id(url)
        webpage = self._download_webpage(url, video_id)

        container_id = self._search_regex(
            r'<div class="videoplayerjw"[^>]+data-container="([0-9]+)"',
            webpage, 'container ID')
        sequenz_id = self._search_regex(
            r'<div class="videoplayerjw"[^>]+data-sequenz="([0-9]+)"',
            webpage, 'sequenz ID')

        title = self._html_search_meta('fulltitle', webpage, default=None)
        if not title or title == "c't":
            title = self._search_regex(
                r'<div[^>]+class="videoplayerjw"[^>]+data-title="([^"]+)"',
                webpage, 'title')

        # videout/feed still and exlusively works for older videos
        # for new ct uplink episodes, whe need the work-around below.
        try:
            doc = self._download_xml(
                'http://www.heise.de/videout/feed', video_id, query={
                    'container': container_id,
                    'sequenz': sequenz_id,
                })
        except Exception as e:
            if e.cause.code == 404:
                if title.rfind('c\'t') != -1:
                    return self.ctUplinkHelper(title, video_id)
            else:
                raise e

        formats = []
        for source_node in doc.findall('.//{http://rss.jwpcdn.com/}source'):
            label = source_node.attrib['label']
            height = int_or_none(self._search_regex(
                r'^(.*?_)?([0-9]+)p$', label, 'height', default=None))
            video_url = source_node.attrib['file']
            ext = determine_ext(video_url, '')
            formats.append({
                'url': video_url,
                'format_note': label,
                'format_id': '%s_%s' % (ext, label),
                'height': height,
            })
        self._sort_formats(formats)

        description = self._og_search_description(
            webpage, default=None) or self._html_search_meta(
            'description', webpage)

        return {
            'id': video_id,
            'title': title,
            'description': description,
            'thumbnail': (xpath_text(doc, './/{http://rss.jwpcdn.com/}image') or
                          self._og_search_thumbnail(webpage)),
            'timestamp': parse_iso8601(
                self._html_search_meta('date', webpage)),
            'formats': formats,
        }

    def ctUplinkHelper(self, title, video_id):
        # e.g. "18.5" from "c't uplink 18.5:"
        episode_str = re.findall(r'[0-9]{1,2}.[0-9]{1,2}', title)

        feeds = []
        for i, feed in enumerate([
                'https://www.heise.de/ct/uplink/ctuplink.rss',
                'https://www.heise.de/ct/uplink/ctuplinkvideo.rss',
                'https://www.heise.de/ct/uplink/ctuplinkvideohd.rss']):
            xml = self._download_xml(feed, video_id,
                "Downloading alternative XML (%s)" % (['audio', 'SD video', 'HD video'][i]),
                fatal=False)
            if xml is not False:
                feeds.append(xml)

        if len(feeds) == 0:
            return

        titles = feeds[0].findall('./channel/item/title')
        descriptions = feeds[0].findall('./channel/item/description')

        # try to find the real matching title.
        # only rely on the episode_str, e.g. "18.5".
        episode_index = -1
        for index, item in enumerate(titles):
            if titles[index].text.rfind(episode_str[0]) != -1:
                episode_index = index
                break

        # in case episode not found at all
        if episode_index == -1:
            return

        formats = []
        for feed in feeds:
            url = feed.findall('./channel/item/guid')[episode_index].text
            formats.append({
                'url': url,
                'ext': determine_ext(url, '')})

        self._sort_formats(formats)

        return {
            'id': video_id,
            'title': title,
            'description': descriptions[episode_index].text,
            'formats': formats
        }