ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add a .png if the image returns an imgur gallery

Browse Source
This commit is contained in:
Luke Robles 2020-09-16 11:37:28 -07:00
parent f3eb5f4d7c
commit afcf7d4ab9
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/get_from_reddit.py
View File

@ -45,5 +45,10 @@ def get_image(boards, nsfw=False):
if response['data']['domain'] not in domains: if response['data']['domain'] not in domains:
# If we dont find an approved domain, re-try the request # If we dont find an approved domain, re-try the request
return get_image(boards, nsfw=nsfw) return get_image(boards, nsfw=nsfw)
image_url = response['data']['url'].replace('http://', 'https://')
return response['data']['url'].replace('http://', 'https://') if not image_url.endswith('png') or not image_url.endswith('jpg') not image_url.endswith('gif'):
image_url = image_url + '.png'
return image_url