python 装饰器

python中提供的装饰器（decorator)作为修改函数的一种便捷的方式。

装饰器本质上就是一个函数，这个函数接受其他的函数作为参数，并将其以一个新的修改后的函数进行替换。

（一）我们首先定义一个最简单的函数

# -*- coding: utf-8 -*-
"""
Created on Mon Oct 26 18:00:41 2015
@author: zenwan
"""def square_sum(a, b):
    return a**2 + b**2

（二）定义一个装饰函数，使用语法糖@来装饰函数

使用语法糖@来装饰函数，相当于square_sum = deco(square_sum)

但发现新函数只在第一次被调用，且原函数多调用一次。

示例代码：

# -*- coding: utf-8 -*-
"""
Created on Mon Oct 26 18:00:41 2015

@author: zenwan
"""


def deco(fun):
    print "hello"
    return fun


@deco
def square_sum(a, b):
    return a**2 + b**2


print square_sum(3, 4)
print square_sum(3, 4)  # 新函数只在第一次被调用，且原函数多调用了一次

运行结果：

hello
25
25

　（三）使用内嵌的包装函数来确保每次新的函数都可以被调用

注意：内嵌包装函数的形参和返回值与原函数一样，装饰函数返回内嵌包装函数对象

# -*- coding: utf-8 -*-
"""
Created on Mon Oct 26 18:00:41 2015

@author: zenwan
"""


def deco(fun):
    def _deco(a, b):
        print "hello"
        ret = fun(a, b)
        return ret
    return _deco


@deco
def square_sum(a, b):
    return a**2 + b**2

print square_sum(3, 4)
print square_sum(3, 4)

运行结果：

hello
25
hello
25

　　（四） 参数数量不确定的函数进行装饰

inspect.getcallargs 返回一个将参数名字和值作为键值对的字典，这意味着我们的装饰器不必检查参数是基于位置的参数还是基于关键字的参数，而只需要在字典中即可。

示例代码：

# -*- coding: utf-8 -*-
"""
Created on Mon Oct 26 18:00:41 2015

@author: zenwan
"""
import inspect
import functools

def deco(fun):
    @functools.wraps(fun)
    def _deco(*args, **kwargs):
        fun_args = inspect.getcallargs(fun, *args, **kwargs)
        print fun_args
        print "hello"
        return fun(*args, **kwargs)
    return _deco


@deco
def square_sum(a, b):
    return a**2 + b**2

print square_sum(3, 4)
print square_sum(3, 4)

运行结果：

{'a': 3, 'b': 4}
hello
25
{'a': 3, 'b': 4}
hello
25

 （未完待续）