python - 覆盖父类中使用过的类

假设有一个类在一个我不能编辑的地方使用另一个类

# NiceClass.py

class LesserClass:
    ...
    # stuff
    ...

class NiceClass:
    ...
    # Lots of use of lesser class in here...
    ...

现在，我想在继承版本的NiceClass中使用我自己的类LesserClass而不是到处使用MyLesserClass。

# MyNiceClass.py

from NiceClass import NiceClass
from NiceClass import LesserClass as oldLesserClass

class LesserClass(oldLesserClass):
    ...
    # Some extra stuff
    ...

class MyNiceClass(NiceClass):
    ...
    # Everywhere LesserClass was used I now want to use MyLesserClass
    ...

但是LesserClass中的所有非重写方法都将使用旧NiceClass中的MyNiceClass。
如果我只是将粘贴在LesserClass的整个定义中的内容复制到NiceClass.py中，一切都会按我的要求工作。
就像我只想继承源代码而不是整个命名空间也许继承是错误的？

最佳答案

我希望这仅限于使用类NiceClass的LesserClass下的方法。
现在，如果您希望MyNiceClass中的方法使用MyLesserClass而不是LesserClass，那么您可以更新这些方法的__globals__指令，并使名称'LesserClass'指向MyLesserClass。
下面是一个简单的例子，通过重写__getattribute__来演示这一点：

class A:
    a = 'A.a'
    b = 'A.b'


class B:
    def func_a(self):
        print(A.a)

    def func_b(self):
        print(A.b)


class C:
    a = 'C.a'
    b = 'C.b'


class D(B):
    def func_a(self):
        print(C.a)

    def __getattribute__(self, attr):
        value = object.__getattribute__(self, attr)
        if callable(value):
            value = update_namespace(value, {'old': {'name': 'A', 'obj': A}, 'new': {'obj': C}})
        return value


def update_namespace(func, namespace):
    def wrapper(*args, **kwargs):
        # Update the globals
        func.__globals__[namespace['old']['name']] = namespace['new']['obj']
        val = func(*args, **kwargs)
        # Restore it back to the actual value
        func.__globals__[namespace['old']['name']] = namespace['old']['obj']
        return val
    return wrapper


d = D()
d.func_a()  # This should print C.a
d.func_b()  # This should print C.b

输出：

C.a
C.b