Python 装饰器 装饰类的方法

Python 装饰器 装饰类的方法

在用scrapy写爬虫时，需要装饰器来装饰Spider类中的parse方法，但是parse方法中用到了self.parse这个回掉函数，担心装饰器修改回掉函数self.parse的入口，所以做了个测试，发现装饰器还是很稳的。

测试代码

check是一个类内的方法（不是类方法）的装饰器，因此self是必须写的位置参数，response才是调用时的位置参数。
这里监测了response的值，如果是401，就会输出一下self.parse的地址，并调用它，且让它参数为300。如果不是200，就输出self.parse的地址，结束，不再调用函数本身。如果response是200，就会调用函数本身，并且传递参数进去，注意这里是要写self的。
Spider是一个类，里面有一个叫parse的方法，被check装饰。
测试用例，先输出了一下self.parse的地址，又用三个样例试了三次。
多次输出self.parse是看地址是否有变化。
三个样例含义：

• 200：看看装饰器正确运行内部函数的结果。
• 203：这个参数的输入，被装饰后，会产生不同的行为。
• 401：会先输出自己的信息，随后调用以response=300来调用方法

用401的下次调用和以上200、203结果做对比：如果结果与200相同，表明装饰器不能很好的处理回掉函数；如果结果与203相同，表明装饰器是没问题的。
结果是后者。

结论

一旦类中的方法被装饰器装饰，那么其调用入口就永久变成了装饰器的入口。可以放心使用，各种调用都是没问题的。

import functools
def check(func):
    @functools.wraps(func)
    def inner(self, response, *args, **kwargs):
        if response == 401:
            print(401)
            print(self.parse)
            print("**************************")
            self.parse(300)
        elif response != 200:
            print(200)
            print(self.parse)
            print("**************************")
        else:
            return func(self, response, *args, **kwargs)
    return inner


class Spider():
    @check
    def parse(self, response):
        print('foo')

if __name__ == '__main__':
    s = Spider()
    print(s.parse)
    s.parse(200)
    print("--------------------")
    s.parse(203)
    print("--------------------")
    s.parse(401)

输出如下：

<bound method Spider.parse of <__main__.Spider object at 0x000001AAFB9A7320>>
foo
--------------------
200
<bound method Spider.parse of <__main__.Spider object at 0x000001AAFB9A7320>>
**************************
--------------------
401
<bound method Spider.parse of <__main__.Spider object at 0x000001AAFB9A7320>>
**************************
200
<bound method Spider.parse of <__main__.Spider object at 0x000001AAFB9A7320>>
**************************