铁乐学Python_day12_装饰器

例：
def login(user, pwd):
    ‘‘‘
    功能：登录调用
    参数：分别有user和pwd，作用分别是用户和密码；
    return: 返回值是登录成功与否（True，False）
    ‘‘‘
    print(user, pwd, "欢迎登录")

print(login.__name__) #查看函数名字
print(login.__doc__)  #查看函数注释

login

    功能：登录调用
    参数：分别有user和pwd，作用分别是用户和密码；
    return: 返回值是登录成功与否（True，False）

from functools import wraps
# 函数工具库的warps模块来实现装饰器函数名字仍是被装饰的原函数名字

def deco(func):
    @wraps(func) #加在最内层函数正上方
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)
    return wrapper

@deco
def index():
    ‘‘‘首页的注释‘‘‘
    print(‘from index‘)

print(index.__doc__)
print(index.__name__)

首页的注释
index

例：下面是一个在调用函数时先进行打印格林威治时间的Decorator：

import time  # 引入时间模块

def wrapper(func):
    ‘‘‘
    此装饰器主要用于测试被装饰函数的工作时间
    ‘‘‘
    def inner(*args, **kwargs):
        start_time = time.time()
        ret = func(*args, **kwargs)
        time.sleep(0.2322) # 模拟函数执行后经过的时间
          end_time = time.time()
        sec = end_time - start_time
        print(‘此程序执行时间为%s‘ % round(sec, 5))
        return ret # 相当于是返回func()
    return inner

@wrapper # 语法糖装饰上后，相当于是将下方的 hello = wrapper(hello)
def hello():
    print(‘hero say hello‘)

hello() 
# 相当于是执行wrapper(hello)(),就相当于是执行wrapper里层的inner函数体。
# 所以会执行额外添加的功能，同时函数也正常调用了。

# 执行效果如下：
hero say hello
此程序执行时间为0.23301


上面例子是嵌套的装饰器，它比较简单。
因为我们在内层的inner函数中有设置动态参数（*args和**kwargs)
被装饰的函数有设置参数的时候，也可以正常传递，如：
@wrapper # 语法糖装饰上后，相当于是将下方的 hello = wrapper(hello)
def hello(name):
    print(‘hero say hello %s‘ % name)

hello(‘tiele‘)
# 相当于是执行wrapper(hello)(),就相当于是执行wrapper里层的inner函数体。
# 所以会执行额外添加的功能，同时函数也正常调用了。

返回：
hero say hello tiele
此程序执行时间为0.23301


但有一点要注意的
经过decorator装饰之后的函数，它们的__name__已经从原来的‘函数名‘变成了‘inner‘：

print(hello.__name__)

hero say hello tiele
此程序执行时间为0.23301
inner

因为直接用查看函数名字与注释的方法去查看经过被装饰器装饰了的函数会得出的是错误的信息（返回的是装饰器的名字与注释),这不符合我们的需求，（虽然直接在inner函数体内打印func.__name__可以得出原函数名，但在全局环境下使用.__name__查看还是会出错）所以还得用上python内置的functools.wraps去支持全局查看函数名时能看到正确的原函数名：

import time  # 引入时间模块
import functools # 引入函数工具模块

def wrapper(func):
    ‘‘‘
    此装饰器主要用于测试被装饰函数的工作时间
    ‘‘‘
    # 加在最内层函数的正上方，因为要的是最内函数的名字
    @functools.wraps(func)
    def inner(*args, **kwargs):
        start_time = time.time()
        ret = func(*args, **kwargs)
        time.sleep(0.2322) # 模拟函数执行后经过的时间
        end_time = time.time()
        sec = end_time - start_time
        print(‘此程序执行时间为%s‘ % round(sec, 5))
        return ret # 相当于是返回func()
    return inner

@wrapper # 语法糖装饰上后，相当于是将下方的 hello = wrapper(hello)
def hello(name):
    print(‘hero say hello %s‘ % name)

hello(‘tiele‘)
# 相当于是执行wrapper(hello)(),就相当于是执行wrapper里层的inner函数体。
# 所以会执行额外添加的功能，同时函数也正常调用了。

print(‘函数名为‘+ hello.__name__)

hero say hello tiele
此程序执行时间为0.23301
函数名为hello

from functools import wraps
# 函数工具库的warps模块来实现装饰器函数名字仍是被装饰的原函数名字
import time
# 引进时间模块

def timer(*args, **kwargs):
    def wrapper(f):
        @wraps(f)
        # 加在最内层函数的正上方，因为要的是最内函数的名字
        def inner(*args, **kwargs):
            ‘‘‘
            此功能用于测试函数起始时间与结束时间
            :param args:
            :param kwargs:
            :return:
            ‘‘‘
            if flag:
                StartTime = time.time()
                ret = f(*args, **kwargs)
                time.sleep(0.3)
                EndTime = time.time()
                sec = EndTime - StartTime
                print(‘此程序执行效率%f‘ % sec)
            else:
                ret = f(*args, **kwargs)
            return ret
        return inner
    return wrapper

flag = True
@timer()
def hello_word():
    ‘‘‘
    向世界say hello
    :return: None
    ‘‘‘
    print(‘hello,word.hello,hero.‘)

hello_word()
print(hello_word.__name__)
print(hello_word.__doc__)

hello,word.hello,hero.
此程序执行效率0.300017
hello_word

    向世界say hello
    :return: None



flag为False，装饰器的计算时间功能便不会执行。

def timer(flag):
    # 带flag参数，可以控制装饰器的开关
    def wrapper(f):
        @wraps(f)
        # 加在最内层函数的正上方，因为要的是最内函数的名字
        def inner(*args, **kwargs):
            ‘‘‘
            此功能用于测试函数起始时间与结束时间
            :param args:
            :param kwargs:
            :return:
            ‘‘‘
            if flag:
                StartTime = time.time()
                ret = f(*args, **kwargs)
                time.sleep(0.3)
                EndTime = time.time()
                sec = EndTime - StartTime
                print(‘此程序执行效率%f‘ % sec)
            else:
                ret = f(*args, **kwargs)
            return ret
        return inner
    return wrapper

@timer(False)
def hello_word():
    ‘‘‘
    向世界say hello
    :return: None
    ‘‘‘
    print(‘hello,word.hello,hero.‘)

hello_word()

hello,word.hello,hero.

def wrapper1(func):
    def inner():
        print(‘wrapper1 ,before func‘)
        func()
        print(‘wrapper1 ,after func‘)
    return inner

def wrapper2(func):
    def inner():
        print(‘wrapper2 ,before func‘)
        func()
        print(‘wrapper2 ,after func‘)
    return inner

@wrapper2
@wrapper1
def f():
    print(‘in f‘)

f()

效果：

wrapper2 ,before func
wrapper1 ,before func
in f
wrapper1 ,after func
wrapper2 ,after func