Python-12-装饰器

一、定义

器即函数

装饰即修饰，意指为其他函数添加新功能

装饰器定义：本质就是函数，功能是为其他函数添加新功能

原则：

• 1.不修改被装饰函数的源代码（开放封闭原则）

• 2.为被装饰函数添加新功能后，不修改被修饰函数的调用方

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

1. 高阶函数

默认满足以下两个条件中的一个就是高阶函数：

• 函数的传入参数是一个函数名
• 函数的返回值是一个函数名

#高阶函数应用1:把函数当做参数传给高阶函数
import time
def foo():
    print('from the foo')

def timmer(func):
    start_time=time.time()
    func()
    stop_time=time.time()
    print('函数%s 运行时间是%s' %(func,stop_time-start_time))
timmer(foo)
#总结:我们确实为函数foo增加了foo运行时间的功能,但是foo原来的执行方式是foo(),现在我们需要调用高阶函数timmer(foo),改变了函数的调用方式


#高阶函数应用2:把函数名当做参数传给高阶函数,高阶函数直接返回函数名
import time
def foo():
    print('from the foo')

def timmer(func):
    start_time=time.time()
    return func
    stop_time=time.time()
    print('函数%s 运行时间是%s' %(func,stop_time-start_time))
foo=timmer(foo)
foo()
#总结:我们确实没有改变foo的调用方式,但是我们也没有为foo增加任何新功能

函数返回值是函数名

2. 函数嵌套

在函数内部定义函数

3. 闭包

闭包:在一个作用域里放入定义变量,相当于打了一个包

三、装饰器实现

def timmer(func):
    def wrapper(*args, **kwargs):
        print(func)
        res = func(*args, **kwargs)
        return res

    return wrapper

@timmer  # test = timmer(test)
def test(name, age):
    print('姓名：%s，年龄：%s' % (name, age))
    return 1

msg = test('Tom', 18)  # 就是在运行wrapper
print(msg)

四、应用

 网站登录认证

user_list = [
    {'name': 'alex', 'passwd': '123'},
    {'name': 'Tom', 'passwd': '123'},
    {'name': 'Jerry', 'passwd': '123'}
]

current_user = {'username': None, 'login': False}


def auth_deco(func):
    def wrapper(*args, **kwargs):
        if current_user['username'] and current_user['login']:
            res = func(*args, **kwargs)
            return res
        i = 0
        while i < 3:
            username = input('用户名: ').strip()
            passwd = input('密码: ').strip()
            for user_dic in user_list:
                if username == user_dic['name'] and passwd == user_dic['passwd']:
                    current_user['username'] = username
                    current_user['login'] = True
                    res = func(*args, **kwargs)
                    return res
            else:
                print('用户名或者密码错误,重新登录')
            i += 1
        print('请稍后重试')

    return wrapper


@auth_deco
def index():
    print('欢迎来到主页面')


@auth_deco
def home():
    print('这里是你家')


@auth_deco
def shopping_car():
    print('查看购物车啊亲')


@auth_deco
def order():
    print('查看订单啊亲')


home()

user_list = [
    {'name': 'alex', 'passwd': '123'},
    {'name': 'Tom', 'passwd': '123'},
    {'name': 'Jerry', 'passwd': '123'}
]

current_user = {'username': None, 'login': False}


def auth(auth_type='file'):
    def auth_deco(func):
        def wrapper(*args, **kwargs):
            if auth_type == 'file':
                if current_user['username'] and current_user['login']:
                    res = func(*args, **kwargs)
                    return res
                i = 0
                while i < 3:
                    username = input('用户名: ').strip()
                    passwd = input('密码: ').strip()
                    for user_dic in user_list:
                        if username == user_dic['name'] and passwd == user_dic['passwd']:
                            current_user['username'] = username
                            current_user['login'] = True
                            res = func(*args, **kwargs)
                            return res
                    else:
                        print('用户名或者密码错误,重新登录')
                    i += 1
                print('请稍后登录')
            elif auth_type == 'ldap':
                print('巴拉拉小魔仙')
                res = func(*args, **kwargs)
                return res

        return wrapper

    return auth_deco


# auth(auth_type='file')就是在运行一个函数,然后返回auth_deco,所以@auth(auth_type='file')
# 就相当于@auth_deco,只不过现在,我们的auth_deco作为一个闭包的应用,外层的包auth给它留了一个auth_type='file'参数
@auth(auth_type='ldap')
def index():
    print('欢迎来到主页面')


@auth(auth_type='ldap')
def home():
    print('这里是你家')


@auth(auth_type='file')
def shopping_car():
    print('查看购物车啊亲')


@auth(auth_type='file')
def order():
    print('查看订单啊亲')


order()

五、类的装饰器

装饰器不仅可以对函数进行装饰，还可以装饰类

def deco(obj):
    print('====')
    obj.x = 1
    obj.y = 2
    return obj

@deco
class Foo(object):
    pass

print(Foo.__dict__)  
# {'__module__': '__main__', '__dict__': <attribute '__dict__' of 'Foo' objects>, 
# '__weakref__': <attribute '__weakref__' of 'Foo' objects>, '__doc__': None, 'x': 1, 'y': 2}