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

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

news 2025/7/5 13:49:43

基本概念

在 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):passdef test(self):print("test")

类装饰器

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

class Foo(object):def __init__(self, func):self._func = funcdef __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