From 277c1a80fd10d870062ec1c56ab6eb3f20409e67 Mon Sep 17 00:00:00 2001 From: Mouse Reeve Date: Sat, 9 May 2020 12:59:06 -0700 Subject: [PATCH] Fixes loading covers --- fedireads/connectors/abstract_connector.py | 2 +- fedireads/connectors/fedireads_connector.py | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/fedireads/connectors/abstract_connector.py b/fedireads/connectors/abstract_connector.py index e0a1a20b..83729bb3 100644 --- a/fedireads/connectors/abstract_connector.py +++ b/fedireads/connectors/abstract_connector.py @@ -231,7 +231,7 @@ def get_data(url): ) if not resp.ok: resp.raise_for_status() - data = response.json() + data = resp.json() return data diff --git a/fedireads/connectors/fedireads_connector.py b/fedireads/connectors/fedireads_connector.py index 8f99ee94..7ead431b 100644 --- a/fedireads/connectors/fedireads_connector.py +++ b/fedireads/connectors/fedireads_connector.py @@ -84,14 +84,15 @@ class Connector(AbstractConnector): def get_cover_from_data(self, data): - cover_url = data.get('cover') - if not cover_url: + cover_data = data.get('attachment') + if not cover_data: return None + cover_url = cover_data[0].get('url') response = requests.get(cover_url) if not response.ok: response.raise_for_status() - image_name = uuid4() + cover_url.split('.')[-1] + image_name = str(uuid4()) + cover_url.split('.')[-1] image_content = ContentFile(response.content) return [image_name, image_content]