从方法创建的类实例中重写嵌套类的方法

问题描述

我正在使用Python Docker SDK将命令发送到本地Docker安装。

client = docker.from_env()
client.images.pull(repository=MY_IMAGE,tag="master")

我想做的是覆盖.pull方法，以便我可以添加一个选项，使其在某些情况下不会拉动。

最简单的方法是在每个.pull

之前添加一个if语句

enable_pull=True

if enable_pull:
    client.images.pull(...)
else:
    pass

...但是我想按原样保留代码，并以某种方式覆盖该方法。我知道如何针对“常规”配置进行操作-例如，覆盖from_env()很容易：

class MyClass(docker):
    def from_env(self,*args,**kwargs):
        # my code
        return super().from_env()

但是我不知道如何覆盖.pull，因为它是.images的一种方法。

解决方法

您这里有个train-wreck ：（放大以表示观点）

docker.from_env().images.pull(repository=MY_IMAGE,tag="master")

因此应将此类内容隐藏在您自己的方法之一中

def pull(self,*args,**kwargs):
    docker.from_env().images.pull(*args,**kwargs)

并这样称呼：

self.pull(repository=MY_IMAGE,tag="master")

但是您想要启用/禁用，所以：

def pull(self,**kwargs):
    if self.enable_pull:
        docker.from_env().images.pull(*args,**kwargs)