Source code for pyelasticsearch.exceptions

from requests import Timeout, ConnectionError


class ElasticHttpError(Exception):
    """Exception raised when ES returns a non-OK (>=400) HTTP status code"""
    # TODO: If helpful in practice, split this into separate subclasses for 4xx
    # and 5xx errors. On second thought, ES, as of 0.19.9, returns 500s on
    # trivial things like JSON parse errors (which it does recognize), so it
    # wouldn't be good to rely on its idea of what's a client error and what's
    # a server error. We'd have to test the string for what kind of error it is
    # and choose an exception class accordingly.

    # This @property technique allows the exception to be pickled (like by
    # Sentry or celery) without having to write our own serialization stuff.
    @property
    def status_code(self):
        """The HTTP status code of the response that precipitated the error"""
        return self.args[0]

    @property
    def error(self):
        """A string error message"""
        return self.args[1]

    def __unicode__(self):
        return u'Non-OK response returned (%d): %r' % (self.status_code,
                                                       self.error)


class ElasticHttpNotFoundError(ElasticHttpError):
    """Exception raised when a request to ES returns a 404"""


class InvalidJsonResponseError(Exception):
    """
    Exception raised in the unlikely event that ES returns a non-JSON response
    """
    @property
    def response(self):
        return self.args[0]

    def __unicode__(self):
        return u'Invalid JSON returned from ES: %r' % (self.response,)