Reputation: 343
I would like to know how to convert parent object that was return by some function to child class.
class A(object):
def __init__():
pass
class B(A):
def functionIneed():
pass
i = module.getObject() # i will get object that is class A
j = B(i) # this will return exception
j.functionIneed()
I cannot change class A. If I could I would implement functionIneed to class A, but it is impossible because of structure of code.
Upvotes: 14
Views: 32321
Reputation: 3825
This comment shares a link, I copy-pasted the contents here in case of link-rot. Please consider visiting the original link if possible.
import copy
class Base(object):
def __init__(self):
self.base_init = True
def foo(self):
return 'base foo'
def bar(self):
return 'base bar'
class Child(Base):
def __new__(cls, other):
if isinstance(other, Base):
other = copy.copy(other)
other.__class__ = Child
return other
return object.__new__(cls)
def __init__(self, other):
self.child_init = True
def bar(self):
return 'child bar'
b = Base()
assert b.base_init == True
assert b.foo() == 'base foo'
assert b.bar() == 'base bar'
assert b.__class__ == Base
c = Child(b)
assert c.base_init == True
assert c.child_init == True
assert c.foo() == 'base foo'
assert c.bar() == 'child bar'
assert c.__class__ == Child
assert b.__class__ == Base
Upvotes: 0
Reputation: 58
Just thinking outside the box:
Instead of a new class with the function you want, how about just adding the function to the class or instance you already have?
There is a good description of this in Adding a Method to an Existing Object Instance
Upvotes: 0
Reputation: 798716
Python does not support "casting". You will need to write B.__init__()
so that it can take an A
and initialize itself appropriately.
Upvotes: 26
Reputation: 181
You did not correctly define your classes. Should be like this:
class A(object):
def __init__(self):
pass
class B(A):
def __init__(self):
super(B,self).__init__()
def functionIneed(self):
pass
Then you can
j=B()
j.fuctionIneed()
as expected
You forgot to refer to the ins
Upvotes: 0
Reputation: 4422
How about:
i = module.getObject() # i will get object that is class A
try:
i.functionIneed()
except AttributeError:
# handle case when u have a bad object
Read up on duck typing.
Upvotes: -1
Reputation: 3877
You said you want to implement something like this:
class B(A):
def functionIneed():
pass
But really what you would be making is something more like this (unless you had intended on making a class or static method in the first place):
class B(A):
def functionIneed(self):
pass
Then you can call B.functionIneed(instance_of_A)
. (This is one of the advantages of having to pass self
explicitly to methods.)
Upvotes: 5
Reputation: 24007
I have a strong suspicion, nay, conviction, that there is something horribly wrong with your program design that it requires you to do this. In Python, unlike Java, very few problems require classes to solve. If there's a function you need, simply define it:
def function_i_need(a):
"""parameter a: an instance of A"""
pass # do something with 'a'
However, if I cannot dissuade you from making your function a method of the class, you can change an instance's class by setting its __class__
attribute:
>>> class A(object):
... def __init__(self):
... pass
...
>>> class B(A):
... def functionIneed(self):
... print 'functionIneed'
...
>>> a = A()
>>> a.functionIneed()
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: 'A' object has no attribute 'functionIneed'
>>> a.__class__ = B
>>> a.functionIneed()
functionIneed
This will work as long as B has no __init__
method, since, obviously, that __init__
will never be called.
Upvotes: 7