KanyeRest and KanyeParse

api_content.py

import json
import logging

import requests
from django.conf import settings


logger = logging.getLogger(__name__)


class KanyeParse:
    def __init__(self, data):
        self.data = data
    
    # A very simple parser
    def get_parsed_data(self):
        parsed_data = json.loads(self.data)
        return parsed_data["quote"]


class KanyeRest:
    def __init__(self):
        self.data = None
        self.url = "https://api.kanye.rest/"

    def get_url(self):
        return self.url

    def fetch_data(self):
        try:
            response = requests.get(url=self.url)
            self.data = response.content
        except requests.exceptions.Timeout:
            logger.exception(f"Timeout occurred")
        except Exception:
            logger.exception(f"Error occurred")
        return self.data

    def get_data(self):
        """ If there is data, parse it, otherwise return an empty list """
        self.data = self.fetch_data()
        if self.data:
            self.data = KanyeParse(self.data).get_parsed_data()
            return self.data
        else:
            return []