]> jfr.im git - yt-dlp.git/blame - youtube_dl/extractor/drtuber.py
Merge pull request #6132 from alarig/master
[yt-dlp.git] / youtube_dl / extractor / drtuber.py
CommitLineData
d740f7e1 1from __future__ import unicode_literals
2
3import re
4
5from .common import InfoExtractor
eade1d7e 6from ..utils import str_to_int
d740f7e1 7
8
9class DrTuberIE(InfoExtractor):
1bf8cf5c 10 _VALID_URL = r'https?://(?:www\.)?drtuber\.com/video/(?P<id>\d+)/(?P<display_id>[\w-]+)'
d740f7e1 11 _TEST = {
12 'url': 'http://www.drtuber.com/video/1740434/hot-perky-blonde-naked-golf',
13 'md5': '93e680cf2536ad0dfb7e74d94a89facd',
14 'info_dict': {
15 'id': '1740434',
33422c05 16 'display_id': 'hot-perky-blonde-naked-golf',
d740f7e1 17 'ext': 'mp4',
76d1466b 18 'title': 'hot perky blonde naked golf',
eade1d7e
S
19 'like_count': int,
20 'dislike_count': int,
21 'comment_count': int,
2914e5f0 22 'categories': ['Babe', 'Blonde', 'Erotic', 'Outdoor', 'Softcore', 'Solo'],
d740f7e1 23 'thumbnail': 're:https?://.*\.jpg$',
11fc065c 24 'age_limit': 18,
d740f7e1 25 }
26 }
27
28 def _real_extract(self, url):
29 mobj = re.match(self._VALID_URL, url)
30 video_id = mobj.group('id')
1bf8cf5c 31 display_id = mobj.group('display_id')
d740f7e1 32
1bf8cf5c 33 webpage = self._download_webpage(url, display_id)
d740f7e1 34
35 video_url = self._html_search_regex(
36 r'<source src="([^"]+)"', webpage, 'video URL')
37
38 title = self._html_search_regex(
76d1466b
S
39 [r'class="hd_title" style="[^"]+">([^<]+)</h1>', r'<title>([^<]+) - \d+'],
40 webpage, 'title')
d740f7e1 41
42 thumbnail = self._html_search_regex(
43 r'poster="([^"]+)"',
44 webpage, 'thumbnail', fatal=False)
45
eade1d7e
S
46 like_count = str_to_int(self._html_search_regex(
47 r'<span id="rate_likes">\s*<img[^>]+>\s*<span>([\d,\.]+)</span>',
48 webpage, 'like count', fatal=False))
49 dislike_count = str_to_int(self._html_search_regex(
50 r'<span id="rate_dislikes">\s*<img[^>]+>\s*<span>([\d,\.]+)</span>',
51 webpage, 'like count', fatal=False))
52 comment_count = str_to_int(self._html_search_regex(
53 r'<span class="comments_count">([\d,\.]+)</span>',
54 webpage, 'comment count', fatal=False))
55
2914e5f0
S
56 cats_str = self._search_regex(
57 r'<span>Categories:</span><div>(.+?)</div>', webpage, 'categories', fatal=False)
58 categories = [] if not cats_str else re.findall(r'<a title="([^"]+)"', cats_str)
d740f7e1 59
60 return {
61 'id': video_id,
1bf8cf5c 62 'display_id': display_id,
d740f7e1 63 'url': video_url,
64 'title': title,
65 'thumbnail': thumbnail,
eade1d7e
S
66 'like_count': like_count,
67 'dislike_count': dislike_count,
68 'comment_count': comment_count,
d740f7e1 69 'categories': categories,
11fc065c 70 'age_limit': self._rta_search(webpage),
d740f7e1 71 }