]> gitweb @ CieloNegro.org - youtube-dl.git/blob - youtube_dl/extractor/sockshare.py
Fix imports and general cleanup
[youtube-dl.git] / youtube_dl / extractor / sockshare.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from ..compat import (
7     compat_urllib_parse,
8     compat_urllib_request,
9 )
10 from ..utils import (
11     determine_ext,
12     ExtractorError,
13 )
14
15 from .common import InfoExtractor
16
17
18 class SockshareIE(InfoExtractor):
19     _VALID_URL = r'https?://(?:www\.)?sockshare\.com/file/(?P<id>[0-9A-Za-z]+)'
20     _FILE_DELETED_REGEX = r'This file doesn\'t exist, or has been removed\.</div>'
21     _TEST = {
22         'url': 'http://www.sockshare.com/file/437BE28B89D799D7',
23         'md5': '9d0bf1cfb6dbeaa8d562f6c97506c5bd',
24         'info_dict': {
25             'id': '437BE28B89D799D7',
26             'title': 'big_buck_bunny_720p_surround.avi',
27             'ext': 'avi',
28             'thumbnail': 're:^http://.*\.jpg$',
29         }
30     }
31
32     def _real_extract(self, url):
33         video_id = self._match_id(url)
34         url = 'http://sockshare.com/file/%s' % video_id
35         webpage = self._download_webpage(url, video_id)
36
37         if re.search(self._FILE_DELETED_REGEX, webpage) is not None:
38             raise ExtractorError('Video %s does not exist' % video_id,
39                                  expected=True)
40
41         confirm_hash = self._html_search_regex(r'''(?x)<input\s+
42             type="hidden"\s+
43             value="([^"]*)"\s+
44             name="hash"
45             ''', webpage, 'hash')
46
47         fields = {
48             "hash": confirm_hash,
49             "confirm": "Continue as Free User"
50         }
51
52         post = compat_urllib_parse.urlencode(fields)
53         req = compat_urllib_request.Request(url, post)
54         # Apparently, this header is required for confirmation to work.
55         req.add_header('Host', 'www.sockshare.com')
56         req.add_header('Content-type', 'application/x-www-form-urlencoded')
57
58         webpage = self._download_webpage(
59             req, video_id, 'Downloading video page')
60
61         video_url = self._html_search_regex(
62             r'<a href="([^"]*)".+class="download_file_link"',
63             webpage, 'file url')
64         video_url = "http://www.sockshare.com" + video_url
65         title = self._html_search_regex((
66             r'<h1>(.+)<strong>',
67             r'var name = "([^"]+)";'),
68             webpage, 'title', default=None)
69         thumbnail = self._html_search_regex(
70             r'<img\s+src="([^"]*)".+?name="bg"',
71             webpage, 'thumbnail')
72
73         formats = [{
74             'format_id': 'sd',
75             'url': video_url,
76             'ext': determine_ext(title),
77         }]
78
79         return {
80             'id': video_id,
81             'title': title,
82             'thumbnail': thumbnail,
83             'formats': formats,
84         }