优化1： 不使用语法糖，在f1中传入多个参数
def f1(func,source):
def wrapper(*args, **kwargs):
name=input("your name: ")
pwd=input("your password: ")
if name=="tom" and pwd=="123":
print("correct name and password! ")
if source=="teachers":
print("欢迎{}老师!".format(name))
elif source=="students":
print("欢迎{}同学!".format(name))
elif source=="company employees":
print("欢迎{}同事!".format(name))
else:
print("please choose proper source")
res=func(*args,**kwargs)
else:
print("wrong name or password!")
return res
return wrapper
def index(*args,**kwargs):
print("index function output is ",*args,**kwargs)
return "the index1 function is done"
def index2(*args,**kwargs):
print("index2 function output is ",*args,**kwargs)
return "the index2 function is done"
def index3(*args,**kwargs):
print("index3 function output is ",*args,**kwargs)
return "the index2 function is done"
index=f1(index,"teachers")
index2=f1(index2,"students")
index3=f1(index3,"company employees")
index2(1,2 )
'''output：
your name: tom
your password: 123
correct name and password!
欢迎tom同学!
index2 function output is 1 2'''
优化2： 回归语法糖，在f1和wrapper外面继续套入函数， 通过在装饰器函数
def source_auth(source):
def f1(func):
def wrapper(*args, **kwargs):
name=input("your name: ")
pwd=input("your password: ")
if name=="tom" and pwd=="123":
print("correct name and password! ")
if source=="teachers":
print("欢迎{}老师!".format(name))
elif source=="students":
print("欢迎{}同学!".format(name))
elif source=="company employees":
print("欢迎{}同事!".format(name))
else:
print("please choose proper source")
res=func(*args,**kwargs)
else:
print("wrong name or password!")
return res
return wrapper
return f1
f1=source_auth(source="teachers")
@f1
def index(*args,**kwargs):
print("index function output is ",*args,**kwargs)
return "the index1 function is done"
f1=source_auth(source="students")
@f1
def index2(*args,**kwargs):
print("index2 function output is ",*args,**kwargs)
return "the index2 function is done"
f1=source_auth(source="company employees")
@f1
def index3(*args,**kwargs):
print("index3 function output is ",*args,**kwargs)
return "the index2 function is done"
# index=f1(index,"teachers")
# index2=f1(index2,"students")
# index3=f1(index3,"company employees")
index2(1,2 )
'''output：
your name: tom
your password: 123
correct name and password!
欢迎tom同学!
index2 function output is 1 2'''
优化3：继续精简
def source_auth(source):
def f1(func):
def wrapper(*args, **kwargs):
name=input("your name: ")
pwd=input("your password: ")
if name=="tom" and pwd=="123":
print("correct name and password! ")
if source=="teachers":
print("欢迎{}老师!".format(name))
elif source=="students":
print("欢迎{}同学!".format(name))
elif source=="company employees":
print("欢迎{}同事!".format(name))
else:
print("please choose proper source")
res=func(*args,**kwargs)
else:
print("wrong name or password!")
return res
return wrapper
return f1
@source_auth(source="teachers") # 此处，也可输入多个变量
def index(*args,**kwargs):
print("index function output is ",*args,**kwargs)
return "the index1 function is done"
@source_auth(source="students")
def index2(*args,**kwargs):
print("index2 function output is ",*args,**kwargs)
return "the index2 function is done"
@source_auth(source="company employees")
def index3(*args,**kwargs):
print("index3 function output is ",*args,**kwargs)
return "the index2 function is done"
index2(1,2 )
'''output：
your name: tom
your password: 123
correct name and password!
欢迎tom同学!
index2 function output is 1 2'''
