在父类中覆盖使用的类

Question

假设有一个类NiceClass在某个我无法编辑的地方使用了另一个类LesserClass。

# NiceClass.py

class LesserClass:
    ...
    # stuff
    ...

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

现在我想在继承版本的 NiceClass 中，使用自己的类 MyLesserClass 替代原来的 LesserClass。

# 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
    ...

但是，MyNiceClass 中所有未被覆盖的方法都将使用旧版 NiceClass.py 中的 LesserClass。

如果我只是将整个 NiceClass 的定义复制粘贴到 MyNiceClass.py 中，那么一切都会按照我的意愿进行。

这就像我只想继承源代码而不是整个命名空间。也许继承并不是正确的方法？

- sciferr

---

您的问题仍不清楚，请尽量提供更多的明确信息。你期望得到什么结果？有什么具体问题需要解决？ - Ankur Sharma

Answer 1

我希望这仅限于使用类LesserClass的NiceClass方法。如果您想让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

- Ashwini Chaudhary