Python装饰器——提高

上一篇文章我们对装饰器有了初步了解，接下来开始是对装饰器的进一步实践

1.少了元信息怎么办

当你写了一个装饰器作用在某个函数上,但是这个函数的重要的元信息比如名字、文档字符串、注解和参数签名都丢失了。这到底是怎样的一种现象呢？直接上代码吧。

# -*- coding:UTF-8 -*- 
def show_me(func):
    def wrapper():
        print "It is in wrapper."
        func()
    return wrapper

@show_me
def func1():
    '''
    this is func1
    '''
    print "running func1"

if __name__ == '__main__':
    func1()
    print func1.__name__
    print func1.__doc__

打印结果：

元数据

我们可以看到func1.__name__输出为wrapper，func1.__doc__输出为None，哇，全乱套了，怎么办。不要怕，Python大法自有办法。

温馨提醒：任何时候你定义装饰器的时候,都应该使用 functools 库中的 @wraps 装饰器来注解底层包装函数。

也就是说，装饰器函数应该这样做补充：

from functools import wraps ##导入这个包

def show_me(func):
    @wraps(func)
    def wrapper():
        print "It is in wrapper."
        func()
    return wrapper

加上这个后，你可以再运行一下，应该会得到完整的元数据.

2.被装饰函数有参数

我们可以看到，上文的被装饰函数（也就是func函数）是没有参数的，但是在实际应用中，我们大多数的函数都会有参数，这就要用到Python中的可变参数了。那么对于被装饰函数带有参数的装饰器该怎么写呢，可见如下代码：

# -*- coding:UTF-8 -*- 
from functools import wraps

def show_me(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print "强哥好帅!"
        return func(*args, **kwargs)
    return wrapper

@show_me
def func4(a, b=2):
    print "a = %s"%a
    print "b = %s"%b


if __name__ == '__main__':
    func4(3, 8)

运行结果如下：

>>强哥好帅!
>>a = 3
>>b = 8

可变参数是个好东西，在Python中(星号)*和参数有许多有趣的用法，值得一看。

3.装饰器带有参数

有时候，@语法糖后面还可以带参数。比如某个函数功能只有拥有经理权限的人才能访问。可以编写这样的一个装饰器@has_permission("manager"),下面来举一个简单粗鲁的栗子：

# -*- coding:UTF-8 -*- 
from functools import wraps

def has_permission(position=""):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if position == "manager":
                print "我是经理我怕谁!"
                return func(*args, **kwargs)
            else:
                print "You don not have permission to access!"
        return wrapper
    return decorator

@has_permission(position="manager")
def func5(name=""):
    print "My name is : %s"%name

if __name__ == '__main__':
    func5("zhujq")

结果为：

>>我是经理我怕谁!
>>My name is : zhujq

当position改为其他时（比如“Boss”），程序便只能输出：

 You don not have permission to access!

4.其他

Python还提供了类装饰器与@staticmethod，@classmethod，@property和这三个在面向对象编程中常用的装饰器，我们下次再讲吧～～