Python 装饰器

1、装饰器

在不改变源码的情况下修改已经存在的函数，例如增加调试信息，增加日志记录，查看传入参数等。
装饰器实际上是一个将一个函数作为参数并且返回另一个函数的函数

2、装饰器的使用

1. 在需要装饰的函数前添加装饰器的名字   @decoratro_name
2. 可以通过人工赋值即普通函数调用的方式

==推荐使用第一种方式使用装饰器，容易理解且简单方便==

3、示例

(1)、通过装饰器打印输出函数名字同时记录函数开始和结束时间

import datetime
def func_time(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print('Running function:', func.__name__)
        print('start time:',datetime.datetime.now().microsecond)
        result = func(*args, **kwargs)
        print('end time:',datetime.datetime.now().microsecond)
        return result
    return wrapper


@func_time
def my_pow(x,y):
    return pow(x,y)

print(my_pow(1111,10011))

(2)、装饰器本身需要传入参数的例子

import functools
def log(text):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            print('%s %s()' % (text, func.__name__))
            return func(*args, **kwargs)
        return wrapper
    return decorator

@log('execute')
def now():
    print('2018-5-29')

now()
print(now.__name__)

注意：代码中 @functools.wraps(func)的作用是将原始函数的name属性复制到wrapper中，否则调用now.name会返回wrapper而不是now，这将导致一些依赖函数签名的代码执行会出错

(3)、同一个函数可以有多个装饰器

import functools
def document_it(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print('Running function:', func.__name__)
        print('Positional arguments:', args)
        print('Keyword arguments:', kwargs)
        result = func(*args, **kwargs)
        print('result:', result)
        return result
    return wrapper

def square_it(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        return result*result
    return wrapper

@document_it
@square_it
def add_ints(x, y):
    return x+y

print(add_ints(1,5))
#执行结果为：
#Running function: add_ints
#Positional arguments: (1, 5)
#Keyword arguments: {}
#result: 36
#36

#将装饰器顺序更换
@square_it
@document_it
def add_ints(x, y):
    return x+y

print(add_ints(1,5))
#执行结果为：
#Running function: add_ints
#Positional arguments: (1, 5)
#Keyword arguments: {}
#result: 6
#36

注意：靠近函数定义的装饰器会先执行，然后依次执行上面的。