I am a choice object for Django. I make using choices in forms and models easier.
Deprecated: This is not needed since Django 3.0 which includes enumeration types which work similarly.
pip install django-choice-object
Choice classes inherit from Choice. Each attribute can have one or two values, the first being the value to be stored in the database and the second optional one being the display text. If no display text is given then it is created from the attribute name.
from django_choice_object import Choice
from django.db import models
class SomeChoice(Choice):
FIRST_CHOICE = 1
SECOND_CHOICE = 2
THIRD_CHOICE = 3
LAST_CHOICE = 4, "I am the display text"
class SomeModel(models.Model):
some_field = models.IntegerField(choices=SomeChoice)
>>> list(SomeChoice)
[(1, 'First Choice'), (2, 'Second Choice'), (3, 'Third Choice'), (4, 'I am the display text')]
>>> SomeModel.objects.filter(some_field=SomeChoice.FIRST_CHOICE).all()
[object list...]
>>> SomeModel.objects.get(id=1).some_field
1
>>> SomeModel().some_field = SomeChoice.LAST_CHOICE
Grouping choice values is a fairly common need. You can add groups of values to a choice like so:
class ChoiceWitHGroups(Choice):
FIRST = 1
SECOND = 2
THIRD = 3
FOURTH = 4
START_GROUP = {FIRST, SECOND}
END_GROUP = {THIRD, FOURTH}
Any value that ends with _GROUP
and is a set
will be ignored as a choice.
Choices can be ordered in three ways. The default is by the value, but this can be configured using the
_order_by
attribute:
class TestChoiceOrdered(Choice):
LAST = 1, "zzzzzzzzz"
FIRST = 2, "aaaaaaaa"
_order_by = "name"
In the code snippet above the FIRST
attribute would appear first as they are sorted by the
name rather than the value.
You can also specify a custom sorting key:
class WeirdChoice(Choice):
LAST = 'last', 'Last', 10
FIRST = 'first', 'First', 1
@staticmethod
def _get_sort_key(value):
return value[2]
In this case the third item in the tuple would be the item to sort on.
- Set up a virtualenv:
virtualenv .
- Install the requirements:
pip install -r requirements.txt
- Run the tests:
nosetests .