From 713b2250bc62a2368c42c25c312f6ff8ae95f337 Mon Sep 17 00:00:00 2001 From: blank X Date: Sun, 14 Nov 2021 02:04:32 +0700 Subject: [PATCH] Support imgur.com/gallery links --- redditbot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/redditbot.py b/redditbot.py index 39dadc7..478a176 100644 --- a/redditbot.py +++ b/redditbot.py @@ -359,9 +359,9 @@ async def main(): splitted = os.path.splitext(parsed[2]) domain = getattr(random_post, 'domain', parsed[1]) preview = getattr(random_post, 'preview', None) - if domain.endswith('imgur.com'): + if domain == 'imgur.com' or domain.endswith('.imgur.com'): parsed[1] = 'i.imgur.com' - if parsed[2].startswith('/a/'): + if parsed[2].startswith('/a/') or parsed[2].startswith('/gallery/'): albumid = os.path.split(parsed[2].rstrip('/'))[1] async with session.get(f'https://imgur.com/ajaxalbums/getimages/{albumid}/hit.json?all=true') as resp: apidata = (await resp.json())['data']