Source code for sandman.exception

"""Exception specifications for Sandman"""

from flask import make_response, render_template

class InvalidAPIUsage(Exception):
    """Excecption which generates a :class:`flask.Response` object whose
    *data* is JSON rather than HTML"""

    def __init__(self, code=400, message=None, payload=None):
        super(InvalidAPIUsage, self).__init__(message)
        self.message = message
        self.payload = payload
        if code is not None:
            self.code = code

    def to_dict(self):
        """Return a dictionary representation of the exception."""
        as_dict = dict(self.payload or ())
        as_dict['message'] = self.message
        return as_dict

    def abort(self):
        """Return an HTML Response representation of the exception."""
        resp = make_response(render_template('error.html', error=self.code, message=self.message), self.code)
        return resp