Question

假设我有一个库函数，我无法更改它产生类A的对象，我创建了一个继承自A的类B.

使用库函数生成B类对象最直接的方法是什么？

编辑 - 我在评论中被问到更多细节，所以这里有：

PyTables是一个在python中处理分层数据集的包。我最常用的是它能够管理部分在磁盘上的数据。它提供了一个'Array'类型，它只带有扩展切片，但我需要选择任意行。 Numpy提供此功能 - 您可以通过提供与您选择的阵列长度相同的布尔数组来进行选择。因此，我想将Array子类化以添加此新功能。

从更抽象的意义上说，这是我之前考虑过的一个问题。通常的解决方案就像已经建议的那样 - 有一个B的构造函数，它接受一个A和另外的参数，然后拉出A的相关位插入到B中。因为它看起来像一个相当基本的问题，我问了一个问题看看是否有任何我不知道的标准解决方案。

Answer 1

如果子类的初始化程序可以处理它，或者您编写显式升级程序，则可以执行此操作。这是一个例子：

class A(object):
    def __init__(self):
        self.x = 1

class B(A):
    def __init__(self):
        super(B, self).__init__()
        self._init_B()
    def _init_B(self):
        self.x += 1

a = A()
b = a
b.__class__ = B
b._init_B()

assert b.x == 2

Answer 2

由于库函数返回A，因此无法在不更改它的情况下返回B.

您可以做的一件事是编写一个函数来获取A实例的字段并将它们复制到一个新的B实例中：

class A: # defined by the library
    def __init__(self, field):
        self.field = field

class B(A): # your fancy new class
    def __init__(self, field, field2):
        self.field = field
        self.field2 = field2 # B has some fancy extra stuff

def b_from_a(a_instance, field2):
    """Given an instance of A, return a new instance of B."""
    return B(a_instance.field, field2)


a = A("spam") # this could be your A instance from the library
b = b_from_a(a, "ham") # make a new B which has the data from a

print b.field, b.field2 # prints "spam ham"

编辑：根据您的情况，合成而非继承可能是一个不错的选择;那就是你的B类只能包含A的实例而不是继承：

class B2: # doesn't have to inherit from A
    def __init__(self, a, field2):
        self._a = a # using composition instead
        self.field2 = field2

    @property
    def field(self): # pass accesses to a
        return self._a.field
    # could provide setter, deleter, etc

a = A("spam")
b = B2(a, "ham")

print b.field, b.field2 # prints "spam ham"

Answer 3

如果您知道自己正在做什么，您实际上可以更改对象的.__class__属性：

In [1]: class A(object): ...: def foo(self): ...: return "foo" ...: In [2]: class B(object): ...: def foo(self): ...: return "bar" ...: In [3]: a = A() In [4]: a.foo() Out[4]: 'foo' In [5]: a.__class__ Out[5]: __main__.A In [6]: a.__class__ = B In [7]: a.foo() Out[7]: 'bar'

Answer 4

Monkeypatch the library？

例如，

import other_library
other_library.function_or_class_to_replace = new_function

噗，它会返回你想要它返回的任何内容。

Monkeypatch A. new 返回B的实例？

调用obj = A（）后，将结果更改为obj。 class = B？

在Python中将对象转换为子类？

4 个答案: