Python装饰器2-__call__方法与类装饰器

简介: __call__方法、创建类装饰器、装饰器的应用场景

一、__call__方法

上一篇《Python装饰器-闭包与函数装饰器》中主要介绍了装饰器的基本原理以及如何创建函数装饰器,而本篇的主要内容是如何创建一个类,并使其作为装饰器使用。在学习类装饰器前,需要先了解python中的__call__方法。__call__方法是python魔法方法的一种,它的作用是将类的实例化对象变成可调用对象,类似于像函数一样被调用。默认情况下:__call__方法在函数中是实现的,而在类中是没有被实现的。

1.函数与__call__方法

callable( )方法

通过callable( ) 方法可以判断对象是否为可调用对象:

输出结果为True,因此函数introduce是一个可调用对象。

function(args)的等价写法

function(*args)等价于function.__call__(*args)

2.类与__call__方法

callable( )方法

默认情况下,在类中是没有实现__call__方法的,此时通过callable( )方法,传入实例对象,运行结果为FALSE,表示这个实例对象不是一个可调用对象。直接调用实例对象,会报错提示“TypeError: 'xxxx' object is not callable”

在类中定义__call__方法

classIntroduce(object):
def__init__(self, name, city):
self.name=nameself.city=citydefintroduce(self):
print(f"我叫{self.name}, 我来自{self.city}")
def__call__(self, age):
print(f"我今年{age}岁了")
intro=Introduce("刘德华", '香港')
print(callable(intro))
intro(28)
intro.__call__(28)

在Introduce类中定义了__call__方法后,此时:

  • 再通过callable()方法调用实例对象intro,运行结果就是True了,表示intro是一个可调用对象;
  • intro(28)实际是调用__call__方法,执行__call__方法中的逻辑;
  • intro(28)等价于intro.__call__(28)

二、类装饰器

相比函数类装饰器,类装饰器具有灵活度大、高内聚、封装性等优点。

1.类装饰器

在明白了__call__方法的作用及原理后,类装饰器就变得简单易懂了。例如我想要定义一个类装饰器,用来统计函数的运行耗时:

importtimeclassTimer(object):
def__init__(self, obj):
self.obj=objdef__call__(self, *args, **kwargs):
start_time=time.time()
self.obj(*args, **kwargs)
end_time=time.time()
use_time=end_time-start_timeprint("函数%s运行耗时:%s秒"% (self.obj.__name__, use_time))
@Timerdefinfo():
time.sleep(2)
print("life is short, i need python")
info()

运行结果如下:

2.带参数的类装饰器

同样是统计函数运行耗时的例子,假如此时我想在函数运行时灵活地输出一些自定义的信息,而不是固定写死的信息,那么,此时就可以给类装饰器定义一个初始化变量,用于存储想要灵活输出的信息:

classTimer2(object):
def__init__(self, prefix):
self.prefix=prefixdef__call__(self, func):
defwrapper(*args, **kwargs):
start_time=time.time()
ret=func(*args, **kwargs)
end_time=time.time()
use_time=end_time-start_timeprint("%s, 函数运行共耗时: %s秒"% (self.prefix, use_time))
returnretreturnwrapper

例如,我想要在函数运行时输出当前时间:

@Timer2(prefix=f"当前时间{time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())}")
definfo():
time.sleep(2)
print("life is short, i need python")

此时就相当于把prefix的内容和__call__函数原本定义的内容实现了灵活拼接,运行结果如下:

三、装饰器应用场景

函数装饰器与类装饰器之间是可以相互修饰的,也就是说函数装饰器既可以修饰函数、也可以修饰类,类装饰器既可以修饰类、也可以修饰函数。

1.函数装饰器装饰函数

给被装饰对象增加类似于setUp和tearDown的前后置功能

defwrapper_info(func):
definner(*args, **kwargs):
print("开始介绍...")
res=func(*args, **kwargs)
print("介绍结束...")
returnresreturninner@wrapper_infodefintroduce3(name, age, city):
print(f"我叫{name}, 我今年{age}岁了, 我来自{city}")
introduce3('刘德华', 28, '香港')
'''开始介绍...我叫刘德华, 我今年28岁了, 我来自香港介绍结束...'''

2.函数装饰器装饰类

给被装饰对象增加属性

defwrapper(obj):
obj.phone_number=15252188888returnobj@wrapperclassIntroduce:
def__init__(self, name, city):
self.name=nameself.city=citydefintroduce(self):
print(f"我叫{self.name}, 我来自{self.city}")
intro=Introduce("刘德华", '香港')
intro.introduce()
print(intro.phone_number)  # 给Introduce类增加了一个phone_number的属性'''我叫刘德华, 我来自香港15252188888'''

3.类装饰器装饰函数

统计函数运行时间

classTimer2(object):
def__init__(self, prefix=''):
self.prefix=prefixdef__call__(self, func):
defwrapper(*args, **kwargs):
start_time=time.time()
ret=func(*args, **kwargs)
end_time=time.time()
use_time=end_time-start_timeprint("%s, 函数运行共耗时: %s秒"% (self.prefix, use_time))
returnretreturnwrapper@Timer2(prefix="测试类装饰器")
definfo():
time.sleep(2)
print("life is short, i need python")
info()
'''life is short, i need python测试类装饰器, 函数运行共耗时: 2.0003740787506104秒'''

