bridge.py

import os

import quandl

from flask import current_app

from cache import cache


class Bridge:

    def __init__(self):
        if 'QUANDL_API_KEY' in current_app.config:
            quandl.ApiConfig.api_key = current_app.config['QUANDL_API_KEY']
        else:
            quandl.ApiConfig.api_key = os.environ['QUANDL_API_KEY']

    @cache.memoize()
    def request(self, data):
        """

        :param data: JSON Object to feed Quandl API request
        :return: JSON object with the response from he API
        """
        current_app.logger.debug("Cache missed")
        try:
            rows = 1
            if 'rows' in data:
                rows = data.get('rows')
            df = quandl.get(dataset=data.get('dataset'), rows=rows)
            if df['Value'].size == 1:
                return df['Value'].get(0)
            else:
                return df['Value'].to_list()
        except Exception as e:
            raise e

    def __repr__(self):
        return f'{"Bridge()"}'