Python 之 装饰器的写法

普通装饰器

1. 不带参数的普通装饰器

from functools import wraps

def use_logging(func):
    @wraps(func)    # 使得装饰器函数和原函数有一样的原信息（docstring、__name__等），但为方便，下面的例子没有使用。
    def wrapper(*args, **kw):
        print args
        print kw
        return func(*args, **kw)
    return wrapper
 
@use_logging
def foo(x, n=1):
    print x, n
    print 'I am foo'
     
foo(3)
foo(1, n=2)

输出如下：

# foo(3)
(3,)
{}    # 因为foo(x, n=1)是在user_logging之后调用的，foo(3)时并未传其他参数，所以这里输出为空。
3 1
I am foo

# foo(1, n=2)
(1,)
{'n': 2}
1 2
I am foo

2. 带参数的普通装饰器

def use_logging(level):
    def decorator(func):
        def wrapper(*args, **kw):
            if level == 'warning':
                print '%s is running' % func.__name__
            return func(*args, **kw)
        return wrapper
    return decorator

@use_logging(level='warning')
def foo(name='FOO'):
    print 'I am %s' % name
    
@use_logging(level='info')
def bar(name='BAR'):
    print 'I am %s' % name

foo()
bar()

输出：

# foo()
foo is running
I am FOO

# bar()
I am BAR

装饰器类

类装饰器具有灵活度大、高内聚、封装性等优点。当使用 @ 形式将装饰器附加到函数上时，就会调用 __call__() 方法。

1. 不带参数的装饰器类

class FOO(object):
    def __init__(self, func):
        self._func = func
        
    def __call__(self, *args, **kw):
        print args, kw
        kw['x'] = 34
        return self._func(*args, **kw)
        
@FOO
def bar(a, x=1):
    print a, x
    print 'I am bar'
    
bar(3,x=7)

输出：

(3,) {'x': 7}
3 34
I am bar

2. 带参数的装饰器类

class FOO(object):
    def __init__(self, level='debug'):
        self._level = level
        
    def __call__(self, func):
        def _call(*args, **kw):
            print args, kw
            if self._level == 'error':
                kw['x'] = 34
            return func(*args, **kw)
        return _call
        
@FOO('error')   # FOO()
def foo(a, x=1):
    print a, x
    print 'I am foo'
    
@FOO('info')
def bar(a, x=1):
    print a, x
    print 'I am bar'

foo(2)
bar(3,x=7)

输出：

# foo(2)
(2,) {}
2 34
I am foo

# bar(d, x=7)
(3,) {'x': 7}
3 7
I am bar

另外一种带参数的装饰器类的写法如下：

class FOO(object):
    def __init__(self, level):
        self._level = level
        
    def __call__(self, func):
        self._func = func
        return self._call
    
    def _call(self, *args, **kw):
        print args, kw
        if self._level == 'error':
            kw['x'] = 34
        return self._func(*args, **kw)

@FOO('error')
def foo(a, x=1):
    print a, x
    print 'I am foo'
    
@FOO(level='info')
def bar(a, x=1):
    print a, x
    print 'I am bar'

foo(2)
bar(3,x=7)

输出的结果如上面的一致，注意加粗的地方。