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

Fix test for imgur gif link

Browse Source
This commit is contained in:
woorst
2017-05-28 09:25:28 -05:00
parent d123986d4f
commit 371f8db06a
2 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rtv/mime_parsers.py
View File

@@ -133,10 +133,12 @@ class ImgurMIMEParser(BaseMIMEParser):
data = json.loads(r.text)['data'] data = json.loads(r.text)['data']
if 'images' in data: if 'images' in data:
# TODO: handle imgur albums with mixed content, i.e. jpeg and gifv # TODO: handle imgur albums with mixed content, i.e. jpeg and gifv
urls = ' '.join([d['link'] for d in data['images'] if not d['animated']]) links = ' '.join([d['link'] for d in data['images'] if not d['animated']])
return urls, 'image/x-imgur-album' return links.replace('http://', 'https://'), 'image/x-imgur-album'
else : else :
return (data['mp4'], 'video/mp4') if data['animated'] else (data['link'], data['type']) link = data['mp4'] if data['animated'] else data['link']
mime = 'video/mp4' if data['animated'] else data['type']
return link.replace('http://', 'https://'), mime
class InstagramMIMEParser(BaseMIMEParser): class InstagramMIMEParser(BaseMIMEParser):

8
tests/test_mime_parsers.py
View File

@@ -41,12 +41,12 @@ URLS = OrderedDict([
'https://i.imgur.com/yW0kbMi.jpg', 'https://i.imgur.com/yW0kbMi.jpg',
'image/jpeg')), 'image/jpeg')),
('imgur_2', ( ('imgur_2', (
'http://imgur.com/yjP1v4B', 'http://imgur.com/gallery/yjP1v4B',
'https://i.imgur.com/yjP1v4Bh.jpg', 'https://i.imgur.com/yjP1v4B.mp4',
'image/jpeg')), 'video/mp4')),
('imgur_album', ( ('imgur_album', (
'http://imgur.com/a/qx9t5', 'http://imgur.com/a/qx9t5',
'http://i.imgur.com/uEt0YLI.jpg', 'https://i.imgur.com/uEt0YLI.jpg',
'image/x-imgur-album')), 'image/x-imgur-album')),
('instagram_image', ( ('instagram_image', (
'https://www.instagram.com/p/BIxQ0vrBN2Y/?taken-by=kimchi_chic', 'https://www.instagram.com/p/BIxQ0vrBN2Y/?taken-by=kimchi_chic',