]>
Commit | Line | Data |
---|---|---|
bfd14b1b | 1 | # encoding: utf-8 |
1547c8cc | 2 | from __future__ import unicode_literals |
3 | ||
bfd14b1b JMF |
4 | import re |
5 | import json | |
1547c8cc | 6 | import itertools |
bfd14b1b JMF |
7 | |
8 | from .common import InfoExtractor | |
9 | from ..utils import ( | |
bfd14b1b | 10 | compat_str, |
a2fb2a21 | 11 | unified_strdate, |
bfd14b1b JMF |
12 | ExtractorError, |
13 | ) | |
14 | ||
15 | ||
16 | class RutubeIE(InfoExtractor): | |
1547c8cc | 17 | IE_NAME = 'rutube' |
18 | IE_DESC = 'Rutube videos' | |
c0ade33e | 19 | _VALID_URL = r'https?://rutube\.ru/video/(?P<long_id>\w+)' |
bfd14b1b JMF |
20 | |
21 | _TEST = { | |
1547c8cc | 22 | 'url': 'http://rutube.ru/video/3eac3b4561676c17df9132a9a1e62e3e/', |
23 | 'file': '3eac3b4561676c17df9132a9a1e62e3e.mp4', | |
24 | 'info_dict': { | |
25 | 'title': 'Раненный кенгуру забежал в аптеку', | |
26 | 'uploader': 'NTDRussian', | |
27 | 'uploader_id': '29790', | |
bfd14b1b | 28 | }, |
1547c8cc | 29 | 'params': { |
bfd14b1b | 30 | # It requires ffmpeg (m3u8 download) |
1547c8cc | 31 | 'skip_download': True, |
bfd14b1b JMF |
32 | }, |
33 | } | |
34 | ||
bfd14b1b JMF |
35 | def _real_extract(self, url): |
36 | mobj = re.match(self._VALID_URL, url) | |
37 | long_id = mobj.group('long_id') | |
a2fb2a21 | 38 | |
39 | api_response = self._download_webpage('http://rutube.ru/api/video/%s/?format=json' % long_id, | |
40 | long_id, 'Downloading video JSON') | |
41 | video = json.loads(api_response) | |
42 | ||
43 | api_response = self._download_webpage('http://rutube.ru/api/play/trackinfo/%s/?format=json' % long_id, | |
44 | long_id, 'Downloading trackinfo JSON') | |
45 | trackinfo = json.loads(api_response) | |
46 | ||
bfd14b1b JMF |
47 | # Some videos don't have the author field |
48 | author = trackinfo.get('author') or {} | |
49 | m3u8_url = trackinfo['video_balancer'].get('m3u8') | |
50 | if m3u8_url is None: | |
1547c8cc | 51 | raise ExtractorError('Couldn\'t find m3u8 manifest url') |
bfd14b1b JMF |
52 | |
53 | return { | |
a2fb2a21 | 54 | 'id': video['id'], |
55 | 'title': video['title'], | |
56 | 'description': video['description'], | |
57 | 'duration': video['duration'], | |
58 | 'view_count': video['hits'], | |
bfd14b1b JMF |
59 | 'url': m3u8_url, |
60 | 'ext': 'mp4', | |
a2fb2a21 | 61 | 'thumbnail': video['thumbnail_url'], |
bfd14b1b JMF |
62 | 'uploader': author.get('name'), |
63 | 'uploader_id': compat_str(author['id']) if author else None, | |
a2fb2a21 | 64 | 'upload_date': unified_strdate(video['created_ts']), |
65 | 'age_limit': 18 if video['is_adult'] else 0, | |
bfd14b1b | 66 | } |
1547c8cc | 67 | |
68 | ||
69 | class RutubeChannelIE(InfoExtractor): | |
70 | IE_NAME = 'rutube:channel' | |
71 | IE_DESC = 'Rutube channels' | |
72 | _VALID_URL = r'http://rutube\.ru/tags/video/(?P<id>\d+)' | |
73 | ||
74 | _PAGE_TEMPLATE = 'http://rutube.ru/api/tags/video/%s/?page=%s&format=json' | |
75 | ||
76 | def _extract_videos(self, channel_id, channel_title=None): | |
77 | entries = [] | |
78 | for pagenum in itertools.count(1): | |
a2fb2a21 | 79 | api_response = self._download_webpage(self._PAGE_TEMPLATE % (channel_id, pagenum), |
1547c8cc | 80 | channel_id, 'Downloading page %s' % pagenum) |
a2fb2a21 | 81 | page = json.loads(api_response) |
1547c8cc | 82 | results = page['results'] |
83 | if len(results) == 0: | |
84 | break; | |
a2fb2a21 | 85 | entries.extend(self.url_result(result['video_url'], 'Rutube') for result in results) |
1547c8cc | 86 | if page['has_next'] is False: |
87 | break; | |
88 | return self.playlist_result(entries, channel_id, channel_title) | |
89 | ||
90 | def _real_extract(self, url): | |
91 | mobj = re.match(self._VALID_URL, url) | |
92 | channel_id = mobj.group('id') | |
93 | return self._extract_videos(channel_id) | |
94 | ||
95 | ||
96 | class RutubeMovieIE(RutubeChannelIE): | |
97 | IE_NAME = 'rutube:movie' | |
98 | IE_DESC = 'Rutube movies' | |
99 | _VALID_URL = r'http://rutube\.ru/metainfo/tv/(?P<id>\d+)' | |
100 | ||
101 | _MOVIE_TEMPLATE = 'http://rutube.ru/api/metainfo/tv/%s/?format=json' | |
102 | _PAGE_TEMPLATE = 'http://rutube.ru/api/metainfo/tv/%s/video?page=%s&format=json' | |
103 | ||
104 | def _real_extract(self, url): | |
105 | mobj = re.match(self._VALID_URL, url) | |
106 | movie_id = mobj.group('id') | |
a2fb2a21 | 107 | api_response = self._download_webpage(self._MOVIE_TEMPLATE % movie_id, movie_id, |
1547c8cc | 108 | 'Downloading movie JSON') |
a2fb2a21 | 109 | movie = json.loads(api_response) |
1547c8cc | 110 | movie_name = movie['name'] |
111 | return self._extract_videos(movie_id, movie_name) |