Merge branch 'vgtv' of https://github.com/mrkolby/youtube-dl into mrkolby-vgtv
[youtube-dl] / youtube_dl / extractor / eporner.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..utils import (
8     parse_duration,
9     str_to_int,
10 )
11
12
13 class EpornerIE(InfoExtractor):
14     _VALID_URL = r'https?://(?:www\.)?eporner\.com/hd-porn/(?P<id>\d+)/(?P<display_id>[\w-]+)'
15     _TEST = {
16         'url': 'http://www.eporner.com/hd-porn/95008/Infamous-Tiffany-Teen-Strip-Tease-Video/',
17         'md5': '3b427ae4b9d60619106de3185c2987cd',
18         'info_dict': {
19             'id': '95008',
20             'display_id': 'Infamous-Tiffany-Teen-Strip-Tease-Video',
21             'ext': 'flv',
22             'title': 'Infamous Tiffany Teen Strip Tease Video',
23             'duration': 194,
24             'view_count': int,
25             'age_limit': 18,
26         }
27     }
28
29     def _real_extract(self, url):
30         mobj = re.match(self._VALID_URL, url)
31         video_id = mobj.group('id')
32         display_id = mobj.group('display_id')
33
34         webpage = self._download_webpage(url, display_id)
35         title = self._html_search_regex(
36             r'<title>(.*?) - EPORNER', webpage, 'title')
37
38         redirect_code = self._html_search_regex(
39             r'<script type="text/javascript" src="/config5/%s/([a-f\d]+)/">' % video_id,
40             webpage, 'redirect_code')
41         redirect_url = 'http://www.eporner.com/config5/%s/%s' % (video_id, redirect_code)
42         player_code = self._download_webpage(
43             redirect_url, display_id, note='Downloading player config')
44
45         sources = self._search_regex(
46             r'(?s)sources\s*:\s*\[\s*({.+?})\s*\]', player_code, 'sources')
47
48         formats = []
49         for video_url, format_id in re.findall(r'file\s*:\s*"([^"]+)",\s*label\s*:\s*"([^"]+)"', sources):
50             fmt = {
51                 'url': video_url,
52                 'format_id': format_id,
53             }
54             m = re.search(r'^(\d+)', format_id)
55             if m:
56                 fmt['height'] = int(m.group(1))
57             formats.append(fmt)
58         self._sort_formats(formats)
59
60         duration = parse_duration(self._search_regex(
61             r'class="mbtim">([0-9:]+)</div>', webpage, 'duration',
62             fatal=False))
63         view_count = str_to_int(self._search_regex(
64             r'id="cinemaviews">\s*([0-9,]+)\s*<small>views',
65             webpage, 'view count', fatal=False))
66
67         return {
68             'id': video_id,
69             'display_id': display_id,
70             'title': title,
71             'duration': duration,
72             'view_count': view_count,
73             'formats': formats,
74             'age_limit': self._rta_search(webpage),
75         }