当前位置：首页 > article >正文

【代码模板】Python Decorator / 装饰器

article 2024/9/28 12:51:36

基本概念

在 python 里面，函数是一个对象，但是是 callable 的对象，因此后面可以接 () 传入参数。这也说明函数可以被当作参数传入其他的函数里面，同时函数也可以作为返回值。

一般装饰器

可以将 decorator 理解为一个函数，他的输入是 @dec 下面接着的函数，输出由 dec 函数定义决定。

def dec(f): 
	pass 

@dec 
def double(x): 
	return x * 2 

# 上述 @dec 等价于
double = dec(double)

@decorator 还可以带参数。

@timeit(10) 
def double(x): 
	return x * 2 

# 等价于 
double = timeit(10)(double) 
# timeit(10)先执行，通常返回一个函数，然后再输入double进一步执行并返回

装饰类

def decorator(cls):
    print("这里可以写被装饰类新增的功能") 
    return cls

# 定义一个类 A,并使用decorator装饰器装饰
@decorator # 装饰器的本质 A = decorator(A)，装饰器返回类本身，还是之前的类，只是在返回之前增加了额外的功能
class A(object):
    def __init__(self):
        pass

    def test(self):
        print("test")

类装饰器

装饰器本身可以是一个类。使用类装饰器主要依靠类的 __call__ 方法，当使用 @ 形式将装饰器附加到函数上时，就会调用此方法。

class Foo(object):
    def __init__(self, func):
        self._func = func

    def __call__(self):
        print ('class decorator runing')
        self._func()
        print ('class decorator ending')

@Foo
def bar():
    print ('bar')

bar()

# 程序输出:
# class decorator runing
# bar
# class decorator ending