python装饰器的学习笔记

此博文是我对装饰器的一些理解，如果有错误欢迎及时留言，我会第一时间向大家学习。

一、什么是装饰器

1、从字面意义来看：

是用来给函数装饰打扮的函数

 

2、理论上可以理解为：

（1）、不改变函数的运行方式

（2）、不改变函数体内容的情况下给函数添加新的功能

 

3、装饰器的结构：

装饰器=高阶函数+函数嵌套+闭包

 

二、装饰器的种类

下面就利用两个例子来实现一下带参装饰器和不带参装饰器使用。

1、不带参数的装饰器

user ={'name':'test','passwd':'123'}

user_status = {'name':None,'login':False}

def auth_func(func):
    def warper(*args,**kwargs):
        if user_status['name'] and user_status['login']:
            res = func(*args,**kwargs)
            return res
        name = input("username: ")
        passwd = input("password: ")
        if user['name'] == name and user['passwd'] == passwd:
            user_status['name'] = name
            user_status['login'] = True
            res = func(*args,**kwargs)
            return res
        else:
            print("用户名或密码错误")
    return warper

@auth_func
def index(name):
    print("欢迎%s 来到主页" % name)

@auth_func
def home(name):
    print("我在home下")

index("test")
home('test')

#输出
username: test
password: 123
欢迎test 来到主页
我在home下

 

2、带参数的装饰器

auth_user_dict ={'username':None,'password':False}

def load_fileBD():
    with open("userDB",'r') as f:
        return f.read()


def auth(auth_type = 'file'):
    def auth_func(func):
        def wraper(*args,**kwargs):
            if auth_type == 'file':
                fild_data = load_fileBD()
                fild_data = eval(fild_data)
                if auth_user_dict['username'] and auth_user_dict['password']:
                    res = func(*args,**kwargs)
                    return res
                username = input('username: ')
                passwd = input('password: ')
                for user_info in fild_data:
                    if username == user_info['username'] and passwd == user_info['password']:
                        auth_user_dict['username'] = username
                        auth_user_dict['password'] = passwd
                        res = func(*args,**kwargs)
                        return res
                else:
                    print("用户名密码错误")
            elif auth_type == 'mysql':
                print("从mysql中读取用户并验证，以print代表执行验证")
                res = func(*args,**kwargs)
                return res
            else:
                print("没有这种类型的数据库")
        return wraper
    return auth_func

@auth('mysql')
def home(name):
    print("%s is login in home" % name)

home('test')

#输出
从mysql中读取用户并验证，以print代表执行验证
test is login in home