61、Python之函数高级：为函数添加方法，实现属性可变的装饰器

本文主要是介绍61、Python之函数高级：为函数添加方法，实现属性可变的装饰器，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

引言

今天文章的标题，初读起来可能有些拗口，什么叫“为函数添加方法”？但是，如果真正对“Python函数也是对象”这个理念有清晰的理解的话，其实，也是不难理解的，本质上就是给一个对象新增一个自定义方法。通过这样做，我们就可以实现在运行过程中，对装饰器的属性进行动态修改了。

本文的主要内容有：

1、函数对象添加自定义方法

2、属性可变的动态装饰器

函数对象添加自定义方法

其实，这一点比较简单，只需要把函数当作Python中的一个普通对象即可。但是，如果一直执着于函数本身，则理解起来，会有些拧巴。

我们来演示下给函数添加自定义方法的做法，直接看代码：

def add(a, b):base = 0if hasattr(globals()['add'], 'base'):base = globals()['add'].baseprint('执行加法运算')return base + a + bdef change_base(x):globals()['add'].base = xprint(f'变更后：add.base={x}')print(add(10, 20))
print('=' * 20)
add.change_base = change_base
add.change_base(100)
print('=' * 20)
print(add(10, 20))

执行结果：

首先要说明的，这是一个很拧巴的把函数当作普通对象进行使用的演示代码。

可以看到，我们给函数对象添加了一个change_base()的方法，用于修改add函数对象的base属性（当然，如果不存在base属性，就会动态添加一个base属性）。

属性可变的动态装饰器

既然在Python中一个函数对象可以当作普通对象使用，动态添加属性和方法都是可以的。那么，我们基于这样的特性，就可以进一步扩充作为装饰器的闭包函数对象，实现动态修改闭包属性的效果。

假设有这样一个业务需求，还是基于免费用户、普通vip、超级vip的用户的区分。免费用户转换为vip/svip门槛有些高，我们为了提高营收，增加了这样一项功能：支持免费用户按次付费，体验vip、svip的功能体验，从而增加付费用户的转化率。

这个业务需求，就需要我们在免费用户单次付费后，动态调整延迟时长了，也就是我们本文的主角：属性可变的动态装饰器。

我们稍微简化一下，之前的代码，简单定义一个延迟的装饰器，然后模拟付费体验一次的场景实现：

import timedef wait(func):paid = Falsedef wrap(*args, **kwargs):nonlocal paidif paid:print('使用专属付费渠道')paid = Falseelse:print('免费用户排队中...')time.sleep(5)return func(*args, **kwargs)def pay_once():nonlocal paidpaid = Truewrap.pay_once = pay_oncereturn wrap@wait
def business():print("开展正常业务")if __name__ == '__main__':print('付费体验前:')business()print('=' * 20)print('付费体验一次vip:')business.pay_once()business()print('=' * 20)print('付费体验一次vip用完后：')business()

执行结果：