Python：使 lib 使用重新定义的类

Question

想象一下使用 python 库。假设这个库有一个脚本和几个感兴趣的类。所有这些类都在单独的文件中定义。假设 D 类被 A、B 和 C 类以多种方式导入和使用。以同样的方式，A 类使用 B 类，B 类使用 C 类。脚本直接使用 A 类，因此直接和间接使用 D 类。现在假设您要自定义 D 并在本地定义一个类 D'派生自 D 类。

让脚本和其他类（A、B、C）使用 D' 类而不是 D 类的最优雅的方法是什么？

谢谢你，祝你有美好的一天！

Answer 1

你可以monkey patch the library。

# a.py
import d
class A():
    my_d = d.D()

# script.py
A.d.D = D_prime

def foo():
    a_cls = a.A() # uses A.d.D is D'

a.py 导入 D 的方式很重要

# a.py
from d import D
class A():
    my_d = D()

# script.py
A.D = D_prime

def foo():
    a_cls = a.A() # uses A.D is D'

其他导入方案可能涉及类似的模式。像这样的东西可能很难修补。

def builder():
    from d import D
    return D()

看看mock library 是如何做到这一点的也可能会有所帮助。如果它是一个玩具，您可以直接使用它。 patch 特别是interesting。

@patch('A.d.D', new=D_prime)
def my_func(...):
    a = A() # a.d.D is D_prime

猴子补丁是一种代码气味，取决于应用程序代码中的单元测试是一种气味。没有一个是“优雅的”。如果你是库作者，请支持依赖注入。

如果你够勇敢，你可能会将一些补丁装饰器提取到不专注于模拟的东西中。因为这个模式存在于 unittest 库中，所以它可以被认为是 Python 的和“优雅的”，上面的警告。

如果您对如何感兴趣，您正在修改模块a 的符号表。详情请参阅globals 和locals。