Python无参装饰器的实现方案及优化

 更新时间:2021年08月16日 09:26:28   作者:Including-xhj  
装饰器(Decorators)是 Python 的一个重要部分,所谓装饰器就是闭包函数的一种应用场景,这篇文章主要给大家介绍了关于Python无参装饰器的相关资料,需要的朋友可以参考下

一、什么是装饰器

定义一个函数,该函数可为其他函数添加额外的功能。

二、何时用装饰器

需要在不修改被装饰对象源代码及其调用方式时,为被装饰对象添加额外的功能。

三、如何写一个装饰器

现在我们有如下一个函数help(),time.sleep()来模拟函数执行时间,print打印传入参数值,方便我们来进行分析。如果现在我们需要为help函数添加一个统计其运行时间的功能,我们可以怎么做?

import time
 
def help(x, y):
    time.sleep(1)
    print(f'x={x} y={y}')
 
help(1, 2)

方案一:

在help函数开头结束分别调用time.time(),两者相减得运行时间。

import time
 
def help(x, y):
    start = time.time()
    time.sleep(1)
    print(f'x={x} y={y}')
    stop = time.time()
    print(stop - start)
 
help(1, 2)

运行结果:

方案一中我们在help中加了相关代码,虽然没有改变它的调用方式,但改变了它的源代码。我们继续想想如何两者都不改变的情况下,完成我们的目标。

对,函数内不能动,我们可以动函数外呀,在help前后加上相关代码,似乎就可以达到我们的目标了,这就是方案二,我们来试试。

方案二:

import time
 
def help(x, y):
    time.sleep(1)
    print(f'x={x} y={y}')
 
start = time.time()
help(1, 2)
stop = time.time()
print(stop - start)

运行结果:

显而易见,似乎没有问题,但是如果我们需要多次调用help函数的话,在它前后都得加上相同的代码,这样代码就会显得十分冗余了。既然help函数前后代码不会变的话,我们可以将它们封装成另一个函数呀,说干就干。

方案三:

import time
 
def help(x, y):
    time.sleep(1)
    print(f'x={x} y={y}')
 
def wrapper():
    start = time.time()
    help(1, 2)
    stop = time.time()
    print(stop - start)
 
wrapper()

运行一下:

 

这样我们就解决了多次调用的问题,但美中不足的是,help函数的调用方式改变了,而且help的参数固定,也只能修饰help函数,我们来一步步试着优化。

优化一(参数优化,实现任意参数): 

对参数优化,我们可以将help的实参通过wrapper的传入,而为了实现任意参数,我们首先想的便是*args,**kwargs来作为函数的参数,于是将方案三进行改进如下(为方便分析,为help多增加了一个参数):

import time
 
 
def help(x, y, z):
    time.sleep(1)
    print(f'x={x} y={y} z={z}')
 
 
def wrapper(*args, **kwargs):
    start = time.time()
    help(*args, **kwargs)
    stop = time.time()
    print(stop - start)
 
 
wrapper(1, 2, 3)

运行一下:

这样我们便将help的参数变得更加灵活了,接着我们来优化。

优化二(实现装饰其他对象):

需要装饰其他对象,意味着我们在help位置的应该是一个可变参数,也就是用户输入的参数,即wapper函数内应变为:

def wrapper(*args, **kwargs):
    start = time.time()
    func(*args, **kwargs)
    stop = time.time()
    print(stop - start)

但是我们期望wrapper能和内部调用的func函数的参数一致,即wrapper的参数我们应该不去改变,那我们func的值从何处传来呢?

没错,我们可以运用闭包函数来传参,修改一下下:

