youtube-dl/youtube_dl/extractor/dw.py

# coding: utf-8
from __future__ import unicode_literals

from .common import InfoExtractor
from ..utils import (
    int_or_none,
    unified_strdate,
)
from ..compat import compat_urlparse


class DWIE(InfoExtractor):
    IE_NAME = 'dw'
    _VALID_URL = r'https?://(?:www\.)?dw\.com/(?:[^/]+/)+(?:av|e)-(?P<id>\d+)'
    _SOURCES_URL = 'http://www.dw.com/playersources/%s-%s'
    _TESTS = [{
        # video
        'url': 'http://www.dw.com/en/intelligent-light/av-19112290',
        'md5': '7372046e1815c5a534b43f3c3c36e6e9',
        'info_dict': {
            'id': '19112290',
            'ext': 'mp4',
            'title': 'Intelligent light',
            'description': 'md5:90e00d5881719f2a6a5827cb74985af1',
            'upload_date': '20160605',
        }
    }, {
        # 720 video
        'url': 'http://www.dw.com/en/hamburg-imposes-germanys-first-diesel-ban/av-44026376',
        'md5': '09299d024c4003145247d02b45a2bd3f',
        'info_dict': {
            'id': '44026376',
            'ext': 'mp4',
            'title': 'Hamburg imposes Germany\'s first diesel ban',
            'description': 'md5:a455cf13a29a11d5abff2c9a57981198',
            'upload_date': '20180531',
        }
    }, {
        # audio
        'url': 'http://www.dw.com/en/worldlink-my-business/av-19111941',
        'md5': '2814c9a1321c3a51f8a7aeb067a360dd',
        'info_dict': {
            'id': '19111941',
            'ext': 'mp3',
            'title': 'WorldLink: My business',
            'description': 'md5:bc9ca6e4e063361e21c920c53af12405',
            'upload_date': '20160311',
        }
    }, {
        # DW documentaries, only last for one or two weeks
        'url': 'http://www.dw.com/en/documentaries-welcome-to-the-90s-2016-05-21/e-19220158-9798',
        'md5': '56b6214ef463bfb9a3b71aeb886f3cf1',
        'info_dict': {
            'id': '19274438',
            'ext': 'mp4',
            'title': 'Welcome to the 90s – Hip Hop',
            'description': 'Welcome to the 90s - The Golden Decade of Hip Hop',
            'upload_date': '20160521',
        },
        'skip': 'Video removed',
    }]

    def _real_extract(self, url):

        def extract_formats():
            media = self._download_json(self._SOURCES_URL % (player_type[0], media_id), media_id)
            extracted_formats = []
            for medium in media:
                file = medium.get('file')
                if file is not None:
                    extracted_formats.append({
                        'url': file
                    })
                    label = medium.get('label')
                    if label is not None:
                        extracted_formats[-1]['format_id'] = label
            return extracted_formats

        media_id = self._match_id(url)
        webpage = self._download_webpage(url, media_id)
        hidden_inputs = self._hidden_inputs(webpage)
        title = hidden_inputs.get('media_title')
        media_id = hidden_inputs.get('media_id') or media_id
        player_type = hidden_inputs.get('player_type')
        file_name = hidden_inputs.get('file_name')
        formats = []

        if player_type == 'video' and hidden_inputs.get('stream_file') == '1':
                formats = self._extract_smil_formats(
                    'http://www.dw.com/smil/v-%s' % media_id, media_id,
                    transform_source=lambda s: s.replace(
                        'rtmp://tv-od.dw.de/flash/',
                        'http://tv-download.dw.de/dwtv_video/flv/'))

        if not formats:
            formats = extract_formats() or [{'url': file_name}] if file_name else []
        self._sort_formats(formats)

        upload_date = hidden_inputs.get('display_date')
        if not upload_date:
            upload_date = self._html_search_regex(
                r'<span[^>]+class="date">([0-9.]+)\s*\|', webpage,
                'upload date', default=None)
            upload_date = unified_strdate(upload_date)

        return {
            'id': media_id,
            'title': title,
            'description': self._og_search_description(webpage),
            'thumbnail': hidden_inputs.get('preview_image'),
            'duration': int_or_none(hidden_inputs.get('file_duration')),
            'upload_date': upload_date,
            'formats': formats,
        }


class DWArticleIE(InfoExtractor):
    IE_NAME = 'dw:article'
    _VALID_URL = r'https?://(?:www\.)?dw\.com/(?:[^/]+/)+a-(?P<id>\d+)'
    _TEST = {
        'url': 'http://www.dw.com/en/no-hope-limited-options-for-refugees-in-idomeni/a-19111009',
        'md5': '8ca657f9d068bbef74d6fc38b97fc869',
        'info_dict': {
            'id': '19105868',
            'ext': 'mp4',
            'title': 'The harsh life of refugees in Idomeni',
            'description': 'md5:196015cc7e48ebf474db9399420043c7',
            'upload_date': '20160310',
        }
    }

    def _real_extract(self, url):
        article_id = self._match_id(url)
        webpage = self._download_webpage(url, article_id)
        hidden_inputs = self._hidden_inputs(webpage)
        media_id = hidden_inputs['media_id']
        media_path = self._search_regex(r'href="([^"]+av-%s)"\s+class="overlayLink"' % media_id, webpage, 'media url')
        media_url = compat_urlparse.urljoin(url, media_path)
        return self.url_result(media_url, 'DW', media_id)