return an object of a different class
alex23
wuwei23 at gmail.com
Tue Feb 15 21:45:52 EST 2011
On Feb 16, 12:23 pm, s... at uce.gov wrote:
> How can I do something like this in python:
>
> #!/usr/bin/python3.1
>
> class MyNumbers:
> def __init__(self, n):
> self.original_value = n
> if n <= 100:
> self = SmallNumers(self)
> else:
> self = BigNumbers(self)
>
> class SmallNumbers:
> def __init__(self, n):
> self.size = 'small'
>
> class BigNumbers:
> def __init__(self, n):
> self.size = 'big'
>
> t = MyNumbers(200)
>
> When I do type(t) it says MyNumbers, while I'd want it to be BigNumbers,
> because BigNumbers and SmallNumbers will have different methods etc...
Firstly, does MyNumbers _have_ to be a class? Or would a function
acting as a class factory be sufficient?
Otherwise, you can change the class of an object, even within its own
methods:
class MyNumbers(object):
def __init__(self, n = 0):
self.original_value = n
self.__class__ = BigNumbers if n > 100 else SmallThing
class BigNumbers(MyNumbers):
size = 'big'
class SmallNumbers(MyNumbers):
size = 'small'
>>> from test import *
>>> s = MyNumbers(50)
>>> b = MyNumbers(200)
>>> type(s)
<class 'test.SmallNumbers'>
>>> s.size
'small'
>>> type(b)
<class 'test.BigNumbers'>
>>> b.size
'big'
Hope this helps.
More information about the Python-list
mailing list