Kembalikan pengecualian dengan jenis dan pesan yang berbeda, menjaga informasi yang ada

Saya sedang menulis modul dan ingin memiliki hierarki eksepsi terpadu untuk pengecualian yang dapat dimunculkan (misalnya mewarisi dari FooErrorkelas abstrak untuk semua foopengecualian spesifik modul). Ini memungkinkan pengguna modul untuk menangkap pengecualian khusus tersebut dan menanganinya dengan jelas, jika perlu. Tetapi banyak pengecualian yang diangkat dari modul dinaikkan karena beberapa pengecualian lain; misalnya gagal pada beberapa tugas karena OSError pada file.

Yang saya butuhkan adalah untuk "membungkus" pengecualian yang ditangkap sehingga memiliki jenis dan pesan yang berbeda , sehingga informasi tersedia lebih lanjut atas hierarki propagasi oleh apa pun yang menangkap pengecualian. Tapi saya tidak ingin kehilangan jenis, pesan, dan jejak stack yang ada; itu semua informasi yang berguna untuk seseorang yang mencoba men-debug masalah. Penangan pengecualian tingkat atas tidak baik, karena saya mencoba untuk menghias pengecualian sebelum membuat jalannya lebih jauh ke tumpukan propagasi, dan penangan tingkat atas sudah terlambat.

Ini sebagian diselesaikan dengan menurunkan foojenis pengecualian spesifik modul saya dari jenis yang ada (misalnya class FooPermissionError(OSError, FooError)), tetapi itu tidak membuatnya lebih mudah untuk membungkus contoh pengecualian yang ada dalam jenis baru, atau memodifikasi pesan.

PEP 3134 dari Python 3134 " Chains Pengecualian dan Embedded Tracebacks" membahas perubahan yang diterima dalam Python 3.0 untuk objek pengecualian "chaining", untuk menunjukkan bahwa pengecualian baru dimunculkan selama penanganan pengecualian yang ada.

Apa yang saya coba lakukan terkait: Saya membutuhkannya juga bekerja di versi Python sebelumnya, dan saya membutuhkannya bukan untuk chaining, tetapi hanya untuk polimorfisme. Apa cara yang tepat untuk melakukan ini?

Python 3 memperkenalkan chaining pengecualian (seperti yang dijelaskan dalam PEP 3134 ). Ini memungkinkan, saat mengajukan pengecualian, mengutip pengecualian yang ada sebagai "penyebab":

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

Pengecualian yang tertangkap dengan demikian menjadi bagian dari (adalah "penyebab") pengecualian baru, dan tersedia untuk kode apa pun yang menangkap pengecualian baru.

Dengan menggunakan fitur ini, __cause__atribut ditetapkan. Penangan pengecualian bawaan juga tahu bagaimana melaporkan "penyebab" dan "konteks" pengecualian bersama dengan traceback.

Dalam Python 2 , tampaknya use case ini tidak memiliki jawaban yang baik (seperti yang dijelaskan oleh Ian Bicking dan Ned Batchelder ). Kekecewaan.

Anda dapat menggunakan sys.exc_info () untuk mendapatkan traceback, dan meningkatkan pengecualian baru Anda dengan traceback tersebut (seperti yang disebutkan oleh PEP). Jika Anda ingin mempertahankan jenis dan pesan yang lama, Anda dapat melakukannya dengan pengecualian, tetapi itu hanya berguna jika apa pun yang menangkap pengecualian Anda mencarinya.

Sebagai contoh

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

Tentu saja, ini benar-benar tidak berguna. Jika ya, kita tidak membutuhkan PEP itu. Saya tidak akan merekomendasikan melakukannya.

Anda bisa membuat jenis pengecualian Anda sendiri yang meluas ke mana pun pengecualian yang Anda tangkap.

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

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

Tetapi sebagian besar waktu, saya pikir akan lebih mudah untuk menangkap pengecualian, menanganinya, dan apakah raisepengecualian asli (dan menjaga traceback) atau raise NewException(). Jika saya memanggil kode Anda, dan saya menerima salah satu pengecualian khusus Anda, saya berharap bahwa kode Anda sudah menangani pengecualian apa pun yang harus Anda tangkap. Jadi saya tidak perlu mengaksesnya sendiri.

Sunting: Saya menemukan analisis ini tentang cara melempar pengecualian Anda sendiri dan menyimpan pengecualian asli. Tidak ada solusi cantik.

Saya juga menemukan bahwa berkali-kali saya perlu beberapa "pembungkus" untuk kesalahan yang diajukan.

Ini termasuk dalam lingkup fungsi dan terkadang hanya membungkus beberapa baris di dalam fungsi.

Dibuat pembungkus untuk digunakan decoratordan context manager:

Penerapan

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

Contoh penggunaan

penghias

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

saat memanggil dometode, jangan khawatir IndexError, cukupMyError

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

manajer konteks

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

di dalam do2, di context manager, jika IndexErrordinaikkan, itu akan dibungkus dan dinaikkanMyError

Solusi paling tegas untuk kebutuhan Anda adalah:

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

Dengan cara ini Anda nantinya dapat mencetak pesan Anda dan kesalahan spesifik dilemparkan oleh fungsi unggah