python 学习之:修饰器

2020-09-08 10:47:51 浏览数 (1)

一、前言

        修饰器是啥?大佬太多,这里不赘述,只记录一下今天自己写对类里面的方法如何修饰。

二、修饰器如何对类里面的方法进行修饰?

        1、我这里是在不同的文件夹下不同的py文件中,先定义一个修饰器,如下:

代码语言:javascript复制
def logs_decorator(func):
    def logs_d():
        print("成功获取了列表......")
        func()
    return logs_d

        2、另一个py文件中,定义一个类,类中还有方法(我这里是做接口测试,所以会有requests库):

代码语言:javascript复制
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,那么这就麻烦了,按照上面的代码,我们直接执行,会出错 :

        提示修饰器中的方法缺少一个参数。

解决方法: 说起来也是很简单,我们直接在修饰器中添加一个参数就可以,如下:

代码语言:javascript复制
def logs_decorator(func):
    def logs_d(self):
        print("成功获取了列表......")
        func(self)
    return logs_d

0 人点赞