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)

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