Source code for fictive.heap.ext.flask

"""
Flask extension around the Heap API client

"""

import collections

from fictive.flask.extension import AbstractFlaskExtension

from fictive.heap.client import HeapClient


class Heap(AbstractFlaskExtension):
    """
    Flask extension around the Heap API client

    """

    CONFIG_KEY = 'HEAP'
    EXTENSION_KEY = 'heap'

    @classmethod
    def get_client(cls, app, *args, **kwargs):
        args = args + cls.appenv(app)['args'][len(args):]
        kwargs = dict(collections.ChainMap(kwargs, cls.appenv(app)['kwargs']))
        return HeapClient(*args, **kwargs)