笔记——python装饰器的用法

装饰器可以简单理解为修饰，为函数添加额外的功能，但是又不用修改原来函数的代码，只需要在原来函数定义前面使用@语法糖即可。它经常用于插入日志等类似的场景，可以达到重复使用而又不修改主要函数代码的目的。

看第一个例子：一个函数

def printName():
    print('LiHaiyu')

printName()

输出：LiHaiyu

看第二个例子：简单的装饰器函数

def mydecorator(f):
    def wrapper(): # 定义修饰函数
        print('Before calling the function')
        f()
        print('After calling the function')
    return wrapper #注意返回函数形成闭包

@mydecorator
def printName():
    print('LiHaiyu')

printName()

输出：

提醒：@mydecorator 必须置于被修饰函数定义之前。相当于将 printName 函数传递给装饰器函数，当调用 printName() 时相当于执行装饰器函数。

第三个例子：函数带参数

def mydecorator(f):
    def wrapper(*args, **kwargs):
        print('Before calling the function')
        f(*args, **kwargs)
        print('Before calling the function')
    return wrapper

@mydecorator
def printName(name):
    print(name)

printName("LiHaiyu")

输出：同上

提醒：*args 表示传递一个列表参数，**kwargs 表示传递一个字典参数。所以在 call function 的时候也应该传递参数

第四个例子：装饰器带参数

def mydecorator(msg = 'Message'):
    def decorated(f):
        def wrapper(*args,**kwargs):
            print('The message is : '+ msg)
            print('Before calling the function')
            f(*args,**kwargs)
            print('After calling the function')
        return wrapper
    return decorated

@mydecorator(msg = 'print name')
def printName(name):
    print(name)

printName('LiHaiyu')

输出：

提醒：返回函数

参考链接：https://www.youtube.com/watch?v=mZ5IwFfqvz8

笔记——python装饰器的用法

猜你喜欢