需求
1、公司又很多网站,每个页面用一个函数来代替,之前任何人都能登入。
2、现在要改变每个页面的功能,其中有两个页面需要验证用户登录。
3、页面功能:a、首页不需要登录;b、home页面和bbs页面需要登录。
代码1、带函数返回值的装饰器:
1、问题1:原码中调用home时,会有返回值"from home"的,但是在装饰器过程中,return返回值会没有。
2、解决方式:由于home是在func的时候调用的,所以需要在func后面定义个变量,print返回结果。
1 def auth(func): 2 def wrapper(*args,**kwargs): 3 username = input("username:").strip() 4 password = input("password:").strip() 5 6 if user==username and passwd == password: 7 print("\033[32;1mUser has passed authentication\033[0m") 8 #func(*args,**kwargs) #没有返回值的时候用 9 #return func(*args,**kwargs) #如果函数在这里结束了,就可以return。 10 res = func(*args,**kwargs) #也可以赋值变量,然后做其他的修饰,最后在return返回。 11 print("the home is return: %s" % res) 12 return res #调用函数home时,可以返回值 13 else: 14 exit("\033[31;1mInvalid username or password\033[0m") 15 16 return wrapper 17 18 19 def index(): 20 print('welcome to index page') 21 22 @auth 23 def home(): 24 print('welcome to home page') 25 return "from home" #原代码中是有返回值的。 26 27 @auth 28 def bbs(): 29 print('welcome to bbs page') 30 31 32 index() 33 print(home()) #原代码可以继续调用函数的返回值 34 bbs()
多种权限认证
1、用一个函数体实现多种方式进行认证:
2、模拟本地用户名和密码和ldap用户认证密码:
1 #!/usr/bin/env python 2 # -*- coding:utf8 -*- 3 # Author:Dong Ye 4 5 6 import time 7 user,passwd = 'dy','asd' 8 #装饰器 9 def auth(auth_type): #定义一个装饰器标签函数,auth_type是装饰器的标签(auth_type = "local" & auth_type="ldap") 10 print("auth func:",auth_type) 11 def outer_wrapper(func): #定义接收@auth传过来的home的内存地址,并赋值给func 12 def wrapper(*args,**kwargs): #定义嵌套函数,做功能装饰home() & bbs(),并且home(args1) & bbs(args1)的参数会传到这里 13 #print("wrapper func args:",*args,**kwargs) 14 if auth_type == 'local': 15 username = input('username: ').strip() 16 password = input('password: ').strip() 17 if user == username and passwd == password: 18 print("\033[32;1mUser has passed authentication\033[0m") 19 return func(*args,**kwargs) #执行源代码home() & bbs(),并且home(args1) & bbs(args1)的参数会传到这里 20 else: 21 exit("\033[31;1mInvalid username or password\033[0m") 22 23 elif auth_type=='ldap': 24 print('是以ldap的方式登录') 25 return wrapper 26 return outer_wrapper 27 28 #注释: 29 #home----传参:内存地址和装饰器标签(local&ldap)----> auth(解封装),auth_type=装饰器标签 -----home的内存参数---->outer_warpper(解封装),func= home---->调用wrapper(*args,**kwargs)-->func函数func(*args,**kwargs)=home(*args,**kwargs) 30 #wrapper ----内存地址1---->outer_wrapper----内存地址1和2---->auth--内存地址(1、2)通过装饰器标签(local&ldap)---->home重新赋值 31 def index(): 32 print('welcome to index page') 33 34 #原代码 35 #home认证用本地认证 36 #auth_type = "local" 赋值给了auth(auth_type) 37 #@auth home = auth(outer_wrapper(func)) func = home 38 #通过高阶函数的功能将wrapper的内存地址返回给outer_wrapper 39 #最后将outer_wrapper的内存地址返回给auth函数,并重新赋值给home 40 @auth(auth_type = "local") 41 def home(): 42 print('welcome to home page') 43 return "from home" 44 45 46 #bbs用远程的ldap 47 @auth(auth_type="ldap") #bbs = auth(bbs) 48 def bbs(): 49 print('welcome to bbs page') 50 51 52 53 54 55 #原调用 56 index() 57 home() #其实执行的是通过wraooer、outer_warpper、auth返回的内存地址,找到wrapper函数体执行的 58 bbs()