笔记—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




阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页