[polskieradio] Add test with video
[youtube-dl] / youtube_dl / extractor / polskieradio.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..compat import (
8     compat_str,
9     compat_urllib_parse_unquote,
10 )
11 from ..utils import (
12     int_or_none,
13     strip_or_none,
14     unified_timestamp,
15 )
16
17
18 class PolskieRadioIE(InfoExtractor):
19     _VALID_URL = r'https?://(?:www\.)?polskieradio\.pl/\d+/\d+/Artykul/(?P<id>[0-9]+)'
20     _TESTS = [{
21         'url': 'http://www.polskieradio.pl/7/5102/Artykul/1587943,Prof-Andrzej-Nowak-o-historii-nie-da-sie-myslec-beznamietnie',
22         'info_dict': {
23             'id': '1587943',
24             'title': 'Prof. Andrzej Nowak: o historii nie da się myśleć beznamiętnie',
25             'description': 'md5:12f954edbf3120c5e7075e17bf9fc5c5',
26         },
27         'playlist': [{
28             'md5': '2984ee6ce9046d91fc233bc1a864a09a',
29             'info_dict': {
30                 'id': '1540576',
31                 'ext': 'mp3',
32                 'title': 'md5:d4623290d4ac983bf924061c75c23a0d',
33                 'timestamp': 1456594200,
34                 'upload_date': '20160227',
35                 'duration': 2364,
36             },
37         }],
38     }, {
39         'url': 'http://www.polskieradio.pl/265/5217/Artykul/1635803,Euro-2016-nie-ma-miejsca-na-blad-Polacy-graja-ze-Szwajcaria-o-cwiercfinal',
40         'info_dict': {
41             'id': '1635803',
42             'title': 'Euro 2016: nie ma miejsca na błąd. Polacy grają ze Szwajcarią o ćwierćfinał',
43             'description': 'md5:01cb7d0cad58664095d72b51a1ebada2',
44         },
45         'playlist_mincount': 12,
46     }, {
47         'url': 'http://polskieradio.pl/9/305/Artykul/1632955,Bardzo-popularne-slowo-remis',
48         'only_matching': True,
49     }, {
50         'url': 'http://www.polskieradio.pl/7/5102/Artykul/1587943',
51         'only_matching': True,
52     }, {
53         # with mp4 video
54         'url': 'http://www.polskieradio.pl/9/299/Artykul/1634903,Brexit-Leszek-Miller-swiat-sie-nie-zawali-Europa-bedzie-trwac-dalej',
55         'only_matching': True,
56     }]
57
58     def _real_extract(self, url):
59         playlist_id = self._match_id(url)
60
61         webpage = self._download_webpage(url, playlist_id)
62
63         content = self._search_regex(
64             r'(?s)<div[^>]+class="audio atarticle"[^>]*>(.+?)<script>',
65             webpage, 'content')
66
67         timestamp = unified_timestamp(self._html_search_regex(
68             r'(?s)<span[^>]+id="datetime2"[^>]*>(.+?)</span>',
69             webpage, 'timestamp', fatal=False))
70
71         entries = []
72
73         media_urls = set()
74
75         for data_media in re.findall(r'<[^>]+data-media=({[^>]+})', content):
76             media = self._parse_json(data_media, playlist_id, fatal=False)
77             if not media.get('file') or not media.get('desc'):
78                 continue
79             media_url = self._proto_relative_url(media['file'], 'http:')
80             if media_url in media_urls:
81                 continue
82             media_urls.add(media_url)
83             entries.append({
84                 'id': compat_str(media['id']),
85                 'url': media_url,
86                 'title': compat_urllib_parse_unquote(media['desc']),
87                 'duration': int_or_none(media.get('length')),
88                 'vcodec': 'none' if media.get('provider') == 'audio' else None,
89                 'timestamp': timestamp,
90             })
91
92         title = self._og_search_title(webpage).strip()
93         description = strip_or_none(self._og_search_description(webpage))
94
95         return self.playlist_result(entries, playlist_id, title, description)