Повторно підняти виняток з іншим типом та повідомленням, зберігши наявну інформацію

Я пишу модуль і хочу мати єдину ієрархію винятків за винятками, які він може викликати (наприклад, успадкування від FooErrorабстрактного класу за всіма fooвинятками модуля). Це дозволяє користувачам модуля вловлювати ці конкретні винятки та чітко обробляти їх, якщо потрібно. Але багато винятків, викладених із модуля, виникають через якийсь інший виняток; наприклад, невдача в якомусь завданні через OSError у файлі.

Мені потрібно «обернути» виняток, що потрапляє таким чином, що він має інший тип і повідомлення , щоб інформація була доступна в подальшому для ієрархії поширення, що б не сприймало виняток. Але я не хочу втрачати існуючий тип, повідомлення та стек; це вся корисна інформація для того, хто намагається налагодити проблему. Обробник винятків верхнього рівня не є корисним, оскільки я намагаюся прикрасити виняток, перш ніж він пробиватиметься вперед до стека розповсюдження, і обробник верхнього рівня запізнився.

Частково це вирішується шляхом отримання fooспецифічних типів виключень мого модуля з існуючого типу (наприклад class FooPermissionError(OSError, FooError)), але це не полегшує перенесення існуючого екземпляра винятків у новий тип, а також не змінення повідомлення.

PEP 3134 Python "Виключення ланцюга та вбудовані відстеження" обговорює зміну, прийняту в Python 3.0 для "ланцюжків" об'єктів винятків, щоб вказати, що під час обробки існуючого винятку було піднято нове виключення.

Те, що я намагаюся зробити, пов'язане: мені це потрібно і в більш ранніх версіях Python, і мені це потрібно не для прикування, а лише для поліморфізму. Який правильний спосіб це зробити?

Python 3 представив ланцюжок виключень (як описано в PEP 3134 ). Це дозволяє, піднімаючи виняток, цитувати існуючий виняток як "причину":

try:
    frobnicate()
except KeyError as exc:
    raise ValueError("Bad grape") from exc

Виняте виняток, таким чином, стає частиною (є "причиною") нового винятку і є доступним для будь-якого коду, який сприймає новий виняток.

За допомогою цієї функції встановлюється __cause__атрибут. Вбудований обробник винятків також знає, як повідомити про "причину" та "контекст" виключення разом із простеженням.

У Python 2 , здається, цей випадок використання не має хорошої відповіді (як описано Іен Бікінг та Нед Батчелдер ). Бампер.

Ви можете використовувати sys.exc_info (), щоб отримати зворотний зворот, і підняти новий виняток за допомогою вказаного прослідкування (як згадує PEP). Якщо ви хочете зберегти старий тип і повідомлення, ви можете зробити це за винятком, але це корисно лише в тому випадку, якщо все, що виловило ваше виключення, шукає.

Наприклад

import sys

def failure():
    try: 1/0
    except ZeroDivisionError, e:
        type, value, traceback = sys.exc_info()
        raise ValueError, ("You did something wrong!", type, value), traceback

Звичайно, це насправді не так корисно. Якби це було, нам цей ПЕП не знадобився. Я б не рекомендував це робити.

Ви можете створити свій власний тип виключення, який поширюється на той виняток, який ви потрапили.

class NewException(CaughtException):
    def __init__(self, caught):
        self.caught = caught

try:
    ...
except CaughtException as e:
    ...
    raise NewException(e)

Але більшу частину часу я вважаю, що було б простіше зловити виняток, обробити його і або raiseвихідний виняток (і зберегти прослідкування), або raise NewException(). Якби я дзвонив вашому коду, і я отримав одне з ваших спеціальних винятків, я би сподівався, що ваш код уже обробляє будь-який виняток, який вам довелося спіймати. Тому мені не потрібно самостійно отримувати доступ до нього.

Редагувати: Я знайшов цей аналіз способів викинути власний виняток і зберегти початковий виняток. Немає гарних рішень.

Я також виявив, що мені багато разів потрібна певна обгортка для виниклих помилок.

Це включало як в область функцій, так і іноді загортати лише деякі рядки всередині функції.

Створена обгортка для використання decoratorта context manager:

Впровадження

import inspect
from contextlib import contextmanager, ContextDecorator
import functools    

class wrap_exceptions(ContextDecorator):
    def __init__(self, wrapper_exc, *wrapped_exc):
        self.wrapper_exc = wrapper_exc
        self.wrapped_exc = wrapped_exc

    def __enter__(self):
        pass

    def __exit__(self, exc_type, exc_val, exc_tb):
        if not exc_type:
            return
        try:
            raise exc_val
        except self.wrapped_exc:
            raise self.wrapper_exc from exc_val

    def __gen_wrapper(self, f, *args, **kwargs):
        with self:
            for res in f(*args, **kwargs):
                yield res

    def __call__(self, f):
        @functools.wraps(f)
        def wrapper(*args, **kw):
            with self:
                if inspect.isgeneratorfunction(f):
                    return self.__gen_wrapper(f, *args, **kw)
                else:
                    return f(*args, **kw)
        return wrapper

Приклади використання

декоратор

@wrap_exceptions(MyError, IndexError)
def do():
   pass

при виклику doметоду, не хвилюйтеся IndexError, простоMyError

try:
   do()
except MyError as my_err:
   pass # handle error 

контекстний менеджер

def do2():
   print('do2')
   with wrap_exceptions(MyError, IndexError):
       do()

всередині do2, у context manager, якщо IndexErrorпіднято, воно буде обгорнуте і піднятоMyError

Найбільш прямим рішенням ваших потреб повинно бути таке:

try:
     upload(file_id)
except Exception as upload_error:
     error_msg = "Your upload failed! File: " + file_id
     raise RuntimeError(error_msg, upload_error)

Таким чином ви зможете пізніше надрукувати своє повідомлення та конкретну помилку, яку викинула функція завантаження