[python] decorator implemented with class to wrap class method

# python2
Python 2.7.5 (default, Dec  8 2017, 16:39:59) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-25)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> class tracer:
...     def __init__(self, func):
...             self.func = func
...             self.count = 0
...     def __call__(self, *args, **kargs):
...             self.count += 1
...             return self.func(*args, **kargs)
... 
>>> class C:
...     @tracer
...     def f(self):
...             pass
... 
>>> c = C()
>>> c.f()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in __call__
TypeError: f() takes exactly 1 argument (0 given)
>>> C.f()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in __call__
TypeError: f() takes exactly 1 argument (0 given)
>>> C.f.func
<function f at 0x7f0efb4ad0c8>
>>> c.f.func
<function f at 0x7f0efb4ad0c8>
>>> 

this is because decoration occurs at function defining phase:

@tracer
def f(self):
...

is the same as

f = tracer(f)

when c.f() is called, it is same as tracer(f)(), this won’t trigger function f __get__ method to pass self of c implicitly to f, thus will result in TypeError because no arguments passed.

how to solve this?
in __call__, the instance must be passed as first argument, how to retrieve the instance from call like c.f()? the answer is to use descriptor protocol

class tracer:
    def __init__(self, func):
        self.calls = 0
        self.func = func
    def __call__(self, *args, **kargs):
        self.calls += 1
        print('call %s to %s' % (self.calls, self.func.__name__))
        return self.func(*args, **kargs)
    def __get__(self, instance, owner):
        def wrapper(*args, **kargs):
            return self(instance, *args, **kargs)
        return wrapper
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值