python学习之装饰器

 

目前在通过慕课网学习python的装饰器知识，感觉不是很容易理解。从网上搜罗了很多的资料，看起来感觉也是一直半解。

1）在python中，可以使用函数作为入参或者返回值。这个有点类似于C、C++中函数指针或者函数引用

2）以下面的无参数装饰器为例，语句@performance def factorial(n): 相当于factorial = performance(factorial)。这一点是装饰器中比较重要的一点。从字面上理解来看，就相当于把函数factorial装饰了一下，以后该怎么着还是怎么着。就好比一个人要出门参加一场宴会，你总得要“装饰”（打扮）一下自己吧，穿上合适的衣服，抹点香水之类的东东，然后就出门了，到了宴会的地方，还是一样，该吃就吃，该喝就喝。

借助于慕课网上两个实例（无参数装饰器和带参数装饰器），现在只能先记下来，慢慢深入理解。（声明：以下问题以及代码均来自于慕课网课程python进阶的“编写无参数decorator”和“编写带参数decorator”）

问题1，请编写一个@performance，它可以打印出函数调用的时间，——无参数装饰器

import time
def performance(f):
    def fn(*args, **kw):
        t1 = time.time()
        r = f(*args, **kw)
        t2 = time.time()
        print 'call %s() in %fs' % (f.__name__, (t2 - t1))
        return r
    return fn

@performance
def factorial(n):
    return reduce(lambda x,y: x*y, range(1, n+1))
print factorial(10)

 

问题2，上一个只能打印秒，请给@performance增加一个参数，允许传入's'或者'ms'，——带参数装饰器

import time
def performance(unit):
    def perf_decorator(f):
        def wrapper(*args, **kw):
            t1 = time.time()
            r = f(*args, **kw)
            t2 = time.time()
            t = (t2 - t1) * 1000 if unit=='ms' else (t2 - t1)
            print 'call %s() in %f %s' % (f.__name__, t, unit)
            return r
        return wrapper
    return perf_decorator

@performance('ms')
def factorial(n):
    return reduce(lambda x,y: x*y, range(1, n+1))
print factorial(10)