Nâng cao ngoại lệ với một loại và thông báo khác, lưu giữ thông tin hiện có

Tôi đang viết một mô-đun và muốn có một hệ thống phân cấp ngoại lệ thống nhất cho các ngoại lệ mà nó có thể nêu ra (ví dụ: kế thừa từ một FooErrorlớp trừu tượng cho tất cả các foongoại lệ cụ thể của mô-đun). Điều này cho phép người dùng mô-đun nắm bắt những ngoại lệ cụ thể đó và xử lý chúng một cách rõ ràng, nếu cần. Nhưng nhiều trường hợp ngoại lệ được nêu ra từ mô-đun được nêu ra vì một số ngoại lệ khác; ví dụ: thất bại ở một số tác vụ vì OSError trên một tệp.

Những gì tôi cần là để bọc lại, ngoại lệ bắt được sao cho nó có một loại và thông điệp khác , để thông tin có sẵn hơn nữa trong hệ thống phân cấp lan truyền bởi bất cứ điều gì bắt được ngoại lệ. Nhưng tôi không muốn mất loại, tin nhắn và dấu vết ngăn xếp hiện có; đó là tất cả thông tin hữu ích cho ai đó đang cố gắng gỡ lỗi. Trình xử lý ngoại lệ cấp cao nhất là không tốt, vì tôi đang cố gắng trang trí ngoại lệ trước khi nó tiến xa hơn đến ngăn xếp lan truyền và trình xử lý cấp cao nhất đã quá muộn.

Điều này được giải quyết một phần bằng cách lấy foocác loại ngoại lệ cụ thể của mô-đun của tôi từ loại hiện có (ví dụ class FooPermissionError(OSError, FooError)), nhưng điều đó không làm cho việc bọc trường hợp ngoại lệ hiện tại trong một loại mới cũng như sửa đổi thông báo dễ dàng hơn.

Chuỗi PEP 3134 của Python và Tracebacks nhúng ngoại vi thảo luận về một thay đổi được chấp nhận trong Python 3.0 cho các đối tượng ngoại lệ của Chained, để chỉ ra rằng một ngoại lệ mới đã được đưa ra trong quá trình xử lý ngoại lệ hiện có.

Những gì tôi đang cố gắng làm có liên quan: Tôi cần nó cũng hoạt động trong các phiên bản Python trước đó và tôi cần nó không phải để xâu chuỗi, mà chỉ cho đa hình. Cách đúng đắn để làm điều này là gì?

Python 3 đã giới thiệu chuỗi ngoại lệ (như được mô tả trong PEP 3134 ). Điều này cho phép, khi đưa ra một ngoại lệ, trích dẫn một ngoại lệ hiện có là nguyên nhân gây ra lỗi:

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

Do đó, ngoại lệ bị bắt trở thành một phần của (là nguyên nhân của hệ thống), ngoại lệ mới và có sẵn cho bất kỳ mã nào bắt được ngoại lệ mới.

Bằng cách sử dụng tính năng này, __cause__thuộc tính được đặt. Trình xử lý ngoại lệ tích hợp cũng biết cách báo cáo ngoại lệ của Nguyên nhân và bối cảnh của vụ ngoại lệ cùng với truy nguyên.

Trong Python 2 , có vẻ như trường hợp sử dụng này không có câu trả lời hay (như được mô tả bởi Ian Bicking và Ned Batchelder ). Bummer.

Bạn có thể sử dụng sys.exc_info () để lấy dấu vết và đưa ra ngoại lệ mới của bạn với dấu vết đã nói (như PEP đề cập). Nếu bạn muốn giữ nguyên kiểu và tin nhắn cũ, bạn có thể làm như vậy với ngoại lệ, nhưng điều đó chỉ hữu ích nếu bất cứ điều gì bắt được ngoại lệ của bạn tìm kiếm nó.

Ví dụ

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

Tất nhiên, điều này thực sự không hữu ích. Nếu có, chúng tôi sẽ không cần PEP đó. Tôi không khuyên bạn nên làm điều đó.

Bạn có thể tạo loại ngoại lệ của riêng bạn mở rộng bất kỳ ngoại lệ nào bạn đã bắt gặp.

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

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

Nhưng hầu hết thời gian, tôi nghĩ sẽ đơn giản hơn để bắt ngoại lệ, xử lý nó và raisengoại lệ ban đầu (và giữ nguyên dấu vết) hoặc raise NewException(). Nếu tôi đang gọi mã của bạn và tôi đã nhận được một trong những trường hợp ngoại lệ tùy chỉnh của bạn, tôi hy vọng rằng mã của bạn đã xử lý bất kỳ ngoại lệ nào bạn phải nắm bắt. Vì vậy, tôi không cần phải truy cập nó.

Chỉnh sửa: Tôi tìm thấy phân tích này về các cách để ném ngoại lệ của riêng bạn và giữ ngoại lệ ban đầu. Không có giải pháp đẹp.

Tôi cũng thấy rằng nhiều lần tôi cần một số "gói" cho các lỗi được nêu ra.

Điều này bao gồm cả trong một phạm vi chức năng và đôi khi chỉ bao gồm một số dòng bên trong một chức năng.

Tạo một trình bao bọc được sử dụng a decoratorvà context manager:

Thực hiện

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

Ví dụ sử dụng

người trang trí

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

Khi gọi dophương thức, đừng lo lắng IndexError, chỉ cầnMyError

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

quản lý bối cảnh

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

bên trong do2, trong context manager, nếu IndexErrorđược nâng lên, nó sẽ được bọc và nâng lênMyError

Giải pháp tối ưu nhất cho nhu cầu của bạn nên là:

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

Bằng cách này, sau này bạn có thể in tin nhắn của mình và lỗi cụ thể được ném bởi chức năng tải lên