Python でclass内にdecoratorを持たせたい

Question

###前提・実現したいこと
Python でclass内にdecoratorを持たせたい

###発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "deco2.py", line 1, in <module>
    class MyClass(object):
  File "deco2.py", line 12, in MyClass
    @decorator
TypeError: decorator() takes exactly 2 arguments (1 given)

###該当のソースコード

class MyClass(object):
    def __init__(self):
        self.my_var = "my_var" # 追加しました
	# pass

    def decorator(self, func):
        def inner(*args, **kwargs):
            print "pre func"
            print self.my_var # 追加しました
            func(*args, **kwargs)
            print "after_func"
        return inner

    @decorator  #ここでエラーになります
    def print_hoge(self):
        print "hoge"


temp = MyClass()
temp.print_hoge()

###試したこと

def decorator(func):
    def inner(*args, **kwargs):
        print "pre func"
        func(*args, **kwargs)
        print "after_func"
    return inner


@decorator
def print_hoge():
    print "hoge"

print_hoge()

>>pre_func
>>hoge
>>after_func

###補足情報(言語/FW/ツール等のバージョンなど)
いつも参考にさせていただいております。
非常に簡単な質問かも知れないのですが、実力が足りず解決方法が分からない為教えてください。

【試したこと】の欄のソースでは上手くいくのに、class内に入れた際うまく行きません。

よろしくお願い致します。

Answer 1

decoratorメソッドのself引数を消すといかがですか。

class MyClass(object):
    def __init__(self):
        pass

    def decorator(func):
        def inner(*args, **kwargs):
            print "pre func"
            func(*args, **kwargs)
            print "after_func"
        return inner

    @decorator
    def print_hoge(self):
        print "hoge"


temp = MyClass()
temp.print_hoge()

self 等にアクセスしたい場合は、以下のようにします。

#coding: utf-8

class MyClass(object):
    def __init__(self):
        self.my_var = "my_var" # 追加しました

    def decorator(func):
        def inner(self, *args, **kwargs):
            print "pre func"
            print self.my_var # 追加しました
            func(self, *args, **kwargs)
            print "after_func"
        return inner

    @decorator
    def print_hoge(self, *args, **kwargs):
        print "hoge"


temp = MyClass()
temp.print_hoge()

selfがなくて良い理由ですが、上のコードは実質以下のコードと同義です。
selfを使うのはあくまでselfが定義しているメソッド内で、今回の例ではdecoratorメソッドは素直に参照できます。

    # デコレータをつけないバージョン
    def print_hoge(self, *args, **kwargs):
        print "hoge"
    print_hoge = decorator(print_hoge)
    # print_hoge = self.decorator(print_hoge)  selfをつけるとエラー。

Comments

[super_hogehoge]

ご回答ありがとうございます!!
ただ、すいません、私の例が悪かったです。確かに教えていただいたコードで出力されますが、self引数を消すと、
```python
class MyClass(object):
def __init__(self):
self.my_var = "my_var" # 追加しました
pass

def decorator(func):
def inner(*args, **kwargs):
print "pre func: "
print self.my_var # 追加しました
func(*args, **kwargs)
print "after_func "
return inner

@decorator
def print_hoge(self):
print "hoge"

temp = MyClass()
temp.print_hoge()
```

のようなclass内部の変数にアクセスできず、selfを付けると@self.decoratorと書いた
箇所でエラーになってしまうので困っていました。

例が適切な物ではなくご迷惑をお掛けしました…
上記問題についても知見をお持ちでご教授いただけると非常に助かります。

お手数をお掛けしますがよろしくお願い致します。

[super_hogehoge]

ご回答ありがとうございます！
正直、私の力不足でclass内methodなのにself無くてよいの？等しっくり来ていないところ
もあるのですが　問題は解決していただけたのでご回答をベストアンサーとさせていただきます。
迅速に回答していただきとても助かりました。ありがとうございました！

[super_hogehoge]

>selfを使うのはあくまでselfが定義しているメソッド内で、今回の例では
>decoratorメソッドは素直に参照できます。

なるほど・・・てっきりclass内の関数は全てselfを引数に持たないといけないのかと
誤認していました。回答して下さった上にこちらの疑問にフォローしていただき
非常にすっきり問題を解決することができました。　ありがとうございました！