Choices datatype for model

Author:: menendez
Posted:: March 26, 2008
Language:: Python
Version:: .96
Score:: 1 (after 1 ratings)

Download
Raw

This class will automatically create a django choices tuple like this:

STATUS_CHOICES = django_choices(Draft=1, Public=2, Closed=3)

Additionally, it includes a method that converts the choices tuple to a dictionary. Like this:

STATUS = STATUS_CHOICES.to_dict()

Those types can come in handy when you need to use those magic values in your code. Best done within the model once so everyone can use it. Code based on: http://www.djangosnippets.org/snippets/455/.

By the way, if you want to just have the method without having to convert to the newer syntax.. it's backward compatible. Just add django_choices in front of the first paren for your choices tuple.

class django_choices(tuple):
    '''
    Creates a choices data type like what django needs for models. 
    '''
    def __new__(self, *args, **kwargs):
        self.choices = []
        if kwargs:
          for val in kwargs.values():
              for key in kwargs.keys():
                if kwargs[key] == val:
                  self.choices.append((val, key))
        return tuple.__new__(self, self.choices or args)
        
    def to_dict(self):
        '''
        Automatically converts the choices to a dictionary. Useful for type lookups.
        '''
        d = {}
        for choice in self:
            d[choice[1].lower()] = choice[0]
        return d

Choices datatype for model

More like this

Comments