Django snippets: Choices class

Choices class

class Choices(object):
    def __init__(self, *args):
        self._choices = []
        for i, arg in enumerate(args):
            setattr(self, arg[0], i)
            self._choices.append((i, arg[1]))

    def __iter__(self):
        return iter(self._choices)

More like this

Automatic Manager Choice Filters by nipuL 3 years ago
Model Choices Helper by pmclanahan 2 years ago
Handling choices the right way by mallipeddi 4 years, 3 months ago
ModelMixin by eallik 1 year, 3 months ago
Online boolean switch in the admin list by sasha 4 years, 5 months ago

Yet another class to simplify field choices creation. Keeps order, allows i18n.

Before:

ONLINE = 0
OFFLINE = 1
STATES = (
    (ONLINE, _('online')),
    (OFFLINE, _('offline'))
)

state = models.IntegerField(choices=STATES, default=OFFLINE)

After:

STATES = Choices(
    ('ONLINE', _('online')),
    ('OFFLINE', _('offline'))
)

state = models.IntegerField(choices=STATES, default=STATES.OFFLINE)

Author:: dc
Posted:: November 10, 2008
Language:: Python
Django Version:: 1.0
Tags:: choices models
Score:: 7 (after 7 ratings)

Tools

Comments

petar (on November 11, 2008):

This class if freaking awesome!