Python: Programatically create subclasses based on init values

Question

I have a base class from which I would like to make many subclasses. The subclasses differ only in the arguments used to call the base class during instantiation. The example below shows how to make a subclass Apple. Is there a way to do this programatically, without having to code the subclass' __init__ method? This seems like a job for metaclasses, but in this case I cannot modify the base class.

apple = {'color': 'red', 'shape': 'sphere'}
pear = {'color': 'yellow', 'shape': 'cone'}
melon = {'color': 'green', 'shape': 'prolate'}

class Fruit(object):
    def __init__(self, color, shape):
        self.color = color
        self.shape = shape        

class Apple(Fruit):
    def __init__(self):
        Fruit.__init__(self, **apple)

If they only differ by initial values, what's the point of them being subtypes? Why not just def apple(): return Fruit("red", "sphere")? — nmclean
– nmclean, Commented Aug 23, 2013 at 12:37
What is the motivation for wanting to use subclasses? This appears like an obvious use case for instances of Fruit, not subclasses. — Joel Cornett
– Joel Cornett, Commented Aug 23, 2013 at 12:37
The internal representation should not define the API; so if OP wants subclasses, there might be a reason. — Alfe
– Alfe, Commented Aug 23, 2013 at 12:42
There are reasons, but I admit they're pretty trivial: the code where the subclasses are used is much clearer if the classes bear the names of their object, and I was simply curious if this was possible. — user807435
– user807435, Commented Aug 23, 2013 at 14:53

user9876 · Accepted Answer · 2013-08-23 12:50:34Z

6

See the type() function.

def make_fruit(name, kwargs):
    def my_init(self):
        Fruit.__init__(self, **kwargs)
    return type(name, (Fruit,), {'__init__': my_init})

Apple = make_fruit('Apple', apple)

answered Aug 23, 2013 at 12:50

user9876

11.1k7 gold badges47 silver badges67 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Alfe Over a year ago

Nicer. And I like using def instead of lambda for defining the __init__().

user807435 Over a year ago

I prefer this over @falsetru's response since it doesn't use a lambda. Using the dict argument in type to define __init__ is the key. Thanks.

falsetru · Accepted Answer · 2013-08-23 12:49:18Z

3

Using type:

class Fruit(object):
    def __init__(self, color, shape):
        self.color = color
        self.shape = shape        

apple = {'color': 'red', 'shape': 'sphere'}
pear = {'color': 'yellow', 'shape': 'cone'}
melon = {'color': 'green', 'shape': 'prolate'}

g = globals()
for clsname, attrs in [('Apple', apple), ('Pear', pear), ('Melon', melon)]:
    def temp(attrs):
        g[clsname] = type(clsname, (Fruit,), {
            '__init__': lambda self: Fruit.__init__(self, **attrs)
        })
    temp(attrs)

>>> a = Apple()
>>> p = Pear()
>>> m = Melon()
>>> assert a.color == 'red' and a.shape == 'sphere'
>>> assert p.color == 'yellow' and p.shape == 'cone'
>>> assert m.color == 'green' and m.shape == 'prolate'

answered Aug 23, 2013 at 12:49

falsetru

371k69 gold badges769 silver badges659 bronze badges

1 Comment

Alfe Over a year ago

Good solution (I can abort my writing then …). But I have to disapprove of modifying the result of globals(). This is not working in all cases. (locals() isn't any better.) There are lot of Qs in SO about this. The only really working way of changing the current scope programmatically is via exec and that's an ugly hack.

Community · Accepted Answer · 2017-05-23 11:44:57Z

0

I think the solution you're seeking does not exists: I all you subclasses share the same constructor, then the only thing which makes the class unique is its name. And I don't think you want to have a generic constructor which check the class name to choose what to do.

So I think:

either you re-define the constructor in each subclass, and precise which parameter to pass to the parent constructor,
either you put the specific constant values in some class member, and the constructor use it to call the parent constructor

See interesting thing in this other thread: Class factory in Python

edited May 23, 2017 at 11:44

CommunityBot

11 silver badge

answered Aug 23, 2013 at 12:44

Alexis

7159 silver badges35 bronze badges

Collectives™ on Stack Overflow

Python: Programatically create subclasses based on init values

3 Answers 3

2 Comments

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

2 Comments

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related