装饰器

2020-03-20  本文已影响0人  堪怜咏絮才

1.装饰器的概念

装饰器是一个闭包:
  内层函数引用外层函数的变量(参数也算变量),然后返回内层函数,就是闭包。
装饰器的本质还是函数
装饰器的基本用法:
  1.定义一个装饰器函数(也可以用类)
  2.定义业务函数(类)
  3.用装饰器装饰器业务函数

2.打印日志

def log(func):
    def wrapper(*args, **kwargs):
        print(f'{func.__name__}函数开始执行了')
        func(*args, **kwargs)
        print(f"{func.__name__}函数执行结束了")

    return wrapper

@log
def add(x, y):
    return x + y

add(1, 2)
执行结果:
add函数开始执行了
add函数执行结束了

3.记录函数执行时间

import time

def log(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        func(*args, **kwargs)
        end_time = time.time()
        total_time = end_time - start_time
        print(f'函数{func.__name__}总共耗时{total_time}s')

    return wrapper

@log
def add(x, y):
    time.sleep(5)
    return x + y

add(1, 2)
执行结果:
函数add总共耗时5.000309467315674s

4.带参的装饰器

def shout(category):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if category == 'dog':
                print('a dog is shouting')
            elif category == 'cat':
                print('a cat is shouting')
            else:
                return
            func(*args, **kwargs)

        return wrapper

    return decorator


@shout('dog')
def dog():
    print('this is a dog')


@shout('cat')
def cat():
    print('this is a cat')


dog()
cat()
执行结果:
  a dog is shouting
this is a dog
a cat is shouting
this is a cat

5.类装饰器

class Logger(object):

    def __init__(self, level='INFO'):
        self.level = level

    def __call__(self, func):
        def wrapper(*args, **kwargs):
            print(f"[{self.level}]:function {func.__name__}() is running")
            result = func(*args, **kwargs)
            return result

        return wrapper

@Logger('WARNING')
def add(x, y):
    return x + y

print(add(1, 2))
执行结果:
[WARNING]:function add() is running
3
上一篇 下一篇

猜你喜欢

热点阅读