1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2024-11-24 19:22:06 +00:00

Apply suggestions from code review

Co-authored-by: dirkf <fieldhouse@gmx.net>
This commit is contained in:
P. S 2024-07-15 18:38:13 +02:00 committed by GitHub
parent 5f38e78a96
commit d8414d01bd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,3 +1,4 @@
# coding: utf-8
from __future__ import unicode_literals from __future__ import unicode_literals
import itertools import itertools
@ -34,24 +35,30 @@ class Tube8IE(InfoExtractor):
} }
}] }]
_EMBED_REGEX = r'<iframe [^>]*\bsrc=["\'](?P<url>(?:https?:)?//(?:www\.)?tube8\.com/embed/(?:[^/]+/)*\d+)'
@classmethod
def _extract_urls(cls, webpage):
return [m.group('url') for m in re.finditer(cls._EMBED_REGEX, webpage)]
def _real_extract(self, url): def _real_extract(self, url):
video_id = self._match_valid_url(url).group('id') video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id) webpage = self._download_webpage(url, video_id)
age_verify_msg = self._search_regex(
r'(your elected officials in \w+(?:\s+\w+){,2} are requiring us to verify your age before allowing you access to our website)',
webpage, 'age verification message', default=None)
if age_verify_msg:
self.raise_geo_restricted('%s said: "%s"' % (self.IE_NAME, age_verify_msg))
playervars = self._search_json( playervars = self._search_json(
r'\bplayervars\s*:', webpage, 'playervars', video_id) r'\bplayervars\s*:', webpage, 'playervars', video_id)
extra_info = self._search_json( info = self._search_json_ld(webpage, video_id, expected_type='VideoObject', default={})
r'application/ld\+json[\"\']?>\s?', webpage, 'extra_info', video_id) for k in ('url', 'description'):
info.pop(k, None)
uploader = traverse_obj(extra_info, ( info['uploader'] = clean_html(get_element_by_class('submitByLink', webpage))
'@graph', lambda _, v: v.get('author'), 'author'))[0]
thumbnail = traverse_obj(extra_info, (
'@graph', lambda _, v: v.get('thumbnail'), 'thumbnail'))[0]
timestamp = parse_iso8601(traverse_obj(extra_info, (
'@graph', lambda _, v: v.get('datePublished'), 'datePublished'))[0])
# Borrowed from youporn extractor # Borrowed from youporn extractor
def get_fmt(x): def get_fmt(x):
@ -63,9 +70,9 @@ class Tube8IE(InfoExtractor):
defs_by_format = dict(traverse_obj(playervars, ( defs_by_format = dict(traverse_obj(playervars, (
'mediaDefinitions', lambda _, v: v.get('format'), T(get_fmt)))) 'mediaDefinitions', lambda _, v: v.get('format'), T(get_fmt))))
title = traverse_obj(playervars, 'video_title') info['title'] = strip_or_none(playervars.get('video_title')) or info.get('title') or playervars['video_title']
if not thumbnail: if 'thumbnail' not in info.keys():
thumbnail = traverse_obj(playervars, 'image_url') info['thumbnail'] = traverse_obj(playervars, ('image_url', T(url_or_none)))
# Borrowed from youporn extractor # Borrowed from youporn extractor
def get_format_data(f): def get_format_data(f):
@ -78,29 +85,26 @@ class Tube8IE(InfoExtractor):
for mp4_url in traverse_obj( for mp4_url in traverse_obj(
get_format_data('mp4'), get_format_data('mp4'),
(lambda _, v: not isinstance(v['defaultQuality'], bool), 'videoUrl'), (lambda _, v: not isinstance(v['defaultQuality'], bool), 'videoUrl'),
(Ellipsis, 'videoUrl')): (Ellipsis, 'videoUrl'),
mobj = re.search(r'(?P<height>\d{3,4})[pP]_(?P<bitrate>\d+)[kK]_\d+', mp4_url) expected_type=url_or_none):
if mobj: height, tbr = map(int_or_none, self._search_regex(r'(?i)(?P<height>\d{3,4})p_(?P<bitrate>\d+)k_\d+', mp4_url, 'media details', group=('height', 'bitrate'), fatal=False))
height = int(mobj.group('height')) fmt_id = '%dp' % height if height else 'mp4'
tbr = int(mobj.group('bitrate')) formats.append({
formats.append({ 'format_id': fmt_id,
'format_id': '%dp-%dk' % (height, tbr), 'url': mp4_url,
'url': mp4_url, 'ext': 'mp4',
'ext': 'mp4', 'height': height,
'tbr': tbr, 'tbr': tbr,
'height': height, })
})
self._sort_formats(formats) self._sort_formats(formats)
return { info.update({
'id': video_id, 'id': video_id,
'title': title,
'thumbnail': thumbnail,
'formats': formats, 'formats': formats,
'uploader': uploader, 'age_limit': 18,
'timestamp': timestamp, })
} return info
# Currently only user channels # Currently only user channels