Python之装饰器

一、装饰器

1.1 简单的装饰器

import time
def func():
    print('嘻嘻更健康')
def timmer(f):
    def inner():
        start_time = time.time()
        time.sleep(0.1)
        f()
        end_time = time.time()
        print('----> 执行效率%s' % (end_time - start_time))
    return inner
func = timmer(func)  # inner
func() # inner()

1.2语法糖

import time
def timmer(f):
    def inner():
        start_time = time.time()
        time.sleep(0.1)
        f()
        end_time = time.time()
        print('----> 执行效率%s' % (end_time - start_time))
    return inner
@timmer  # func = timmer(func)
def func():
    print('嘻嘻更健康')
func() # inner()

1.3函数带参数的装饰器

import time
def timmer(f):
    def inner(*args,**kwargs):
        start_time = time.time()
        time.sleep(0.1)
        f(*args,**kwargs)
        end_time = time.time()
        print('----> 执行效率%s' % (end_time - start_time))
    return inner
@timmer   # func1 = timmer(func1)
def func1(a,b):
    print('%s和%s嘻嘻更健康' %(a,b))
func1('萌哥','杀毒软件')

1.4 带返回值的装饰器

import time
def timmer(f):
    def inner(*args,**kwargs):
        start_time = time.time()
        time.sleep(0.1)
        ret = f(*args,**kwargs)  # 222
        end_time = time.time()
        print('----> 执行效率%s' % (end_time - start_time))
        return ret
    return inner

@timmer  # func = timmer(func)
def func(a):
    return 222
print(func('萌哥'))

1.5 通用装饰器

def wrapper(func):
    def inner(*args,**kwargs):
        '''执行函数前操作'''
        ret = func(*args,**kwargs)
        '''执行函数后的操作'''
        return ret
    return inner

@wrapper
def func():
    print(66)
func()

1.6 查看函数信息

from functools import wraps

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__)

二、开放封闭原则

1.对扩展是开放的
必须允许代码扩展、添加新功能。
2.对修改是封闭的
防止误操作

三、带参数的装饰器

flag =False
def outer(flag):
    def timer(func):
        def inner(*args,**kwargs):
            if flag:
                print('''执行函数之前要做的''')
            re = func(*args,**kwargs)
            if flag:
                print('''执行函数之后要做的''')
            return re
        return inner
    return timer

@outer(flag)
def func():
    print(111)
func()

四、多个装饰器

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()