【Python基础】之装饰器

偏执的太偏执、 2022-04-24 13:18 213阅读 0赞

这是我初次接触装饰器，先从初学者的角度介绍装饰器，关于装饰器的应用场景举例，后面再补充。

# 1.装饰器的作用 #

*  装饰器可以让一个函数在不做任何变动的情况下新增额外的功能。

如下代码，func\_name函数是打印传入函数的函数名，f1函数就是打印一个字符串。

def func_name(func):
        """打印传入函数的函数名"""
        print(func.__name__)
    
    def f1():
        print("这是待装饰的函数f1")

现在要求，每次调用一个函数的时候，都要打印该函数的函数名，那么我们可以在f1函数内部作如下修改：

def f1():
        func_name(f1)
        print("这是待装饰的函数")
    
    f1() #f1 这是待装饰的函数

看似很简单，但是实际应用场景中可能有几十个函数需要作这样的修改，都在函数内部作改动是很麻烦的，也容易出错，那么我们可以利用装饰器在f1函数的外部进行修改。

# 2.写一个装饰器 #

如下代码就是对1中的f1函数作一个修改。可以看到我们定义了两个函数func\_name和函数f1，其中我们用func\_name函数装饰了函数f1。

那么在装饰的过程中，f1发生了怎么样的变化呢。首先我们要定义一个装饰器func\_name，然后在待装饰的函数正上方输入@func\_name(注意，这里是@+函数名，不带括号)。

当我们调用函数f1()时，一共发生了三件事，

*  第一，执行func\_name函数，并把@下面一行的函数名称（注意，不带括号，是函数名）传给func\_name，这时func\_name会返回函数名称wrap（注意，是函数名，不带括号），
 *  第二步，将func\_name返回的函数名称wrap赋值给f1，即f1 = func\_name(f1)， 而func\_name(f1)=wrap，所以这时，f1 = wrap()，
 *  第三步，执行f1()函数，即wrap()函数，但是现在wrap()中的func()函数还是原来传入func\_name的f1函数。

def func_name(func):
        def wrap():
            print(func.__name__)
            func()
        return wrap
    
    @func_name
    def f1():
        print("这是待装饰的函数")
    
    f1()# f1 这是待装饰的函数

# 3.带参数的装饰器 #

如果f1带参数呢？比如下方的f1，有一个参数args

def f1(args):
        print(args)
        print("这是待装饰的函数")

那么我们可以在装饰器中的wrap()和func()中也加入相同的参数。

def func_name(func):
        def wrap(args): #加入参数
            print(func.__name__)
            func(args) #加入参数
        return wrap
    
    @func_name
    def f1(args):
        print(args)
        print("这是待装饰的函数")
    
    f1("我是参数")
    """
    输出:
    f1
    我是参数
    这是待装饰的函数
    """

但是如果我们需要用func\_name装饰十几个函数，而十几个函数传入的参数格式、格式都不相同，怎么办呢？这时我们可以在wrap()和func()中传入动态参数\*args, \*\*kwargs，这样勿论待装饰的函数传入多少参数，装饰器都能够正确接受。

有关动态参数见：[https://blog.csdn.net/lincoco49/article/details/89364034][https_blog.csdn.net_lincoco49_article_details_89364034]

def func_name(func):
        def wrap(*args, **kwargs):#动态参数
            print(func.__name__)
            func(*args, **kwargs) #动态参数
        return wrap
    
    
    @func_name
    def f1(args):#一个形参
        print(args)
        print("这是待装饰的函数")
    
    @func_name
    def f2(a, b): #两个形参
        ret = a+b
        print(ret)
    
    f1("我是参数")
    """
    f1
    我是参数
    这是待装饰的函数
    """
    
    f2(1,1)
    """
    f2
    2
    """

# 4.写装饰器易错总结： #

*  @后面跟的是函数名，不加括号
 *  装饰器返回的也是一个函数名，不加括号，加了括号就会执行函数了

[https_blog.csdn.net_lincoco49_article_details_89364034]: https://blog.csdn.net/lincoco49/article/details/89364034