Python扩展方法一二事

时间:2023-03-08 22:06:07

前言

跟着一个有强迫症的老板干活是一件极其幸福的事情(你懂的)。最近碰到一个问题,简单的说就是对一个对象做出部分修改后仍然返回此对象,于是我就写了一个方法,老板看了之后只有一句话:不雅观,改成直接对此对象调用此方法。我脑海里千万个不情愿,然而没有办法,不得不低头,精通C#、Java、Scala等多种语言HelloWorld的我,一想便知这是扩展方法。于是开始Google之,看似简单的问题,其实里面也有一些细节需要注意,在此记录之。

Level 1

原理很简单,将方法的第一个对象改成self(self即为需要扩展方法的对象的实例),其余不变,无需返回此对象,在此方法外部将此方法赋给此类。当需要调用的时候引入此包即可。大致如下:

class A():
def m2():
... ... def m1(self):
// TODO: Do sth to this instance.
... A.m1 = m1 a = A()
a.m1()

当然m1方法也可以添加其他参数,只需要放在self后面即可。

Level 2

过了几天,老板又提了一个新的需求:某个类有一个m1方法,但是老板想修改此方法,使其更加完善。看上去没有什么难的吗,同样是个扩展方法,于是我写代码如下:

class A():
def m1():
... def m2():
...
... def m1(self):
// TODO: Do sth to this instance.
self.m1()
... A.m1 = m1 a = A()
a.m1()

满心欢喜本以为这么简单的就解决了问题。调试,悲剧的是直接报错,错误为递归调用。细细想来确实如此,在此作用域内直接对self对象调用m1方法,应当是已经扩展后的方法,那么当然会产生循环调用的问题。

我想到应当可以通过先修改类中m1方法名称来解决此问题,但是具体不知道如何操作,于是在*中提了个问题,很快就有老外大牛回复了。见https://*.com/questions/46460250/how-to-call-a-method-that-has-been-overridden-by-an-extension-method#。所以问题的核心在于我们可以通过下述代码定义一个类的方法:

_m1 = A.m1

同样调用的时候将实例作为第一个参数传入。于是代码进行如下改造即可:

class A():
def m1():
... def m2():
...
... _m1 = A.m1 def m1(self):
// TODO: Do sth to this instance.
_m1(self)
... A.m1 = m1 a = A()
a.m1()

这样即解决了递归调用的问题,但是此处有一个细节需要注意,_m1必须定义在m1方法上部,由于_m1是定义在m1扩展方法之上的,所以此处仍是A类中的m1方法,若是定义在m1扩展方法中,则又会出现递归调用的问题。

Level 3

下午看了一下上面在*中提的问题,又有一个新的答案,分析了一下他主要介绍了两点知识,覆盖类的方法和覆盖实例的方法,即扩展方法对整个类生效和只对某个实例生效。

对整个类扩展

除了直接写A.plot = plot外,还可以写成:

setattr(A, 'plot', plot)

A代表需要扩展的类,'plot'为扩展后的方法名,plot为重写的扩展方法。当然如果扩展后的方法名在原类中已有,则覆盖之;若无则为新的方法。

对具体实例扩展

让我比较意外的是python可以对某个实例进行方法扩展,这在其他语言中似乎是基本没有的。

当然我们不能用a.plot = plot的形式为a这个A的实例扩展方法,但是可以通过下述方式对a扩展方法:

a.plot = types.MethodType(plot, a)

这样只有a有此扩展方法,或者是只有a的此方法被修改了,其他实例并不受影响。

总结

本文简单记录了Python扩展方法实现方式、类的方法的重定义、实例方法扩展等细节,供需查看。