python课堂整理22----装饰器进阶

喜欢ヅ旅行 2021-10-30 05:24 302阅读 0赞

## 一、函数闭包模拟session ##

user_dic = {'username': 'None', 'login': False}
    def auth_func(func):
        def wrapper(*args, **kwargs):
            if user_dic['username'] and user_dic['login']:
                res = func(*args, **kwargs)
                return res
            else:
                username = input("请输入用户名:")
                passwd = input("请输入密码:")
                if username == 'jinling' and passwd == '123':
                    user_dic['username'] = username
                    user_dic['login'] = True
                    res = func(*args, **kwargs)
                    return res
                else:
                    print('用户名或密码错误')
        return wrapper
    
    @auth_func
    def index():
        print('欢迎来到京东主页')
    @auth_func
    def home(name):
        print('欢迎回家%s' %name)
    index()
    home('jinling')

![1697453-20190730162150570-1964953414.png][]

## 多账号认证： ##

user_list = [
        {'name': 'liu', 'passwd': '521'},
        {'name': 'jinling', 'passwd': '123'},
        {'name': 'lala', 'passwd': '456'},
        {'name': 'sb', 'passwd': '000'},
        {'name': 'wocao', 'passwd': '222'}
        ]
    user_dic = {'username': 'None', 'login': False}
    def auth_func(func):
        def wrapper(*args, **kwargs):
            if user_dic['username'] and user_dic['login']:
                res = func(*args, **kwargs)
                return res
            else:
                username = input('请输入用户名:')
                passwd = input('请输入密码:')
                for user in user_list:
                    if username == user['name'] and passwd == user['passwd']:
                        user_dic['username'] = username
                        user_dic['login'] = True
                        res = func(*args, **kwargs)
                        return res
                else:
                    print('用户名或密码错误')
        return wrapper
    
    @auth_func
    def index():
        print('欢迎来到京东主页')
    @auth_func
    def home():
        print('欢迎回家')
    index()
    home()

![1697453-20190730164112406-1967108960.png][]

## 添加认证方式：带参装饰器 ##

user_list = [
        {'name': 'liu', 'passwd': '521'},
        {'name': 'jinling', 'passwd': '123'},
        {'name': 'lala', 'passwd': '456'},
        {'name': 'sb', 'passwd': '000'},
        {'name': 'wocao', 'passwd': '222'}
        ]
    user_dic = {'username': 'None', 'login': False}
    def auth(auth_type = 'filedb'):
        def auth_func(func):
            def wrapper(*args, **kwargs):
                if auth_type == 'filedb':
                    if user_dic['username'] and user_dic['login']:
                        res = func(*args, **kwargs)
                        return res
                    else:
                        username = input('请输入用户名:')
                        passwd = input('请输入密码:')
                        for user in user_list:
                            if username == user['name'] and passwd == user['passwd']:
                                user_dic['username'] = username
                                user_dic['login'] = True
                                res = func(*args, **kwargs)
                                return res
                        else:
                            print('用户名或密码错误')
                elif auth_type == 'dsb':
                    print('难搞哦')
            return wrapper
    
        return auth_func
    @auth(auth_type='filedb')
    def index():
        print('欢迎来到京东主页')
    @auth(auth_type='dsb')
    def home():
        print('欢迎回家')
    index()
    home()

![1697453-20190730170114653-1141851103.png][]

转载于:https://www.cnblogs.com/dabai123/p/11271252.html

[1697453-20190730162150570-1964953414.png]: /images/20211029/006d71751bc34d2f9fcc70549215d3e4.png
[1697453-20190730164112406-1967108960.png]: /images/20211029/6a1fd22c2dbb43f4a829d4ac00184a5f.png
[1697453-20190730170114653-1141851103.png]: /images/20211029/9bed8b0ecfd443c889f4c2084f1490be.png