python 类方法的链式调用

最新推荐文章于 2025-08-26 17:18:54 发布

原创最新推荐文章于 2025-08-26 17:18:54 发布 · 953 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #类

Python 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了Python中实现类方法链式调用的原理，通过示例展示了如何创建一个类来支持链式调用。文章以selenium库的ActionChains类为例，解释了move_to_element、context_click等方法如何最终由perform方法执行。作者还编写了一个简化版的类Chain，演示了如何在类中返回self以实现方法的连续调用。文章总结称，这种链式调用方式适合用于封装面向过程编程任务。

Python 类方法的链式调用

之前在学selenium的时候里面有一个类ActionChains有一种比较奇特的类方法的调用形式：

from selenium.webdriver import ActionChains
from selenium.webdriver import Chrome
from time import sleep
driver=Chrome(r’/Users/jcgg/Desktop/tools/chromedriver’)
driver.get(r’https://www.baidu.com/’)
element=driver.find_element_by_id(“kw”)
ActionChains(driver).move_to_element(element).context_click().perform()
sleep(10)
driver.close()

这边move_to_element，context_click，perform都是类ActionChains的方法，在查看源码后发现最终执行的方法是perform,前面的两个方法只不过是依次往实例对象ActionChains(driver)列表中添加所需要执行的方法而已，然后再由perform方法统一调用，顺序执行。

这边我写了一个简化后能实现类方法的链式调用的类

class Chain():

    def __init__(self):
        self.to_do_list=[] #初始化添加一个存放需要执行的方法列表

    def func1(self,x):
        self.to_do_list.append(lambda: print(x)) # 添加需要执行的方法,这边可以使用匿名函数来添加，也可以自己定义函数添加，把lambda后面的print()函数替换即可。
        return self

    def func2(self,x):
        self.to_do_list.append(lambda: print(x*2)) # 同上
        return self

    def finish(self):
        self.to_do_list.append(lambda: print("链路式执行")) # 同上
        return self

    def do(self):
        for func in self.to_do_list:
            func()  #这边调用存放在列表里面的方法，不用加参数*args,**args，匿名函数已经带了传入的参数了

if __name__=="__main__":
    Chain().func1(1).func2(2).finish().do()

执行结果：依次执行了func1,func2,finish三个方法
在这里插入图片描述
总结：能够实现链式调用的主要是在类中需要执行的方法返回值都是self,在上述执行的main函数中Chain().func1(1).func2(2).finish()执行返回的是Chain()实例化的对象，最后由这个实例化对象调用了一下do这个方法而已。个人觉得这种用来封装面向过程编程是个不错的选择。