Python之道（2）---Python装饰器

第二篇，来说说装饰器，在介绍装饰器之前先要理解闭包的概念。

闭包

内部函数对外部环境变量（但不是全局作用域变量）进行引用，内部函数就是一个闭包。
- 条件1:内部函数
- 条件2:外部环境变量
闭包就是满足这两个条件的函数
注意：闭包中不能修改外部作用域的局部变量

程序示例：

def addbase(x):
    def addp(y):
        return x+y
    return addp

f=addbase(10)
print(f(8))
print(addbase(10)(9))

运行结果：

18
19

说完闭包，理解装饰器就比较简单了。

装饰器

装饰器（decorator）可以用来增强函数的功能，同时不改变原函数的定义，使用起来非常方便。

程序示例1：

#在不改变sum_1函数的基础上我们希望加入附加功能，
#检验参数，当参数为负数时，返回零值
def dec_1(func):
    def jianyan(num1,num2):
        #参数检验
        if num1<0 or num2<0:
            print("参数最好不要小于0")
            return 0
        return func(num1,num2)
    return jianyan
#定义普通函数
def sum_1(num1,num2):
    return  (num1+num2)

sum_1=dec_1(sum_1)
print(sum_1(-2,0))
print(sum_1(5,4))

运行结果：

参数最好不要小于0
0
9

上面的定义过程略显复杂，Python提供了更加简洁方便的方式，使用@，上面的程序可以改写为下面这样，功能不变：

def dec_1(func):
    def jianyan(num1,num2):
        #参数检验
        if num1<0 or num2<0:
            print("参数最好不要小于0")
            return 0
        return func(num1,num2)
    return jianyan
#使用@语法糖的方式
@dec_1
#定义普通函数
def sum_1(num1,num2):
    return  (num1+num2)

#sum_1=dec_1(sum_1)
print(sum_1(-2,0))
print(sum_1(5,4))

程序运行结果和上面保持一致：

参数最好不要小于0
0
9

装饰器还可以使用更加通用的定义方式，去装饰各种类型的函数，在这里仅仅用于演示我们写了一个比较简单的例子，用装饰器去输出每个函数的参数。

def dec_2(func):
    def printer(*args, **kwargs):
        print("参数是：%s,%s"%(args, kwargs))
        return(func(*args, **kwargs))
    return printer

@dec_2
def sum(num1,num2):
#求和
    return num1+num2


@dec_2
def fool(num1,num2,num3):
#取最后一个参数返回
    return num3

print(sum(1,2))
print(fool(6,7,8))

运行结果：

参数是：(1, 2),{}
3
参数是：(6, 7, 8),{}
8