Fix "invalid escape sequences" error on Python 3.6
[youtube-dl] / youtube_dl / extractor / indavideo.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..utils import (
6     int_or_none,
7     parse_age_limit,
8     parse_iso8601,
9 )
10
11
12 class IndavideoEmbedIE(InfoExtractor):
13     _VALID_URL = r'https?://(?:(?:embed\.)?indavideo\.hu/player/video/|assets\.indavideo\.hu/swf/player\.swf\?.*\b(?:v(?:ID|id))=)(?P<id>[\da-f]+)'
14     _TESTS = [{
15         'url': 'http://indavideo.hu/player/video/1bdc3c6d80/',
16         'md5': 'f79b009c66194acacd40712a6778acfa',
17         'info_dict': {
18             'id': '1837039',
19             'ext': 'mp4',
20             'title': 'Cicatánc',
21             'description': '',
22             'thumbnail': r're:^https?://.*\.jpg$',
23             'uploader': 'cukiajanlo',
24             'uploader_id': '83729',
25             'timestamp': 1439193826,
26             'upload_date': '20150810',
27             'duration': 72,
28             'age_limit': 0,
29             'tags': ['tánc', 'cica', 'cuki', 'cukiajanlo', 'newsroom'],
30         },
31     }, {
32         'url': 'http://embed.indavideo.hu/player/video/1bdc3c6d80?autostart=1&hide=1',
33         'only_matching': True,
34     }, {
35         'url': 'http://assets.indavideo.hu/swf/player.swf?v=fe25e500&vID=1bdc3c6d80&autostart=1&hide=1&i=1',
36         'only_matching': True,
37     }]
38
39     def _real_extract(self, url):
40         video_id = self._match_id(url)
41
42         video = self._download_json(
43             'http://amfphp.indavideo.hu/SYm0json.php/player.playerHandler.getVideoData/%s' % video_id,
44             video_id)['data']
45
46         title = video['title']
47
48         video_urls = video.get('video_files', [])
49         video_file = video.get('video_file')
50         if video:
51             video_urls.append(video_file)
52         video_urls = list(set(video_urls))
53
54         video_prefix = video_urls[0].rsplit('/', 1)[0]
55
56         for flv_file in video.get('flv_files', []):
57             flv_url = '%s/%s' % (video_prefix, flv_file)
58             if flv_url not in video_urls:
59                 video_urls.append(flv_url)
60
61         formats = [{
62             'url': video_url,
63             'height': int_or_none(self._search_regex(
64                 r'\.(\d{3,4})\.mp4(?:\?|$)', video_url, 'height', default=None)),
65         } for video_url in video_urls]
66         self._sort_formats(formats)
67
68         timestamp = video.get('date')
69         if timestamp:
70             # upload date is in CEST
71             timestamp = parse_iso8601(timestamp + ' +0200', ' ')
72
73         thumbnails = [{
74             'url': self._proto_relative_url(thumbnail)
75         } for thumbnail in video.get('thumbnails', [])]
76
77         tags = [tag['title'] for tag in video.get('tags') or []]
78
79         return {
80             'id': video.get('id') or video_id,
81             'title': title,
82             'description': video.get('description'),
83             'thumbnails': thumbnails,
84             'uploader': video.get('user_name'),
85             'uploader_id': video.get('user_id'),
86             'timestamp': timestamp,
87             'duration': int_or_none(video.get('length')),
88             'age_limit': parse_age_limit(video.get('age_limit')),
89             'tags': tags,
90             'formats': formats,
91         }
92
93
94 class IndavideoIE(InfoExtractor):
95     _VALID_URL = r'https?://(?:.+?\.)?indavideo\.hu/video/(?P<id>[^/#?]+)'
96     _TESTS = [{
97         'url': 'http://indavideo.hu/video/Vicces_cica_1',
98         'md5': '8c82244ba85d2a2310275b318eb51eac',
99         'info_dict': {
100             'id': '1335611',
101             'display_id': 'Vicces_cica_1',
102             'ext': 'mp4',
103             'title': 'Vicces cica',
104             'description': 'Játszik a tablettel. :D',
105             'thumbnail': r're:^https?://.*\.jpg$',
106             'uploader': 'Jet_Pack',
107             'uploader_id': '491217',
108             'timestamp': 1390821212,
109             'upload_date': '20140127',
110             'duration': 7,
111             'age_limit': 0,
112             'tags': ['vicces', 'macska', 'cica', 'ügyes', 'nevetés', 'játszik', 'Cukiság', 'Jet_Pack'],
113         },
114     }, {
115         'url': 'http://index.indavideo.hu/video/2015_0728_beregszasz',
116         'only_matching': True,
117     }, {
118         'url': 'http://auto.indavideo.hu/video/Sajat_utanfutoban_a_kis_tacsko',
119         'only_matching': True,
120     }, {
121         'url': 'http://erotika.indavideo.hu/video/Amator_tini_punci',
122         'only_matching': True,
123     }, {
124         'url': 'http://film.indavideo.hu/video/f_hrom_nagymamm_volt',
125         'only_matching': True,
126     }, {
127         'url': 'http://palyazat.indavideo.hu/video/Embertelen_dal_Dodgem_egyuttes',
128         'only_matching': True,
129     }]
130
131     def _real_extract(self, url):
132         display_id = self._match_id(url)
133
134         webpage = self._download_webpage(url, display_id)
135         embed_url = self._search_regex(
136             r'<link[^>]+rel="video_src"[^>]+href="(.+?)"', webpage, 'embed url')
137
138         return {
139             '_type': 'url_transparent',
140             'ie_key': 'IndavideoEmbed',
141             'url': embed_url,
142             'display_id': display_id,
143         }