Login

Choices class

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

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)
1
2
3
4
5
6
7
8
9
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

Comments

petar (on November 11, 2008):

This class if freaking awesome!

#

Please login first before commenting.