Python 自定义库的类

发布于10月23日

我使用的是一个具有一个我需要定制的函数的Python库.

这不是问题，因为我可以制作自己的编辑版本，如下所示:

from coolLibrary import originalclass

class myclass(originalclass):
    whatever...

我现在面临的问题是，图书馆在多个地方使用这originalclass.有没有办法告诉图书馆在任何地方都使用myclass而不是originalclass？

我的目标是在不编辑原始代码的情况下拥有一个定制代码，以便可以更新库.

推荐答案

如果coolLibrary.originalclass对其自身有明确的引用:

# coolLibrary.py
class originalclass:
    def foo(self):
        print('original foo')

    def bar(self):
        originalclass.foo(self)

以使:

# main.py
import coolLibrary

class myclass(coolLibrary.originalclass):
    def foo(self):
        print('custom foo')

myclass().bar()

输出:

original foo

在定义自定义类之后，您可以使用自定义类修补coolLibrary.originalclass:

coolLibrary.originalclass = myclass

因此:

myclass().bar()

然后输出:

custom foo

请注意，修补原始类会使行为发生全局更改.如果也使用coolLibrary.originalclass的另一个模块依赖于其原始行为才能工作，则您可能希望将补丁仅应用于需要此类自定义行为的代码块.这可以通过unittest.mock.patch上下文管理器来完成:

from unittest.mock import patch

with patch('coolLibrary.originalclass', myclass):
    myclass().bar()