]> gitweb @ CieloNegro.org - youtube-dl.git/blob - youtube_dl/extractor/vimple.py
[ign] Fix extraction of video in articles
[youtube-dl.git] / youtube_dl / extractor / vimple.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import base64
5 import re
6 import xml.etree.ElementTree
7 import zlib
8
9 from .common import InfoExtractor
10 from ..utils import int_or_none
11
12
13 class VimpleIE(InfoExtractor):
14     IE_DESC = 'Vimple.ru'
15     _VALID_URL = r'https?://(player.vimple.ru/iframe|vimple.ru)/(?P<id>[a-f0-9]{10,})'
16     _TESTS = [
17         # Quality: Large, from iframe
18         {
19             'url': 'http://player.vimple.ru/iframe/b132bdfd71b546d3972f9ab9a25f201c',
20             'info_dict': {
21                 'id': 'b132bdfd71b546d3972f9ab9a25f201c',
22                 'title': 'great-escape-minecraft.flv',
23                 'ext': 'mp4',
24                 'duration': 352,
25                 'webpage_url': 'http://vimple.ru/b132bdfd71b546d3972f9ab9a25f201c',
26             },
27         },
28         # Quality: Medium, from mainpage
29         {
30             'url': 'http://vimple.ru/a15950562888453b8e6f9572dc8600cd',
31             'info_dict': {
32                 'id': 'a15950562888453b8e6f9572dc8600cd',
33                 'title': 'DB 01',
34                 'ext': 'flv',
35                 'duration': 1484,
36                 'webpage_url': 'http://vimple.ru/a15950562888453b8e6f9572dc8600cd',
37             }
38         },
39     ]
40
41     def _real_extract(self, url):
42         mobj = re.match(self._VALID_URL, url)
43         video_id = mobj.group('id')
44
45         iframe_url = 'http://player.vimple.ru/iframe/%s' % video_id
46
47         iframe = self._download_webpage(
48             iframe_url, video_id,
49             note='Downloading iframe', errnote='unable to fetch iframe')
50         player_url = self._html_search_regex(
51             r'"(http://player.vimple.ru/flash/.+?)"', iframe, 'player url')
52
53         player = self._request_webpage(
54             player_url, video_id, note='Downloading swf player').read()
55
56         player = zlib.decompress(player[8:])
57
58         xml_pieces = re.findall(b'([a-zA-Z0-9 =+/]{500})', player)
59         xml_pieces = [piece[1:-1] for piece in xml_pieces]
60
61         xml_data = b''.join(xml_pieces)
62         xml_data = base64.b64decode(xml_data)
63
64         xml_data = xml.etree.ElementTree.fromstring(xml_data)
65
66         video = xml_data.find('Video')
67         quality = video.get('quality')
68         q_tag = video.find(quality.capitalize())
69
70         formats = [
71             {
72                 'url': q_tag.get('url'),
73                 'tbr': int(q_tag.get('bitrate')),
74                 'filesize': int(q_tag.get('filesize')),
75                 'format_id': quality,
76             },
77         ]
78
79         return {
80             'id': video_id,
81             'title': video.find('Title').text,
82             'formats': formats,
83             'thumbnail': video.find('Poster').get('url'),
84             'duration': int_or_none(video.get('duration')),
85             'webpage_url': video.find('Share').get('videoPageUrl'),
86         }