一、前言
修饰器是啥?大佬太多,这里不赘述,只记录一下今天自己写对类里面的方法如何修饰。
二、修饰器如何对类里面的方法进行修饰?
1、我这里是在不同的文件夹下不同的py文件中,先定义一个修饰器,如下:
def logs_decorator(func): def logs_d(): print("成功获取了列表......") func() return logs_d
2、另一个py文件中,定义一个类,类中还有方法(我这里是做接口测试,所以会有requests库):
import requests from common.logsDecorator import logs_decorator class getBorrowList(object): @logs_decorator def test_get_borrow_list(self): url = "http://xxx/borrow/list" headers = {"Content-Type": "application/x-www-form-urlencoded"} data = { "page": 1, "post_style": 0, "sign": "8de24e3b196446" } r = requests.post(url=url, json=data, headers=headers) print(r.text) print(r.status_code)
详解:上述代码正确写法上是没有问题的,方法上面直接取得修饰器的名称进行调用。但是因为类这个对象的原因,需要在这个方法上添加self,那么这就麻烦了,按照上面的代码,我们直接执行,会出错 :
提示修饰器中的方法缺少一个参数。
解决方法: 说起来也是很简单,我们直接在修饰器中添加一个参数就可以,如下:
def logs_decorator(func): def logs_d(self): print("成功获取了列表......") func(self) return logs_d