Python 装饰器（一）

╰+攻爆jí腚メ 2022-05-10 08:56 153阅读 0赞

# Python 装饰器（一） #

首先，看一下下面三个函数，假如说我现在要加一个条件，计算每个函数所花费时间。

def add(x, y):  
        return x+y  
      
    def sub(x,y):  
        return x-y  
      
    def multi(x,y):  
        return x+y

如果不考虑效率，我们可以每个函数逐一添加如下：

import time  
    def add(x, y):  
        t1 = time.time()  
        s =  x+y  
        time.sleep(3)  
        t2 = time.time()  
        times = str(t2 - t1)  
        print "spend : " + times + "s"  
        return s  
       
    def sub(x,y):  
        return x-y  
     
    def multi(x,y):  
        return x*y  
        
    if __name__ == "__main__":  
        sum = add(3, 4)  
        print sum

运行结果如下：

spend : 3.0s
    7

如果有很多这样的函数，我们不可能逐一的添加，那么下面我们该如何处理呢？  
这时候就该使用python 装饰器了！  
代码如下：

import time  
      
    def calc(fn):  
        def sum(x, y):  
            t1 = time.time()  
            s =  fn(x, y)  
            time.sleep(3)  
            t2 = time.time()  
            times = str(t2 - t1)  
            print "spend : " + times + "s"  
      return s  
        return sum  
      
    @calc  
    def add(x,y):  
        return x + y  
    @calc  
    def sub(x,y):  
        return x-y  
    @calc  
    def multi(x,y):  
        return x*y  
      
    if __name__ == "__main__":  
        print add.__name__  
        print sub.__name__  
        print multi.__name__  
        a = add(3, 4)  
        b = sub(3,4)  
        c = multi(3,4)  
        print a,b,c

运行结果如下：

sum
    sum
    sum
    spend : 3.0s
    spend : 3.0s
    spend : 3.0s
    7 -1 7

通过上述代码引用，我们可以直观的看出来，

### **装饰器可以在不修改原有代码的基础上，对已有同类型的一些函数 进行额外的补充！！** ###