假设我有一个库函数,我无法更改它产生类A的对象,我创建了一个继承自A的类B.
使用库函数生成B类对象最直接的方法是什么?
编辑 - 我在评论中被问到更多细节,所以这里有:
PyTables是一个在python中处理分层数据集的包.我最常用的是它能够管理部分在磁盘上的数据.它提供了一个'Array'类型,它只带有扩展切片,但我需要选择任意行.Numpy提供此功能 - 您可以通过提供与您选择的阵列长度相同的布尔数组来进行选择.因此,我想将Array子类化以添加此新功能.
从更抽象的意义上说,这是我之前考虑过的一个问题.通常的解决方案就像已经建议的那样 - 有一个B的构造函数,它接受一个A和另外的参数,然后拉出A的相关位以插入到B.因为它看起来像一个相当基本的问题,我问了一个问题看看是否有任何我不知道的标准解决方案.
如果子类的初始化程序可以处理它,或者您编写显式升级程序,则可以执行此操作.这是一个例子:
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
由于库函数返回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"
如果你知道你在做什么,你实际上可以改变.__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'