简单的装饰器

Posted 2020-10-25 右威卫大将军

import functools

def log(text):
    if isinstance(text, str):
        def decorator(func):
            @functools.wraps(func)
            def wrapper(*args, **kw):
                func(*args, **kw)
                print ‘%s %s‘%(text, func.__name__)
            return wrapper
        return decorator
    else:
        @functools.wraps(text)
        def wrapper(*args, **kw):
            print ‘%s %s‘%(‘call‘, text.__name__)
            return text(*args, **kw)
        return wrapper

@log(‘end‘)
def now1():
    print ‘now‘

@log
def now2():
    print ‘now‘

now1()
now2()

返回值：

now
end now1
call now2
now

 

几个重点：

一、带参数的装饰器需要写三层，而不带参数的需要两层。

二、这个装饰器既可以实现带参数，也可以实现不带参数。当传入了字符串时，函数会执行带参数的装饰器部分；没有任何参数时，会将text作为不带参数的装饰器中的‘func‘。

三、第一部分在原函数调用后打印，第二部分在原函数调用前打印。

四、需要用functool.wraps方法将原函数的函数名复制给装饰器，否则执行now1.__name__和now2.__name__会打印‘wrapper‘而不是‘now1‘和‘now2‘。