[vk] Use unicode_literals
[youtube-dl] / youtube_dl / extractor / vk.py
1 # encoding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5 import json
6
7 from .common import InfoExtractor
8 from ..utils import (
9     compat_str,
10     unescapeHTML,
11 )
12
13
14 class VKIE(InfoExtractor):
15     IE_NAME = 'vk.com'
16     _VALID_URL = r'https?://vk\.com/(?:videos.*?\?.*?z=)?video(?P<id>.*?)(?:\?|%2F|$)'
17
18     _TEST = {
19         'url': 'http://vk.com/videos-77521?z=video-77521_162222515%2Fclub77521',
20         'file': '162222515.flv',
21         'md5': '0deae91935c54e00003c2a00646315f0',
22         'info_dict': {
23             'title': 'ProtivoGunz - Хуёвая песня',
24             'uploader': 'Noize MC',
25         },
26     }
27
28     def _real_extract(self, url):
29         mobj = re.match(self._VALID_URL, url)
30         video_id = mobj.group('id')
31         info_url = 'http://vk.com/al_video.php?act=show&al=1&video=%s' % video_id
32         info_page = self._download_webpage(info_url, video_id)
33         m_yt = re.search(r'src="(http://www.youtube.com/.*?)"', info_page)
34         if m_yt is not None:
35             self.to_screen(u'Youtube video detected')
36             return self.url_result(m_yt.group(1), 'Youtube')
37         data_json = self._search_regex(r'var vars = ({.*?});', info_page, 'vars')
38         data = json.loads(data_json)
39
40         return {
41             'id': compat_str(data['vid']),
42             'url': data['url240'],
43             'title': unescapeHTML(data['md_title']),
44             'thumbnail': data['jpg'],
45             'uploader': data['md_author'],
46         }