◎知识点
装饰器概述及使用
◎脚本练习
"""装饰器""" """ 对于某个函数,如果我们希望在不改变该函数代码的前提下,为该函数增加额外的功能,那么就可以使用 装饰器来装饰该函数。 装饰器是一个函数,装饰器接收一个函数作为参数(传入的实参是被装饰的函数),装饰器的内部嵌套定义 另一个函数,内函数中会引用装饰器的参数,并且装饰器的返回值是内函数。这样,就构成了一个闭包。 为了让内函数接收任意类型的参数,将内函数的形参定义为(*args, **kwargs)。在函数中,首先完成 为被装饰函数添加的新功能,然后调用被装饰的函数。 把装饰器应用到被装饰函数的语法为:在被装饰函数的前面添加"@装饰器的函数名"。在被装饰函数add 的前面添加@log后,相当于执行了语句:add = log(add),首先,被装饰的函数add会作为实参传递给 装饰器log,然后,返回装饰器的内函数wrapper,最后,将内函数wrapper赋值给名为add(被装饰函数的函数名) 的变量,这样,再调用被装饰的函数add时,其实调用的是装饰器的内函数wrapper。 """ def log(func): def wrapper(*args, **kwargs): print("函数%s被调用了" % func.__name__) return func(*args, **kwargs) return wrapper @log def add(sum1, sum2): print(sum1, sum2) return sum1 + sum2 print(add(1, 2)) print(add.__name__) # wrapper """ 如果希望被装饰函数的特殊属性__name__的值为其函数名,而不是装饰器的内函数的函数名, 可以在装饰器的内函数前面添加另外一个装饰器:@functools.wraps(装饰器的参数名), 其中functools.wraps指的是标准库functools中的函数wraps。 """ import functools def log2(func): @functools.wraps(func) def wrapper(*args, **kwargs): print("函数%s被调用了" % func.__name__) return func(*args, **kwargs) return wrapper @log2 def add2(sum1, sum2): print(sum1, sum2) return sum1 + sum2 print(add2(1, 2)) print(add2.__name__) # add2 """ 把装饰器应用到被装饰函数时,还可以传递额外的参数。此时,需要编写一个3层嵌套的装饰器。 对于@log3('6月', '18日'),相当于执行语句:add3 = log3(6月', '18日')(add3)。 """ def log3(month, day): def decorator(func): @functools.wraps(func) def wrapper(*args, **kwargs): print("%s%s, 函数%s被调用了" % (month, day, func.__name__)) return func(*args, **kwargs) return wrapper return decorator @log3('6月', '18日') def add3(sum1, sum2): print(sum1, sum2) return sum1 + sum2 print(add3(1, 2))