他のクラスのメソッドをデコレートするデコレータクラスを実装しようとしています。ただし、デコレータで使用可能なdecoratedメソッドを保持するクラスが必要です。どこにも見つからないようです。
次に例を示します。
class my_decorator(object):
def __init__(self, arg1, arg2):
print(self.__class__.__name__ + ".__init__")
self.arg1 = arg1
self.arg2 = arg2
def __call__(self, my_callable):
print(self.__class__.__name__ + ".__call__")
print(type(my_callable))
self.my_callable = my_callable
# self.my_callable_method_class = ?where to get this?
def function_wrapper(*args, **kwargs):
print(self.__class__.__name__ + ".function_wrapper")
print(self.arg1)
self.my_callable.__call__(*args, **kwargs)
print(self.arg2)
return function_wrapper
class MyClass(object):
@my_decorator(arg1="one", arg2="two")
def decorated_method(self):
print(self.__class__.__name__ + ".decorated_method")
print(type(self.decorated_method))
print("hello")
m = MyClass()
m.decorated_method()
これを印刷します:
my_decorator.__init__
my_decorator.__call__
<type 'function'>
my_decorator.function_wrapper
one
MyClass.decorated_method
<type 'instancemethod'>
hello
two
デコレータクラスでは、呼び出し可能オブジェクトは関数型ですが、クラス自体の内部ではインスタンスメソッド型です。instancemethodからim_classを取得できますが、関数にはそのようなものはありません。
デコレータ内からdecoratedメソッドを含むクラスを取得するにはどうすればよいですか?
私はこれを行うことができます:
class my_decorator(object):
def __init__(self, cls, arg1, arg2):
.
.
class MyClass(object):
@my_decorator(cls=MyClass, arg1="one", arg2="two")
def decorated_method(self):
.
.
しかし、それは冗長であり、良くないので、私はそれをしたくありません。
または、これを別の方法で実装する必要がありますか?基本的に、デコレータに対していくつかの引数が必要です。また、デコレータのdecoratedメソッドのクラスが必要です。