From 7fcbf623a0c09dda67e6aa7e3eba48090f4dac28 Mon Sep 17 00:00:00 2001 From: Serene-Arc Date: Wed, 28 Apr 2021 15:17:21 +1000 Subject: [PATCH] Catch additional errors in site downloaders --- bdfr/site_downloaders/gfycat.py | 2 +- bdfr/site_downloaders/redgifs.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/bdfr/site_downloaders/gfycat.py b/bdfr/site_downloaders/gfycat.py index f140660..f77f05b 100644 --- a/bdfr/site_downloaders/gfycat.py +++ b/bdfr/site_downloaders/gfycat.py @@ -34,7 +34,7 @@ class Gfycat(GifDeliveryNetwork): try: out = json.loads(content.contents[0])['video']['contentUrl'] - except (IndexError, KeyError) as e: + except (IndexError, KeyError, AttributeError) as e: raise SiteDownloaderError(f'Failed to download Gfycat link {url}: {e}') except json.JSONDecodeError as e: raise SiteDownloaderError(f'Did not receive valid JSON data: {e}') diff --git a/bdfr/site_downloaders/redgifs.py b/bdfr/site_downloaders/redgifs.py index 3b59818..14ab6ea 100644 --- a/bdfr/site_downloaders/redgifs.py +++ b/bdfr/site_downloaders/redgifs.py @@ -44,7 +44,7 @@ class Redgifs(GifDeliveryNetwork): try: out = json.loads(content.contents[0])['video']['contentUrl'] - except (IndexError, KeyError): + except (IndexError, KeyError, AttributeError): raise SiteDownloaderError('Failed to find JSON data in page') except json.JSONDecodeError as e: raise SiteDownloaderError(f'Received data was not valid JSON: {e}')