Python装饰器原理与基本用法分析

Python  /  管理员 发布于 7年前   216

本文实例讲述了Python装饰器原理与基本用法。分享给大家供大家参考，具体如下：

装饰器：

意义：在不能改变原函数的源代码，和在不改变整个项目中原函数的调用方式的情况下，给函数添加新的功能

由于不允许改变函数的源代码，在忽略调用方式的情况下，我们可能会有以下结果：

def decorator(func):  func()  print("logging")def test1():  print("test1")def test2():  print("Test2")decorator(test1)decorator(test2)

但这改变了原本的调用方式，原本是test1()，现在是decorator(test1)

那么如果我们为了使调用方式不变，是否可以使装饰好的函数decorator的返回值是一个我们需要的函数，再赋值给原来的函数名呢？

于是：

def timmer1(func):  def warpper():    start_time = time.time()    func()    stop_time=time.time()    print("the func run time is %s"%(stop_time-start_time))   return warppertest3=timmer1(test3)

好像上面这段代码并没有改变原来的调用方式，调用原来的test3,相当于运行timmer1中的warpper

如果对于无参数的函数来说，上面的代码已经实现了我们的目的，但对于带参数的函数，上面的代码没有传入参数，所以仍然需要修改

于是：

def timmer2(func):  def warpper(*args,**kwargs):    start_time = time.time()    func(*args,**kwargs)    stop_time=time.time()    print("the func run time is %s"%(stop_time-start_time))  return warpper

在上上面的代码中，由于实质上，test3已经等于wrapper，所以可以直接使用,test3(参数)来传入参数，为了处理参数不确定数量问题，可以使用可变长度参数

上面代码还存在一个问题，无法获取原本函数中的返回值，那么我们还需要加上一些东西：

import timedef timmer2(func):  def warpper(*args,**kwargs):    start_time = time.time()    res=func(*args,**kwargs)    return res    stop_time=time.time()    print("the func run time is %s"%(stop_time-start_time))  return warpper

使用一个变量记录下原函数的返回值。

这样我们就实现了装饰器的基本功能。

补充：

python提供了一个功能：

@装饰器名def 目标装饰函数名():  pass#上面的效果是 目标装饰函数名=装饰器(目标装饰函数名)

所以在需要替换原函数的时候，可以在目标装饰函数定义的上一行加上@装饰器名

所以上面的代码会变成：

def timmer2(func):  def warpper(*args,**kwargs):    start_time = time.time()    func(*args,**kwargs)    stop_time=time.time()    print("the func run time is %s"%(stop_time-start_time))  return warpper@timmer2def test7():  print("test7")@timmer2def test6(x):  print(x)test7()test6(2)

import timedef timmer2(func):  def warpper(*args,**kwargs):    start_time = time.time()    res=func(*args,**kwargs)    return res    stop_time=time.time()    print("the func run time is %s"%(stop_time-start_time))  return warpper@timmer2def test4():  print("test4 run")  return "test4 done"test4()print("--------")print(test4())

第二个补充：

可以一个函数，可以使用多个装饰器

比如：

@装饰器1

@装饰器2

更多关于Python相关内容感兴趣的读者可查看本站专题：《Python面向对象程序设计入门与进阶教程》、《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python编码操作技巧总结》及《Python入门与进阶经典教程》

希望本文所述对大家Python程序设计有所帮助。