Login

Better Django Model Field Choices

Author:
jorjun
Posted:
March 2, 2011
Language:
Python
Version:
1.2
Tags:
choice choices model field
Score:
2 (after 2 ratings)

Nice to name your constant multiple choice fields in models, this is one way of doing that. Sorry I haven't looked into existing alternatives. But this approach worked for me.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from collections import namedtuple

from django.db.models import *


def _get_field_choices(named):
    """
    Namedtuple expected, return Django field choices
    """
    dictionary = named._asdict()
    return [(v, k) for (k, v) in dictionary.items()]

def get_named_choices(name, choices):
    choices = namedtuple(name, choices)(*range(len(choices)))
    return _get_field_choices(choices)


class Photo(Model):
    _image_types = {
        'COLOUR' : 0, 
        'BLACKWHITE' : 1, 
        'MONOCHROME' : 2, 
        'UNKNOWN' : 9, 
    }
    IMAGE_TYPE = get_named_choices('ImageType', _image_types)

    image_type = PositiveSmallIntegerField(
        default=0,
        choices=get_field_choices(IMAGE_TYPE)
    )

"""
usage: myphoto.image_type = Photo.IMAGE_TYPE.BLACKWHITE
"""

More like this

Comments

jorjun (on March 2, 2011):
<p>CORRECTION (sorry)</p> <p>def get_named_choices(name, choices): return namedtuple(name, choices)(*range(len(choices)))</p>

#

jorjun (on March 2, 2011):
<p>class Property: def init(self, kwd): for (key, val) in kwd.iteritems(): if isinstance(val, dict): val = Property(val) self.dict[key] = val</p> <pre>@property def p_django_field_choices(self): return [(v, k) for (k, v) in self.__dict__.items()] </pre>

#

jorjun (on March 2, 2011):
<pre>class Property: def __init__(self, **kwd): for (key, val) in kwd.iteritems(): if isinstance(val, dict): val = Property(**val) self.__dict__[key] = val @property def p_django_field_choices(self): return [(v, k) for (k, v) in self.__dict__.items()] </pre>

#

Please login first before commenting.