写代码要遵循开放封闭原则,那么什么是开放封闭原则呢,简单的说就是:已经实现的功能代码块不允许被修改,但可以被扩展。即:
开放:对扩展开发;封闭:已经实现的代码块
那么问题来了如何在不更改原有代码前提下实现性能的添加,装饰器就是一个很好的用法
例如:我需要在“登录系统前添加验证功能”,在不更改现有的代码该如何实现?
def test(name):
print("登录系统___%s" % name)
test("张三")
1)方法1:也许有人会想到使用闭包实现,如
def outer(func):
def inner(name):
print("验证——————————")
func(name)
return inner
def test(name):
print("登录系统___%s" % name)
test = outer(test)
test("张三")
#输出结果
验证——————————
登录系统___张三
那么问题来了,这样的话在每个文件不同作用域里调用test功能之前都需要添加第12行: test = outer(test),那如果这样的需要添加位置较多的话也不现实,且比较难于查找,出现缺漏的结果也可以自行想象。这里就可以使用方法2
2)方法2:使用装饰器
def outer(func):
def inner(name):
print("验证——————————")
func(name)
return inner
@outer
def test(name):
print("登录系统___%s" % name)
test("张三")
#输出结果
验证——————————
登录系统___张三
这里可以看到只需要在test函数前加上 @outer 就可以实现,也可以这么理解 @outer --> test = outer(test)
当装饰的函数有参数时装饰器内部也必须要定义形参,即 第 2 行,在第 4 行时也必须传递参数进行调用
多个函数使用同样的验证功能时也是如此
def outer(func):
def inner(name):
print("验证——————————")
func(name)
return inner
@outer
def test1(name):
print("登录系统___%s" % name)
@outer
def test2(name):
print("查询余额___%s" % name)
test1("张三")
test2("李四")
#输出结果
验证——————————
登录系统___张三
验证——————————
查询余额___李四
2、多个装饰器
def outer1(func):
print("_____装饰器outer1____")
def inner1():
print("———inner1———————")
func()
return inner1
def outer2(func):
print("_____装饰器outer2____")
def inner2():
print("———inner2———————")
func()
return inner2
@outer1
@outer2
def test():
pass
test()
#输出结果
_____装饰器outer2____
_____装饰器outer1____
———inner1———————
———inner2———————
从结果也可以看到,当有多个装饰器时,装饰器是从内往外装饰,即:1)@outer2--> test = outer2(test) 2) @outer1--> test = outer1(test)
执行到,15,16开始装饰时就会有输出输出23,24行,原因请看上边1)2),在装饰时也会执行outer2(),outer1()两个函数,所以会有输出结果
3、装饰器带返回值
def outer(func):
def inner(a, b):
ret = func(a, b)
return ret
return inner
@outer
def test(a, b):
return a + b
ret = test(5, 2)
print("-----%d" % ret)
#输出结果
-----7
当装饰的函数有返回值时需要将返回值返回,如第 3,4 行
4、通用装饰器
def outer(func):
def inner(*args, **kwargs):
ret = func(*args, **kwargs)
return ret
return inner
@outer
def test1(a):
return a ** 2
@outer
def test2(a, b, c):
return a * b * c
print(test1(3))
print(test2(2, 3, c=5))
#输出结果
9
30
当多个函数都是用同一个装饰器的时候,参数不一致的问题,就的使用通用装饰器来解决,通用即所有的函数都适用的意思。
如上所示,test1与test2函数的参数不一致,也可以用同样的装饰器装饰,第2,3行解决
5、带参数的装饰器