高级装饰器---验证用户登录

需求

1、公司又很多网站，每个页面用一个函数来代替，之前任何人都能登入。

2、现在要改变每个页面的功能，其中有两个页面需要验证用户登录。

3、页面功能：a、首页不需要登录；b、home页面和bbs页面需要登录。

 

代码1、带函数返回值的装饰器：

1、问题1：原码中调用home时，会有返回值"from home"的，但是在装饰器过程中，return返回值会没有。

2、解决方式：由于home是在func的时候调用的，所以需要在func后面定义个变量，print返回结果。

def auth(func):
    def wrapper(*args,**kwargs):
        username = input("username:").strip()
        password = input("password:").strip()

        if user==username and passwd == password:
            print("\033[32;1mUser has passed authentication\033[0m")
            #func(*args,**kwargs)  #没有返回值的时候用
            #return func(*args,**kwargs)    #如果函数在这里结束了，就可以return。
            res = func(*args,**kwargs)  #也可以赋值变量,然后做其他的修饰，最后在return返回。
            print("the home is return: %s" % res)
            return res  #调用函数home时，可以返回值
        else:
            exit("\033[31;1mInvalid username or password\033[0m")

    return wrapper


def index():
    print('welcome to index page')

@auth
def home():
    print('welcome to home page')
    return "from home"   #原代码中是有返回值的。

@auth
def bbs():
    print('welcome to bbs page')


index()
print(home())  #原代码可以继续调用函数的返回值
bbs()

 

多种权限认证

1、用一个函数体实现多种方式进行认证：
2、模拟本地用户名和密码和ldap用户认证密码：

#!/usr/bin/env python
# -*- coding:utf8 -*-
# Author:Dong Ye

 
import time
user,passwd = 'dy','asd'
#装饰器
def auth(auth_type):  #定义一个装饰器标签函数，auth_type是装饰器的标签（auth_type = "local" & auth_type="ldap"）
    print("auth func:",auth_type)
    def outer_wrapper(func):  #定义接收@auth传过来的home的内存地址，并赋值给func
        def wrapper(*args,**kwargs):   #定义嵌套函数，做功能装饰home() & bbs()，并且home(args1) & bbs(args1)的参数会传到这里
            #print("wrapper func args:",*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")
                    return func(*args,**kwargs)   #执行源代码home() & bbs()，并且home(args1) & bbs(args1)的参数会传到这里
                else:
                    exit("\033[31;1mInvalid username or password\033[0m")
 
            elif auth_type=='ldap':
                print('是以ldap的方式登录')
        return wrapper
    return outer_wrapper
 
#注释：
#home----传参：内存地址和装饰器标签（local&ldap）----> auth(解封装),auth_type=装饰器标签 -----home的内存参数---->outer_warpper（解封装），func= home---->调用wrapper（*args,**kwargs）-->func函数func(*args,**kwargs)=home(*args,**kwargs)
#wrapper ----内存地址1---->outer_wrapper----内存地址1和2---->auth--内存地址（1、2）通过装饰器标签（local&ldap）---->home重新赋值
def index():
    print('welcome to index page')
 
#原代码
#home认证用本地认证
#auth_type = "local" 赋值给了auth(auth_type)
#@auth home = auth（outer_wrapper(func)）   func = home
#通过高阶函数的功能将wrapper的内存地址返回给outer_wrapper
#最后将outer_wrapper的内存地址返回给auth函数，并重新赋值给home
@auth(auth_type = "local")
def home():
    print('welcome to home page')
    return "from home"
 
 
#bbs用远程的ldap
@auth(auth_type="ldap")   #bbs = auth(bbs)
def bbs():
    print('welcome to bbs page')
 
 
 
 
 
#原调用
index()
home() #其实执行的是通过wraooer、outer_warpper、auth返回的内存地址，找到wrapper函数体执行的
bbs()