update liveleak mime parser for youtube embeds

This commit is contained in:
woorst
2017-10-29 08:52:34 -05:00
parent d5f344e4a7
commit 4796a7b70f
4 changed files with 354 additions and 3 deletions

View File

@@ -376,11 +376,15 @@ class LiveleakMIMEParser(BaseMIMEParser):
if source:
urls.append((source.get('src'), source.get('type')))
# TODO: Handle pages with multiple videos
# TODO: Handle pages with youtube embeds
if urls:
return urls[0]
else:
return url, None
iframe = soup.find_all(lambda t: t.name == 'iframe' and
'youtube.com' in t['src'])
if iframe:
return YoutubeMIMEParser.get_mimetype(iframe[0]['src'].strip('/'))
else:
return url, None
class ClippitUserMIMEParser(BaseMIMEParser):
@@ -471,6 +475,8 @@ class WorldStarHipHopMIMEParser(BaseMIMEParser):
'youtube.com' in t['src'])
if iframe:
return YoutubeMIMEParser.get_mimetype(iframe[0]['src'])
else:
return url, None