Merge branch 'akamai_pv' of https://github.com/remitamine/youtube-dl into remitamine...
[youtube-dl] / youtube_dl / extractor / tnaflix.py
1 from __future__ import unicode_literals
2
3 import re
4
5 from .common import InfoExtractor
6 from ..compat import compat_str
7 from ..utils import (
8     fix_xml_ampersands,
9     float_or_none,
10     int_or_none,
11     parse_duration,
12     str_to_int,
13     xpath_text,
14 )
15
16
17 class TNAFlixNetworkBaseIE(InfoExtractor):
18     # May be overridden in descendants if necessary
19     _CONFIG_REGEX = [
20         r'flashvars\.config\s*=\s*escape\("([^"]+)"',
21         r'<input[^>]+name="config\d?" value="([^"]+)"',
22     ]
23     _TITLE_REGEX = r'<input[^>]+name="title" value="([^"]+)"'
24     _DESCRIPTION_REGEX = r'<input[^>]+name="description" value="([^"]+)"'
25     _UPLOADER_REGEX = r'<input[^>]+name="username" value="([^"]+)"'
26     _VIEW_COUNT_REGEX = None
27     _COMMENT_COUNT_REGEX = None
28     _AVERAGE_RATING_REGEX = None
29     _CATEGORIES_REGEX = r'<li[^>]*>\s*<span[^>]+class="infoTitle"[^>]*>Categories:</span>\s*<span[^>]+class="listView"[^>]*>(.+?)</span>\s*</li>'
30
31     def _extract_thumbnails(self, flix_xml):
32
33         def get_child(elem, names):
34             for name in names:
35                 child = elem.find(name)
36                 if child is not None:
37                     return child
38
39         timeline = get_child(flix_xml, ['timeline', 'rolloverBarImage'])
40         if timeline is None:
41             return
42
43         pattern_el = get_child(timeline, ['imagePattern', 'pattern'])
44         if pattern_el is None or not pattern_el.text:
45             return
46
47         first_el = get_child(timeline, ['imageFirst', 'first'])
48         last_el = get_child(timeline, ['imageLast', 'last'])
49         if first_el is None or last_el is None:
50             return
51
52         first_text = first_el.text
53         last_text = last_el.text
54         if not first_text.isdigit() or not last_text.isdigit():
55             return
56
57         first = int(first_text)
58         last = int(last_text)
59         if first > last:
60             return
61
62         width = int_or_none(xpath_text(timeline, './imageWidth', 'thumbnail width'))
63         height = int_or_none(xpath_text(timeline, './imageHeight', 'thumbnail height'))
64
65         return [{
66             'url': self._proto_relative_url(pattern_el.text.replace('#', compat_str(i)), 'http:'),
67             'width': width,
68             'height': height,
69         } for i in range(first, last + 1)]
70
71     def _real_extract(self, url):
72         mobj = re.match(self._VALID_URL, url)
73         video_id = mobj.group('id')
74         display_id = mobj.group('display_id') if 'display_id' in mobj.groupdict() else video_id
75
76         webpage = self._download_webpage(url, display_id)
77
78         cfg_url = self._proto_relative_url(self._html_search_regex(
79             self._CONFIG_REGEX, webpage, 'flashvars.config', default=None), 'http:')
80
81         if not cfg_url:
82             inputs = self._hidden_inputs(webpage)
83             cfg_url = 'https://cdn-fck.tnaflix.com/tnaflix/%s.fid?key=%s' % (inputs['vkey'], inputs['nkey'])
84
85         cfg_xml = self._download_xml(
86             cfg_url, display_id, 'Downloading metadata',
87             transform_source=fix_xml_ampersands)
88
89         formats = []
90
91         def extract_video_url(vl):
92             return re.sub('speed=\d+', 'speed=', vl.text)
93
94         video_link = cfg_xml.find('./videoLink')
95         if video_link is not None:
96             formats.append({
97                 'url': extract_video_url(video_link),
98                 'ext': xpath_text(cfg_xml, './videoConfig/type', 'type', default='flv'),
99             })
100
101         for item in cfg_xml.findall('./quality/item'):
102             video_link = item.find('./videoLink')
103             if video_link is None:
104                 continue
105             res = item.find('res')
106             format_id = None if res is None else res.text
107             height = int_or_none(self._search_regex(
108                 r'^(\d+)[pP]', format_id, 'height', default=None))
109             formats.append({
110                 'url': self._proto_relative_url(extract_video_url(video_link), 'http:'),
111                 'format_id': format_id,
112                 'height': height,
113             })
114
115         self._sort_formats(formats)
116
117         thumbnail = self._proto_relative_url(
118             xpath_text(cfg_xml, './startThumb', 'thumbnail'), 'http:')
119         thumbnails = self._extract_thumbnails(cfg_xml)
120
121         title = self._html_search_regex(
122             self._TITLE_REGEX, webpage, 'title') if self._TITLE_REGEX else self._og_search_title(webpage)
123
124         age_limit = self._rta_search(webpage) or 18
125
126         duration = parse_duration(self._html_search_meta(
127             'duration', webpage, 'duration', default=None))
128
129         def extract_field(pattern, name):
130             return self._html_search_regex(pattern, webpage, name, default=None) if pattern else None
131
132         description = extract_field(self._DESCRIPTION_REGEX, 'description')
133         uploader = extract_field(self._UPLOADER_REGEX, 'uploader')
134         view_count = str_to_int(extract_field(self._VIEW_COUNT_REGEX, 'view count'))
135         comment_count = str_to_int(extract_field(self._COMMENT_COUNT_REGEX, 'comment count'))
136         average_rating = float_or_none(extract_field(self._AVERAGE_RATING_REGEX, 'average rating'))
137
138         categories_str = extract_field(self._CATEGORIES_REGEX, 'categories')
139         categories = [c.strip() for c in categories_str.split(',')] if categories_str is not None else []
140
141         return {
142             'id': video_id,
143             'display_id': display_id,
144             'title': title,
145             'description': description,
146             'thumbnail': thumbnail,
147             'thumbnails': thumbnails,
148             'duration': duration,
149             'age_limit': age_limit,
150             'uploader': uploader,
151             'view_count': view_count,
152             'comment_count': comment_count,
153             'average_rating': average_rating,
154             'categories': categories,
155             'formats': formats,
156         }
157
158
159 class TNAFlixNetworkEmbedIE(TNAFlixNetworkBaseIE):
160     _VALID_URL = r'https?://player\.(?:tna|emp)flix\.com/video/(?P<id>\d+)'
161
162     _TITLE_REGEX = r'<title>([^<]+)</title>'
163
164     _TESTS = [{
165         'url': 'https://player.tnaflix.com/video/6538',
166         'info_dict': {
167             'id': '6538',
168             'display_id': '6538',
169             'ext': 'mp4',
170             'title': 'Educational xxx video',
171             'thumbnail': 're:https?://.*\.jpg$',
172             'age_limit': 18,
173         },
174         'params': {
175             'skip_download': True,
176         },
177     }, {
178         'url': 'https://player.empflix.com/video/33051',
179         'only_matching': True,
180     }]
181
182     @staticmethod
183     def _extract_urls(webpage):
184         return [url for _, url in re.findall(
185             r'<iframe[^>]+?src=(["\'])(?P<url>(?:https?:)?//player\.(?:tna|emp)flix\.com/video/\d+)\1',
186             webpage)]
187
188
189 class TNAFlixIE(TNAFlixNetworkBaseIE):
190     _VALID_URL = r'https?://(?:www\.)?tnaflix\.com/[^/]+/(?P<display_id>[^/]+)/video(?P<id>\d+)'
191
192     _TITLE_REGEX = r'<title>(.+?) - TNAFlix Porn Videos</title>'
193     _DESCRIPTION_REGEX = r'<meta[^>]+name="description"[^>]+content="([^"]+)"'
194     _UPLOADER_REGEX = r'<i>\s*Verified Member\s*</i>\s*<h1>(.+?)</h1>'
195     _CATEGORIES_REGEX = r'(?s)<span[^>]*>Categories:</span>(.+?)</div>'
196
197     _TESTS = [{
198         # anonymous uploader, no categories
199         'url': 'http://www.tnaflix.com/porn-stars/Carmella-Decesare-striptease/video553878',
200         'md5': '7e569419fe6d69543d01e6be22f5f7c4',
201         'info_dict': {
202             'id': '553878',
203             'display_id': 'Carmella-Decesare-striptease',
204             'ext': 'mp4',
205             'title': 'Carmella Decesare - striptease',
206             'thumbnail': 're:https?://.*\.jpg$',
207             'duration': 91,
208             'age_limit': 18,
209             'categories': ['Porn Stars'],
210         }
211     }, {
212         # non-anonymous uploader, categories
213         'url': 'https://www.tnaflix.com/teen-porn/Educational-xxx-video/video6538',
214         'md5': 'fcba2636572895aba116171a899a5658',
215         'info_dict': {
216             'id': '6538',
217             'display_id': 'Educational-xxx-video',
218             'ext': 'flv',
219             'title': 'Educational xxx video',
220             'description': 'md5:b4fab8f88a8621c8fabd361a173fe5b8',
221             'thumbnail': 're:https?://.*\.jpg$',
222             'duration': 164,
223             'age_limit': 18,
224             'uploader': 'bobwhite39',
225             'categories': ['Amateur Porn', 'Squirting Videos', 'Teen Girls 18+'],
226         }
227     }, {
228         'url': 'https://www.tnaflix.com/amateur-porn/bunzHD-Ms.Donk/video358632',
229         'only_matching': True,
230     }]
231
232
233 class EMPFlixIE(TNAFlixNetworkBaseIE):
234     _VALID_URL = r'https?://(?:www\.)?empflix\.com/videos/(?P<display_id>.+?)-(?P<id>[0-9]+)\.html'
235
236     _UPLOADER_REGEX = r'<span[^>]+class="infoTitle"[^>]*>Uploaded By:</span>(.+?)</li>'
237
238     _TESTS = [{
239         'url': 'http://www.empflix.com/videos/Amateur-Finger-Fuck-33051.html',
240         'md5': 'b1bc15b6412d33902d6e5952035fcabc',
241         'info_dict': {
242             'id': '33051',
243             'display_id': 'Amateur-Finger-Fuck',
244             'ext': 'mp4',
245             'title': 'Amateur Finger Fuck',
246             'description': 'Amateur solo finger fucking.',
247             'thumbnail': 're:https?://.*\.jpg$',
248             'duration': 83,
249             'age_limit': 18,
250             'uploader': 'cwbike',
251             'categories': ['Amateur', 'Anal', 'Fisting', 'Home made', 'Solo'],
252         }
253     }, {
254         'url': 'http://www.empflix.com/videos/[AROMA][ARMD-718]-Aoi-Yoshino-Sawa-25826.html',
255         'only_matching': True,
256     }]
257
258
259 class MovieFapIE(TNAFlixNetworkBaseIE):
260     _VALID_URL = r'https?://(?:www\.)?moviefap\.com/videos/(?P<id>[0-9a-f]+)/(?P<display_id>[^/]+)\.html'
261
262     _VIEW_COUNT_REGEX = r'<br>Views\s*<strong>([\d,.]+)</strong>'
263     _COMMENT_COUNT_REGEX = r'<span[^>]+id="comCount"[^>]*>([\d,.]+)</span>'
264     _AVERAGE_RATING_REGEX = r'Current Rating\s*<br>\s*<strong>([\d.]+)</strong>'
265     _CATEGORIES_REGEX = r'(?s)<div[^>]+id="vid_info"[^>]*>\s*<div[^>]*>.+?</div>(.*?)<br>'
266
267     _TESTS = [{
268         # normal, multi-format video
269         'url': 'http://www.moviefap.com/videos/be9867c9416c19f54a4a/experienced-milf-amazing-handjob.html',
270         'md5': '26624b4e2523051b550067d547615906',
271         'info_dict': {
272             'id': 'be9867c9416c19f54a4a',
273             'display_id': 'experienced-milf-amazing-handjob',
274             'ext': 'mp4',
275             'title': 'Experienced MILF Amazing Handjob',
276             'description': 'Experienced MILF giving an Amazing Handjob',
277             'thumbnail': 're:https?://.*\.jpg$',
278             'age_limit': 18,
279             'uploader': 'darvinfred06',
280             'view_count': int,
281             'comment_count': int,
282             'average_rating': float,
283             'categories': ['Amateur', 'Masturbation', 'Mature', 'Flashing'],
284         }
285     }, {
286         # quirky single-format case where the extension is given as fid, but the video is really an flv
287         'url': 'http://www.moviefap.com/videos/e5da0d3edce5404418f5/jeune-couple-russe.html',
288         'md5': 'fa56683e291fc80635907168a743c9ad',
289         'info_dict': {
290             'id': 'e5da0d3edce5404418f5',
291             'display_id': 'jeune-couple-russe',
292             'ext': 'flv',
293             'title': 'Jeune Couple Russe',
294             'description': 'Amateur',
295             'thumbnail': 're:https?://.*\.jpg$',
296             'age_limit': 18,
297             'uploader': 'whiskeyjar',
298             'view_count': int,
299             'comment_count': int,
300             'average_rating': float,
301             'categories': ['Amateur', 'Teen'],
302         }
303     }]