]>
Commit | Line | Data |
---|---|---|
0143dc02 PH |
1 | import json |
2 | import os | |
3 | import re | |
4 | import sys | |
5 | ||
6 | from .common import InfoExtractor | |
7 | from ..utils import ( | |
8 | compat_urllib_parse_urlparse, | |
9 | compat_urllib_request, | |
10 | ||
11 | ExtractorError, | |
12 | unescapeHTML, | |
13 | unified_strdate, | |
14 | ) | |
15 | ||
16 | ||
17 | class YouPornIE(InfoExtractor): | |
18 | _VALID_URL = r'^(?:https?://)?(?:\w+\.)?youporn\.com/watch/(?P<videoid>[0-9]+)/(?P<title>[^/]+)' | |
19 | ||
20 | def _print_formats(self, formats): | |
21 | """Print all available formats""" | |
22 | print(u'Available formats:') | |
23 | print(u'ext\t\tformat') | |
24 | print(u'---------------------------------') | |
25 | for format in formats: | |
26 | print(u'%s\t\t%s' % (format['ext'], format['format'])) | |
27 | ||
28 | def _specific(self, req_format, formats): | |
29 | for x in formats: | |
30 | if x["format"] == req_format: | |
31 | return x | |
32 | return None | |
33 | ||
34 | def _real_extract(self, url): | |
35 | mobj = re.match(self._VALID_URL, url) | |
36 | video_id = mobj.group('videoid') | |
37 | ||
38 | req = compat_urllib_request.Request(url) | |
39 | req.add_header('Cookie', 'age_verified=1') | |
40 | webpage = self._download_webpage(req, video_id) | |
41 | ||
42 | # Get JSON parameters | |
43 | json_params = self._search_regex(r'var currentVideo = new Video\((.*)\);', webpage, u'JSON parameters') | |
44 | try: | |
45 | params = json.loads(json_params) | |
46 | except: | |
47 | raise ExtractorError(u'Invalid JSON') | |
48 | ||
49 | self.report_extraction(video_id) | |
50 | try: | |
51 | video_title = params['title'] | |
52 | upload_date = unified_strdate(params['release_date_f']) | |
53 | video_description = params['description'] | |
54 | video_uploader = params['submitted_by'] | |
55 | thumbnail = params['thumbnails'][0]['image'] | |
56 | except KeyError: | |
57 | raise ExtractorError('Missing JSON parameter: ' + sys.exc_info()[1]) | |
58 | ||
59 | # Get all of the formats available | |
60 | DOWNLOAD_LIST_RE = r'(?s)<ul class="downloadList">(?P<download_list>.*?)</ul>' | |
61 | download_list_html = self._search_regex(DOWNLOAD_LIST_RE, | |
62 | webpage, u'download list').strip() | |
63 | ||
64 | # Get all of the links from the page | |
65 | LINK_RE = r'(?s)<a href="(?P<url>[^"]+)">' | |
66 | links = re.findall(LINK_RE, download_list_html) | |
67 | if(len(links) == 0): | |
68 | raise ExtractorError(u'ERROR: no known formats available for video') | |
69 | ||
70 | self.to_screen(u'Links found: %d' % len(links)) | |
71 | ||
72 | formats = [] | |
73 | for link in links: | |
74 | ||
75 | # A link looks like this: | |
76 | # http://cdn1.download.youporn.phncdn.com/201210/31/8004515/480p_370k_8004515/YouPorn%20-%20Nubile%20Films%20The%20Pillow%20Fight.mp4?nvb=20121113051249&nva=20121114051249&ir=1200&sr=1200&hash=014b882080310e95fb6a0 | |
77 | # A path looks like this: | |
78 | # /201210/31/8004515/480p_370k_8004515/YouPorn%20-%20Nubile%20Films%20The%20Pillow%20Fight.mp4 | |
79 | video_url = unescapeHTML( link ) | |
80 | path = compat_urllib_parse_urlparse( video_url ).path | |
81 | extension = os.path.splitext( path )[1][1:] | |
82 | format = path.split('/')[4].split('_')[:2] | |
83 | # size = format[0] | |
84 | # bitrate = format[1] | |
85 | format = "-".join( format ) | |
86 | # title = u'%s-%s-%s' % (video_title, size, bitrate) | |
87 | ||
88 | formats.append({ | |
89 | 'id': video_id, | |
90 | 'url': video_url, | |
91 | 'uploader': video_uploader, | |
92 | 'upload_date': upload_date, | |
93 | 'title': video_title, | |
94 | 'ext': extension, | |
95 | 'format': format, | |
96 | 'thumbnail': thumbnail, | |
97 | 'description': video_description | |
98 | }) | |
99 | ||
100 | if self._downloader.params.get('listformats', None): | |
101 | self._print_formats(formats) | |
102 | return | |
103 | ||
104 | req_format = self._downloader.params.get('format', None) | |
105 | self.to_screen(u'Format: %s' % req_format) | |
106 | ||
107 | if req_format is None or req_format == 'best': | |
108 | return [formats[0]] | |
109 | elif req_format == 'worst': | |
110 | return [formats[-1]] | |
111 | elif req_format in ('-1', 'all'): | |
112 | return formats | |
113 | else: | |
114 | format = self._specific( req_format, formats ) | |
115 | if format is None: | |
116 | raise ExtractorError(u'Requested format not available') | |
117 | return [format] |