Python：decorator装饰器的使用示例

定义一个装饰器

def decorator(func): def wrapper(*arg, **kwargs): print("before") ret = func(*arg, **kwargs) print("after") return ret return wrapper


  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
 

使用示例

# -*- coding: utf-8 -*-


# 作用于普通方法
@decorator
def foo(name): print('my name is:', name)


class Foo(object): # 作用于实例方法 @decorator def foo(self, name): print('my name is:', name) # 作用于类方法 @classmethod @decorator def class_foo(cls, name): print('my name is:', name) # 作用于静态方法 @staticmethod @decorator def static_foo(name): print('my name is:', name)

if __name__ == '__main__': foo('Tom') Foo().foo('Tom') Foo.class_foo('Tom') Foo.static_foo('Tom')

"""
4中方式都可以正常执行，输入如下
before
my name is: Tom
after
"""

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
  
25
  
26
  
27
  
28
  
29
  
30
  
31
  
32
  
33
  
34
  
35
  
36
  
37
  
38
  
39
  
40
  
41
  
42
 

如果把装饰器函数参数修改了

def decorator(func): def wrapper(name): print("before") # ret = func(*arg, **kwargs) # 取消原来的不定参数，写为固定参数 ret = func(name) print("after") return ret return wrapper


  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
 

执行结果

# 普通函数可以正常执行
foo('Tom')

# 静态方法可以正常执行
Foo.static_foo('Tom')

# 实例方法报错
Foo().foo('Tom')
# TypeError: wrapper() takes 1 positional argument but 2 were given

# 类方法报错
Foo.class_foo('Tom')
# TypeError: wrapper() takes 1 positional argument but 2 were given


  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
 

综上，一般情况下需要写成不定参数的形式，兼容性更强

参考
详解Python的装饰器

文章来源: pengshiyu.blog.csdn.net，作者：彭世瑜，版权归原作者所有，如需转载，请联系作者。

原文链接：pengshiyu.blog.csdn.net/article/details/112667286