Python扩展方法一二事

作者: 魏守峰 | 来源:发表于2017-09-28 12:00 被阅读22次

    前言

    跟着一个有强迫症的老板干活是一件极其幸福的事情(你懂的)。最近碰到一个问题,简单的说就是对一个对象做出部分修改后仍然返回此对象,于是我就写了一个方法,老板看了之后只有一句话:不雅观,改成直接对此对象调用此方法。我脑海里千万个不情愿,然而没有办法,不得不低头,精通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方法名称来解决此问题,但是具体不知道如何操作,于是在StackOverflow中提了个问题,很快就有老外大牛回复了。见https://stackoverflow.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

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

    对整个类扩展

    除了直接写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扩展方法实现方式、类的方法的重定义、实例方法扩展等细节,供需查看。

    相关文章

      网友评论

        本文标题:Python扩展方法一二事

        本文链接:https://www.haomeiwen.com/subject/xlmmextx.html