Source code for fictive.mandrill.ext.flask

"""
Facilities for using the mandrill client with Flask

"""
import collections

import mandrill

from fictive.flask.extension import AbstractFlaskExtension


class Mandrill(AbstractFlaskExtension):
    """
    Flask extension wrapper around the mandrill client

    """

    CONFIG_KEY = 'MANDRILL'
    EXTENSION_KEY = 'mandrill'

    @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 mandrill.Mandrill(*args, **kwargs)