laudom/haruhi-dl
2
0
Fork 0
Code Issues 13 Pull requests Packages Projects Releases Wiki Activity

[pornhub] Fix review issues (closes #27393)

Browse source
This commit is contained in:
=?UTF-8?q?Sergey=20M=E2=80=A4?= 2021-02-26 15:36:38 +01:00 committed by Dominika
parent 6ba48137c2
commit c7a0d780b6
1 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
haruhi_dl/extractor/pornhub.py
View file

@ -291,15 +291,13 @@ class PornHubIE(PornHubBaseIE):
video_urls.append((v_url, None)) video_urls.append((v_url, None))
video_urls_set.add(v_url) video_urls_set.add(v_url)
def parse_quality_items(js_str): def parse_quality_items(quality_items):
if (url_or_none(js_str)): q_items = self._parse_json(quality_items, video_id, fatal=False)
return js_str if not isinstance(q_items, list):
media_definitions = self._parse_json(js_str, video_id, fatal=False) return
if isinstance(media_definitions, list): for item in q_items:
for definition in media_definitions: if isinstance(item, dict):
if not isinstance(definition, dict): add_video_url(item.get('url'))
continue
add_video_url(definition.get('url'))
if not video_urls: if not video_urls:
FORMAT_PREFIXES = ('media', 'quality', 'qualityItems') FORMAT_PREFIXES = ('media', 'quality', 'qualityItems')
@ -308,8 +306,10 @@ class PornHubIE(PornHubBaseIE):
default=None) default=None)
if js_vars: if js_vars:
for key, format_url in js_vars.items(): for key, format_url in js_vars.items():
if any(key.startswith(p) for p in FORMAT_PREFIXES): if key.startswith(FORMAT_PREFIXES[-1]):
add_video_url(parse_quality_items(format_url)) parse_quality_items(format_url)
elif any(key.startswith(p) for p in FORMAT_PREFIXES[:2]):
add_video_url(format_url)
if not video_urls and re.search( if not video_urls and re.search(
r'<[^>]+\bid=["\']lockedPlayer', webpage): r'<[^>]+\bid=["\']lockedPlayer', webpage):
raise ExtractorError( raise ExtractorError(