腾讯课堂Coding学院：浅入深谈|Python的装饰器（decorator）

微信邦

来源 / 菜鸟分析

编辑 / Coding学院

请编写一个decorator，能在函数调用的前后打印出'begin call'和'end call'的日志

1. import functools
   
2. 
   
3. def log(text):
   
4.         if isinstance(text,str):
   
5.                 def decorator(func):
   
6.                         functools.wraps(func)
   
7.                         def wrapper(*args,**kwargs):
   
8.                                 func(*args,**kwargs)
   
9.                                 print('%s %s' %(text,func.__name__))
   
10.                         print('%s %s' %('start',func.__name__))
    
11.                         return wrapper
    
12.                 return decorator
    
13.         else:
    
14.                 functools.wraps(func)
    
15.                 def wrapper(*args,**kwargs):
    
16.                         print('call %s:' %(func.__name__))
    
17.                         return func(*args,**kwargs)
    
18.                 return wrapper

复制代码

1. >>> @log('end')
   
2. def now():
   
3.         print('2018-01-28')
   
4.         
   
5. start now
   
6. >>> now()
   
7. 2018-01-28
   
8. end now

复制代码

其中，#@log放到now()函数的定义处，相当于执行了语句：now = log(now)，而且now = log('end')(now)

装饰器本质上是一个Python函数，它可以让其它函数在不作任何变动的情况下增加额外功能，装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景。比如：插入日志、性能测试、事务处理、缓存、权限校验等。有了装饰器我们就可以抽离出大量的与函数功能无关的雷同代码进行重用。

在面向对象（OOP）的设计模式中，decorator被称为装饰模式。OOP的装饰模式需要通过继承和组合来实现，而Python除了能支持OOP的decorator外，直接从语法层次支持decorator。Python的decorator可以用函数实现，也可以用类实现。

decorator可以增强函数的功能，定义起来虽然有点复杂，但使用起来非常灵活和方便。