4.类装饰器装饰类

这里仅仅为了证明和演示类装饰器可以用来装饰类,实际意义并不大。

classDecorator:
def__init__(self, obj):
self.obj=objdef__call__(self, *args, **kwargs):
res=self.obj(*args, **kwargs)
print("测试类装饰器用来装饰类")
returnres# 类装饰器装饰类@DecoratorclassAnimals:
defdogs(self):
print('wong wong')
ani=Animals()
ani.dogs()
'''测试类装饰器用来装饰类wong wong'''

5.函数装饰器装饰类中的方法

同样以统计函数运行时间为例,定义装饰器和引用装饰器都是一样的。

defwrapper(func):
definner(*args, **kwargs):
start_time=time.time()
res=func(*args, **kwargs)
end_time=time.time()
use_time=end_time-start_timeprint("函数运行共耗时: %s秒"% (use_time))
returnresreturninnerclassGoods(object):
def__init__(self, goods, weight, price):
self.goods=goodsself.weight=weightself.price=price@wrapperdefbuying(self):
time.sleep(1)
print(f"商品: {self.goods}, 重量: {self.weight}, 价格: {self.price}")
goods=Goods("西瓜", "10KG", "30元")
goods.buying()
'''商品: 西瓜, 重量: 10KG, 价格: 30元函数运行共耗时: 1.0001726150512695秒'''


相关文章
WK
|
3天前
|
Python
Python中format_map()方法
在Python中,`format_map()`方法用于使用字典格式化字符串。它接受一个字典作为参数,用字典中的键值对替换字符串中的占位符。此方法适用于从字典动态获取值的场景,尤其在处理大量替换值时更为清晰和方便。
WK
61 36
|
8天前
|
存储 索引 Python
|
14天前
|
机器学习/深度学习 数据采集 数据挖掘
11种经典时间序列预测方法:理论、Python实现与应用
本文将总结11种经典的时间序列预测方法,并提供它们在Python中的实现示例。
53 2
11种经典时间序列预测方法:理论、Python实现与应用
|
4天前
|
开发者 Python
探索Python中的装饰器:简化代码,增强功能
【10月更文挑战第22天】在Python的世界里,装饰器是一个强大的工具,它能够让我们以简洁的方式修改函数的行为,增加额外的功能而不需要重写原有代码。本文将带你了解装饰器的基本概念,并通过实例展示如何一步步构建自己的装饰器,从而让你的代码更加高效、易于维护。
|
6天前
|
开发框架 Python
探索Python中的装饰器:简化代码,增强功能
【10月更文挑战第20天】在编程的海洋中,简洁与强大是航行的双桨。Python的装饰器,这一高级特性,恰似海风助力,让代码更优雅、功能更强大。本文将带你领略装饰器的奥秘,从基础概念到实际应用,一步步深入其内涵与意义。
|
2天前
|
测试技术 开发者 Python
深入浅出:Python中的装饰器解析与应用###
【10月更文挑战第22天】 本文将带你走进Python装饰器的世界,揭示其背后的魔法。我们将一起探索装饰器的定义、工作原理、常见用法以及如何自定义装饰器,让你的代码更加简洁高效。无论你是Python新手还是有一定经验的开发者,相信这篇文章都能为你带来新的启发和收获。 ###
6 1
|
2天前
|
设计模式 测试技术 开发者
Python中的装饰器深度解析
【10月更文挑战第24天】在Python的世界中,装饰器是那些能够为函数或类“添彩”的魔法工具。本文将带你深入理解装饰器的概念、工作原理以及如何自定义装饰器,让你的代码更加优雅和高效。
|
2天前
|
测试技术 数据安全/隐私保护 Python
探索Python中的装饰器:简化和增强你的函数
【10月更文挑战第24天】在Python编程的海洋中,装饰器是那把可以令你的代码更简洁、更强大的魔法棒。它们不仅能够扩展函数的功能,还能保持代码的整洁性。本文将带你深入了解装饰器的概念、实现方式以及如何通过它们来提升你的代码质量。让我们一起揭开装饰器的神秘面纱,学习如何用它们来打造更加优雅和高效的代码。
|
2天前
|
测试技术 开发者 Python
探索Python中的装饰器:从入门到实践
【10月更文挑战第24天】 在Python的世界里,装饰器是一个既神秘又强大的工具。它们就像是程序的“隐形斗篷”,能在不改变原有代码结构的情况下,增加新的功能。本篇文章将带你走进装饰器的世界,从基础概念出发,通过实际例子,逐步深入到装饰器的高级应用,让你的代码更加优雅和高效。无论你是初学者还是有一定经验的开发者,这篇文章都将为你打开一扇通往高效编程的大门。
|
4天前
|
程序员 Python
深入浅出Python装饰器
【10月更文挑战第22天】在编程的世界里,我们经常追求代码的优雅和可重用性。Python中的装饰器就是这样一个神奇的工具,它允许我们在不改变原有函数定义的情况下增加额外的功能。本文将通过浅显易懂的方式介绍装饰器的概念、用法及其背后的原理,带你一步步深入理解这个强大的特性。