gryf/tuir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added test cases.

Browse Source
This commit is contained in:
Michael Lazar
2016-08-06 17:16:13 -07:00
parent 71713bdbfd
commit f2b412ab31
4 changed files with 352 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rtv/mime_parsers.py
View File

@@ -180,12 +180,12 @@ class InstagramMIMEParser(BaseMIMEParser):
def get_mimetype(url):
page = requests.get(url)
soup = BeautifulSoup(page.content, 'html.parser')
tags = soup.find_all('meta', attrs={'property': 'og:video:secure_url'})
if tags:
return BaseMIMEParser.get_mimetype(tags[0].get('content'))
tag = soup.find('meta', attrs={'property': 'og:video:secure_url'})
tag = tag or soup.find('meta', attrs={'property': 'og:image'})
if tag:
return BaseMIMEParser.get_mimetype(tag.get('content'))
else:
tags = soup.find_all('meta', attrs={'property': 'og:image'})
return BaseMIMEParser.get_mimetype(tags[0].get('content'))
return url, None
# Parsers should be listed in the order they will be checked