python：クラス内のエラー発生時に特定の処理をさせる

Question

実現したいこと

class内のmethodでエラーが発生したときに、class内の指定したmethodを必ず実行するようにする簡潔な方法はありますでしょうか。

詳細

基本的にはtry文がよくあるやり方だと思うのですが、私が思いついた書き方(下記コードを参照)では、非常に冗長なコードになってしまいます。

一般的な書き方

################# 書き方１ #############
## 問題点
## メソッドを使うたびにtry文が必要
class Hoge_v1:
    def 問題があるメソッド(self):
        print("問題があるメソッドが実行される")
        ここでエラー発生

    def エラーが起きたときに実行させたいメソッド(self):
        print("エラー発生時の処理")


例１ = Hoge_v1()

try:
    例１.問題があるメソッド()
except:
    例１.エラーが起きたときに実行させたいメソッド()

"""出力例
問題があるメソッドが実行される
エラー発生時の処理
"""


################# 書き方２ #############
## 問題点
## クラス内のメソッドが複数ある場合に、全てにtry文が必要
class Hoge_v2:
    def 問題があるメソッド(self):
        try:
            print("問題があるメソッドが実行される")
            ここでエラー発生
        except:
            self.エラーが起きたときに実行させたいメソッド(self)

    def 問題があるメソッド２(self):
        try:
            print("問題があるメソッドが実行される")
            ここでエラー発生
        except:
            self.エラーが起きたときに実行させたいメソッド(self)

    def エラーが起きたときに実行させたいメソッド(self):
        print("エラー発生時の処理")


例２ = Hoge_v2()

try:
    例２.問題があるメソッド()
except:
    例２.エラーが起きたときに実行させたいメソッド()

"""出力例
問題があるメソッドが実行される
エラー発生時の処理
"""

その他

クロージャーやデコレーターを用いたらうまく書けそうな気もするのですが思いつきませんでした。

Answer 1

デコレータを使う例です。

class MyClass:
    def my_exception(func):
        def wrapper(self, *args, **kwargs):
            try:
                func(self, *args, **kwargs)
            except Exception as e:
                ''' 
                共通の例外処理をここに書く
                '''
                print(f"{self.value}にて例外発生")
                print(e)
                print()
            finally:
                pass
        return wrapper
  
    @my_exception
    def bar(self, addition=''):
        self.value = "bar" + addition
        a="#"
        a+0
        return

    @my_exception
    def baz(self):
        self.value = "baz"
        1/0
        return

    my_exception = staticmethod( my_exception )



# 実行例

m = MyClass()
m.bar()
m.bar("-abc")
m.bar(addition="123")
m.baz()

Comments

[BTB]

@sfdustさん
やはりデコレーダーでできたのですね。
求めていたことができました。
助かりました、ありがとうございます。