函数作装饰器，类做装饰器

最新推荐文章于 2020-02-04 21:14:00 发布

hanshiqin

最新推荐文章于 2020-02-04 21:14:00 发布

阅读量164

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/hanshiqin/article/details/88790999

本文深入探讨了Python中装饰器的应用，包括函数和类作为装饰器的实现方式，以及如何使用functools库中的wraps装饰器来保留原函数的元信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

函数作装饰器，类做装饰器

一.用类写装饰器

"""
用类写装饰器
func = Decorator(func)
func('abc',18)
"""

class Decorator(object):
def __init__(self, f):
self.f = f
def __call__(self,*args,**kwargs):
print("decorator start")
self.f(*args,**kwargs)
print("decorator end")

@Decorator
def func(name,age):
print("{0}\' is {1}".format(name,age))
print("func")

func('abc',18)

二.用函数写装饰器

'''

用函数写装饰器
func1 = Decorator1(func)
func1('abc',18)
'''

def Decorator1(f):
def wrapper(*args,**kwargs):
print("decorator start")
f(*args, **kwargs)
print("decorator end")
return wrapper

@Decorator1
def func1(name,age):
print("{0}\' is {1}".format(name, age))

func1('abc',18)

三.消除装饰器对函数的影响

def Decorator1(f):

    def wrapper(*args,**kwargs):
        print("decorator start")
        f(*args, **kwargs)
        print("decorator end")
    return wrapper

@Decorator1
def func1(name,age):
    print("{0}\' is {1}".format(name, age))

func1('abc',18)

print("function \'s name is {}".format(func1.__name__))

输出为：

decorator start
abc' is 18
decorator end
function 's name is wrapper

这是由于装饰后的func其实相当于 func= Decorator1(func)

而 func.__name ,也相当于Decorator1(func) .__name__

消除这种影响的办法是使用 functools 工具包中的 wraps 装饰器

 1 from functools import wraps
 2 def Decorator1(f):
 3     @wraps(f)
 4     def wrapper(*args,**kwargs):
 5         print("decorator start")
 6         f(*args, **kwargs)
 7         print("decorator end")
 8     return wrapper
 9 
10 @Decorator1
11 def func1(name,age):
12     print("{0}\' is {1}".format(name, age))
13 
14 func1('abc',18)
15 print("function \'s name is {}".format(func1.__name__))