您的位置:首页 > 编程语言 > Python开发

python基础——装饰器

2017-08-30 10:38 85 查看
装饰器:本质是函数,(装饰其他函数)就是为其他函数添加附加功能。

装饰器用途:在不修改原函数源代码和调用方式的情况下,为原函数添加新的功能。

原则:1.不能修改被装饰的函数的源代码

2.不能修改被装饰的函数的调用方式

高阶函数+嵌套函数=》装饰器

举例

这种不能传递参数,不是通用型,通俗的说就是在函数外套一个函数,把原函数当做参数传递进去,并在里头调用加上要实现的功能,最后返回内层函数的地址。

import time
#装饰器
def timer(func):#外层函数,func是形参,把原函数的地址传递进来
def deco():
start_time=time.time()#实现的新功能
func()#地址加了括号就是调用,即调用原函数
stop_time=time.time()#实现的新功能
print("the func run time is ,(stop_time-start_time))"#实现的新功能
return deco#返回内层函数的地址
@timer #test1=timer(test1),必须要有这个,这个其实就是test1=deco
def test1():
time.sleep(3)
print("int the test1")
test1()#test1()=deco(),做了一个转换,之所以这样做的原因是deco()不能直接用,它是一个局部函数


通用型,可以传递任意个参数,应用场景:有些函数需要添加相同的功能,但是它们当中有些需要传递参数进去(不同的原函数有不同的功能,但它们添加的新功能相同),有些不需要,这个时候就需要实现能兼容传递参数和不传递参数的装饰器,下面的例子就是这样的一个通用型装饰器。

import time
def timer(func):
def deco(*args,**kwargs):#注意这里需要传递参数
start_time=time.time()
func(*args,**kwargs)#注意这里需要传递参数
stop_time=time.time()
print("the total time is:",(stop_time-start_time))
return deco
@timer# @timer要紧挨着被装饰函数的上方
def test1():
time.sleep(3)
print("int the test1")
@timer
def test2(name,age):#这个原函数需要传递参数
time.sleep(3)
print("test2:",name,age)
test1()
test2("Jorocco",23)


这种情况是装饰器带有参数,当我们的外层函数有参数的时候,此时需要在外面再加一层

import time
user,passwd="Jorocco","123ab"
def auth(auth_type):
def outer_wrapper(func):
def wrapper(*args,**kwargs):
if auth_type=="local":
username=input("Username:").strip()
password=input("Password:").strip()
if user==username and passwd==password:
print("\033[32;1mUser has passed authentication\033[0m")
res=func(*args,**kwargs)#1号位置,有返回值,就需要有这个
print("---after authenticaion")
return res#1号位置,有返回值,就需要有这个,不然就没有1号位置的返回结果
else:
exit("\033[31;1mInvalid usrname or password\033[0m")
elif auth_type=="ldap":
print("gameover")
return wrapper
return outer_wrapper

def index():
print("welcome to index")
@auth(auth_type="local")#auth_type="local"有这个参数
def home():
print("welcome to home")
return 3# 1号位置,有返回值
@auth(auth_type="ldap")
def bbs():
print("welcome to bbs")

index()
print(home())
bbs()
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python