def outter(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        func(*args, **kwargs)
        stop = time.time()
        print(stop - start)
 
    return wrapper

这样我们为其他函数修饰时,只需要将其函数名作为outter函数的参数传入即可:

import time
 
def help(x, y, z):
    time.sleep(1)
    print(f'这是help的{x}{y}{z}')
 
def others(x, y, z):
    time.sleep(1)
    print(f'这是others的{x}{y}{z}')
 
def outter(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        func(*args, **kwargs)
        stop = time.time()
        print(stop - start)
 
    return wrapper
 
help = outter(help)
others = outter(others)
 
help(1, 2, 3)
others(4, 5, 6)

运行一下:

结果符合预期,而且在使用时由于outter内的func是在局部名称空间,outter外的func是在全局名称空间,调用时二者并不冲突,并且使用时可读性较高,我们好像达成开始的目标,似乎能以假乱真了。但我们继续思考一下,我们演示用到的函数十分简单,甚至没有返回值,如果加上返回值后,我们再对其修饰后,能得到原函数的返回值吗?

优化三(得到相同返回值):

回到我们的wrapper中去,既然需要我们func函数的返回值,我们直接将其赋值给res,再return出res的值:

import time
 
def help(x, y, z):
    time.sleep(1)
    print(f'这是help的{x}{y}{z}')
    return 'help'
 
def others(x, y, z):
    time.sleep(1)
    print(f'这是others的{x}{y}{z}')
    return 'others'
 
def outter(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        res=func(*args, **kwargs)
        stop = time.time()
        print(stop - start)
        return res
    return wrapper
 
help = outter(help)
others = outter(others)
 
res1=help(1, 2, 3)
res2=others(4, 5, 6)
print(res1,res2)

没毛病,跑一下:

 总结: 

到这我们完成了一个简单的无参装饰器,装饰后的func既没有改变源代码,也没有改变调用方式。

但是代码稍显冗余,python语法便规定:在被装饰对象正上方单独一行写@装饰器名字,等价于func=outter(func),简化代码。从中我们总结出无参装饰器的一个模板:

def outter(func):
     def wrapper(*args,**kwargs):
         # 1、调用原函数
         # 2、增加的新功能
         res=func(*args,**kwargs)
         return res
     return wrapper
 
#使用时
@outter
def func:
    pass

到此这篇关于Python无参装饰器的文章就介绍到这了,更多相关Python无参装饰器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python获取接口请求耗时的方法详解

    Python获取接口请求耗时的方法详解

    你想知道我们请求一个url的时候,握手和请求资源分别占用多长时间么?今天我们就来使用python写个小案例来看看,感兴趣的可以跟随小编一起了解一下
    2023-04-04
  • Python中变量的作用域详解

    Python中变量的作用域详解

    大家好,本篇文章主要讲的是Python中变量的作用域详解,感兴趣的同学赶快来看一看吧,对你有帮助的话记得收藏一下
    2022-01-01
  • 浅谈Python中os模块及shutil模块的常规操作

    浅谈Python中os模块及shutil模块的常规操作

    这篇文章主要介绍了浅谈Python中os模块及shutil模块的常规操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-04-04
  • Anaconda下Python中h5py与netCDF4模块下载与安装的教程详解

    Anaconda下Python中h5py与netCDF4模块下载与安装的教程详解

    这篇文章主要为大家详细介绍了基于Anaconda,下载并安装Python中h5py与netCDF4这两个模块的方法,感兴趣的小伙伴可以跟随小编一起学习一下
    2024-01-01
  • 我们为什么要减少Python中循环的使用

    我们为什么要减少Python中循环的使用

    这篇文章主要介绍了我们为什么要减少Python中循环的使用,我将阐述 Python 提供的一些简单但是非常有用的结构,一些小技巧以及一些我在数据科学工作中遇到的案例。我将讨论 Python 中的 for 循环,以及如何尽量避免使用它们,需要的朋友可以参考下
    2019-07-07
  • 浅谈Python中的函数(def)及参数传递操作

    浅谈Python中的函数(def)及参数传递操作

    这篇文章主要介绍了浅谈Python中的函数(def)及参数传递操作,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-05-05
  • Python time三种时间转换小结

    Python time三种时间转换小结

    本文主要介绍了Python time三种时间转换小结,主要包括时间戳,结构化时间,字符串时间,文中根据实例编码详细介绍的十分详尽,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-03-03
  • Python编程中的异常处理教程

    Python编程中的异常处理教程

    这篇文章主要介绍了Python编程中的异常处理教程,是Python入门学习中的基础知识,需要的朋友可以参考下
    2015-08-08
  • 解析python中的jsonpath 提取器

    解析python中的jsonpath 提取器

    jsonPath 是一种信息抽取类库,是从JSON文档中抽取指定信息的工具,提供多种语言实现版本,包括:Javascript, Python, PHP 和 Java。本文给大家介绍python的jsonpath 提取器,感兴趣的朋友跟随小编一起看看吧
    2022-01-01
  • scrapy-redis分布式爬虫的搭建过程(理论篇)

    scrapy-redis分布式爬虫的搭建过程(理论篇)

    这篇文章主要介绍了scrapy-redis分布式爬虫的搭建过程(理论篇),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-09-09

最新评论