Source code for dirhunt.exceptions

# -*- coding: utf-8 -*-
import functools
import sys
import traceback


class DirHuntError(Exception):
    body = ''

    def __init__(self, extra_body=''):
        self.extra_body = extra_body

    def __str__(self):
        msg = self.__class__.__name__
        if self.body:
            msg += ': {}'.format(self.body)
        if self.extra_body:
            msg += ('. {}' if self.body else ': {}').format(self.extra_body)
        return msg


class EmptyError(DirHuntError):
    pass


class RequestError(DirHuntError):
    pass


class IncompatibleVersionError(DirHuntError):
    pass


def catch(fn):
    def wrap(*args, **kwargs):
        try:
            fn(*args, **kwargs)
        except DirHuntError as e:
            sys.stderr.write('[Error] Dir Hunt Exception:\n{}\n'.format(e))
    return wrap


def reraise_with_stack(func):
    @functools.wraps(func)
    def wrapped(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except Exception as e:
            traceback.print_exc()
            raise e
    return wrapped