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]