Python クラス内のデコレータメソッドを継承できない

Question

実現したいこと

親クラス内のデコレータメソッドを子クラスで使用したいのですが、想定していたようにできません。どのように書くのが適切でしょうか？

該当のソースコード

class A:
    def decorator(f):
        def wrapper(self, *args, **kwargs):
            print("wrapper")
            f(self, *args, **kwargs)
            return
        return wrapper

class B(A):
    @decorator
    def b(self):
        print("b")
        return

b = B()
b.b()

結果

Traceback (most recent call last):
  File "main.py", line 9, in <module>
    class B(A):
  File "main.py", line 10, in B
    @decorator
NameError: name 'decorator' is not defined

試したこと

デコレータの指定方法を変更すれば動きました。しかしこれがベストではない気がしています。
「ベストではない」と言ったのは、簡潔に言えば親クラス名を複数回書いているためにリファクタリングの観点から問題があると考えたからです。
また、decoratorメソッドをオーバーライドする必要がある際、@~も明示的に変更する必要が生じてしまいます。

class A:
    def decorator(f):
        def wrapper(self, *args, **kwargs):
            print("wrapper")
            f(self, *args, **kwargs)
            return
        return wrapper

class B(A):
    @A.decorator
    def b(self):
        print("b")
        return

b = B()
b.b()

結果

wrapper
b

補足情報（FW/ツールのバージョンなど）

Python: 3.8.5

Comments

[退会済みユーザー]

「しかしこれがベストではない気がしています。」「どのように書くのが適切でしょうか」とのことですが記載の修正後のコードで目的は達せられていると思いますが、どのような点で「ベストではない」とお考えなのでしょうか？逆に言えば、あなたが考える「ベスト」「適切」に近づけるための方向性の基準は何なのでしょうか（コードを現状より少しでも短くしたい、等）

[hoboki]

コメントありがとうございます。
簡潔に言えば親クラス名を2回書いている点です。リファクタリングの観点から問題があると思います。
また、decoratorメソッドをオーバーライドする必要がある際、ただオーバーライドするだけでは済まなくなっています。

Answer 1

検索してみると以下のサイトがみつかりました。
https://stackoverflow.com/questions/7473096/python-decorators-how-to-use-parent-class-decorators-in-a-child-class

読むと、デコレータそのものはクラスAのメソッドではないので、クラスの外に出せばよいという回答になっています。
デコレータが返す関数はメソッドになりますが、もともとpythonではメソッドをクラスの外で定義できる(selfという変数名にも特別の意味はない)ので、このままの形で使えます。

class A():
    pass


def decorator(f):
    def wrapper(self, *args, **kwargs):
        print("wrapper")
        f(self, *args, **kwargs)
        return
    return wrapper    


class B(A):
    @decorator
    def b(self):
        print("b")
        return    


b = B()
b.b()

Comments

[hoboki]

丁寧なご説明ありがとうございます。大変勉強になりました。