관리-도구

편집 파일: decorators.py

import sys
from functools import wraps

from six import reraise

from . import Traceback

class Error:
    def __init__(self, exc_type, exc_value, traceback):
        self.exc_type = exc_type
        self.exc_value = exc_value
        self.__traceback = Traceback(traceback)

@property
    def traceback(self):
        return self.__traceback.as_traceback()

def reraise(self):
        reraise(self.exc_type, self.exc_value, self.traceback)

def return_error(func, exc_type=Exception):
    @wraps(func)
    def return_exceptions_wrapper(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except exc_type:
            return Error(*sys.exc_info())

return return_exceptions_wrapper

returns_error = return_errors = returns_errors = return_error  # cause I make too many typos

@return_error
def apply_with_return_error(args):
    """
    args is a tuple where the first argument is a callable.

eg::

apply_with_return_error((func, 1, 2, 3)) - this will call func(1, 2, 3)

"""
    return args[0](*args[1:])