Question

假设有一个课程NiceClass在我无法编辑的地方使用其他课程LesserClass

# NiceClass.py

class LesserClass:
    ...
    # stuff
    ...

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

现在，我希望在MyLesserClass的继承版本中使用我自己的班级LesserClass代替NiceClass

# 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中所有未被覆盖的方法都会使用旧LesserClass中的NiceClass.py。

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

就像我只想继承源代码而不是整个命名空间。也许遗产是错误的方式？

Answer 1

我希望这仅限于使用NiceClass类LesserClass下的方法。

现在，如果您希望MyNiceClass中的方法使用MyLesserClass而不是LesserClass，那么您可以更新这些方法的__globals__字典，并将名称{{1 }指向'LesserClass'。

这是一个简单的例子，通过覆盖__getattribute__来证明这一点：

MyLesserClass

<强>输出：

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

覆盖父类

1 个答案: