PEP8 applied
[youtube-dl] / youtube_dl / extractor / videobam.py
1 from __future__ import unicode_literals
2
3 import re
4 import json
5
6 from .common import InfoExtractor
7 from ..utils import int_or_none
8
9
10 class VideoBamIE(InfoExtractor):
11     _VALID_URL = r'http://(?:www\.)?videobam\.com/(?:videos/download/)?(?P<id>[a-zA-Z]+)'
12
13     _TESTS = [
14         {
15             'url': 'http://videobam.com/OiJQM',
16             'md5': 'db471f27763a531f10416a0c58b5a1e0',
17             'info_dict': {
18                 'id': 'OiJQM',
19                 'ext': 'mp4',
20                 'title': 'Is Alcohol Worse Than Ecstasy?',
21                 'description': 'md5:d25b96151515c91debc42bfbb3eb2683',
22                 'uploader': 'frihetsvinge',
23             },
24         },
25         {
26             'url': 'http://videobam.com/pqLvq',
27             'md5': 'd9a565b5379a99126ef94e1d7f9a383e',
28             'note': 'HD video',
29             'info_dict': {
30                 'id': 'pqLvq',
31                 'ext': 'mp4',
32                 'title': '_',
33             }
34         },
35     ]
36
37     def _real_extract(self, url):
38         mobj = re.match(self._VALID_URL, url)
39         video_id = mobj.group('id')
40
41         page = self._download_webpage('http://videobam.com/%s' % video_id, video_id, 'Downloading page')
42
43         formats = []
44
45         for preference, format_id in enumerate(['low', 'high']):
46             mobj = re.search(r"%s: '(?P<url>[^']+)'" % format_id, page)
47             if not mobj:
48                 continue
49             formats.append({
50                 'url': mobj.group('url'),
51                 'ext': 'mp4',
52                 'format_id': format_id,
53                 'preference': preference,
54             })
55
56         if not formats:
57             player_config = json.loads(self._html_search_regex(r'var player_config = ({.+?});', page, 'player config'))
58             formats = [{
59                 'url': item['url'],
60                 'ext': 'mp4',
61             } for item in player_config['playlist'] if 'autoPlay' in item]
62
63         self._sort_formats(formats)
64
65         title = self._og_search_title(page, default='_', fatal=False)
66         description = self._og_search_description(page, default=None)
67         thumbnail = self._og_search_thumbnail(page)
68         uploader = self._html_search_regex(r'Upload by ([^<]+)</a>', page, 'uploader', fatal=False, default=None)
69         view_count = int_or_none(
70             self._html_search_regex(r'<strong>Views:</strong> (\d+) ', page, 'view count', fatal=False))
71
72         return {
73             'id': video_id,
74             'title': title,
75             'description': description,
76             'thumbnail': thumbnail,
77             'uploader': uploader,
78             'view_count': view_count,
79             'formats': formats,
80             'age_limit': 18,
81         }