装饰器模式

最新推荐文章于 2024-09-15 22:31:42 发布

新知

最新推荐文章于 2024-09-15 22:31:42 发布

阅读量74

点赞数

分类专栏： python 文章标签： python

原文链接：https://www.cnblogs.com/houzheng/p/10061772.html

版权

python 专栏收录该内容

45 篇文章 2 订阅

订阅专栏

装饰器模式

一无参装饰器

# 装饰器
import time

# 装饰器,记录函数运行时间
def decorator01(fun):
    def wapper():
        stime = time.time()
        fun()
        etime = time.time()
        print("fun run time is {TIME}".format(TIME=etime - stime))
    return wapper  # 必须要返回一个函数的内存地址


# 使用装饰器装饰某个函数,等价于 test01=decorator01(test01),
# 即将test01实际引用变成wapper函数内存地址,所以执行test01实际是执行wapper
@decorator01
def test01():
    time.sleep(2)
    print("test01 is running")

test01()  # 不修改代码和调用方式,实现添加记录时间功能

二函数带参装饰器

# 装饰器
import time

# 装饰器,记录函数运行时间
def decorator01(fun):
    def wapper(*args, **kwargs): # 使用非固定参数,无论参数是什么,都可以传递进来
        stime = time.time()
        fun(*args, **kwargs)
        etime = time.time()
        print("fun run time is {TIME}".format(TIME=etime - stime))
    return wapper  # 必须要返回一个函数的内存地址

# test01() = wapper(), 所以装饰器加参数是给嵌套函数加参数
@decorator01
def test01(args1):
    time.sleep(2)
    print("参数是 {NAME} ".format(NAME=args1))

test01("侯征")  # 不修改代码和调用方式,实现添加记录时间功能

三装饰器本身带参数的装饰器

# 装饰器
import time

# 如果装饰器有参数,最外层是装饰器的参数
def decorator01(*args, **kwargs):
    print("装饰器参数:", *args, **kwargs)
    def out(fun): #第二层才是接受的函数
        def wapper(*args, **kwargs):  # 使用非固定参数,无论参数是什么,都可以传递进来
            stime = time.time()
            fun(*args, **kwargs)
            etime = time.time()
            print("fun run time is {TIME}".format(TIME=etime - stime))

        return wapper  # 必须要返回一个函数的内存地址
    return out  # 要返回装饰函数的内存地址

# 装饰器本身带参数,此时 decorator01(arg)=out,即相当于 @out装饰test01,所以 test01=out(fun)=wapper
@decorator01(1)
def test01(args1):
    time.sleep(2)
    print("参数是 {NAME} ".format(NAME=args1))

test01("侯征")  # 不修改代码和调用方式,实现添加记录时间功能