Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Check if Django model field choices exists

I'm attempting to check if a value exists in the choices tuple set for a model field.

For example lets say I have a Model like this:

class Vote(models.Model):
 VOTE_TYPE = (
  (1, "Up"),
  (-1, "Down"),
 )

 value = models.SmallIntegerField(max_length=1, choices=VOTE_TYPES)

Now lets say in a view I have a variable new_value = 'Up' that I would like to use as the value field in a new Vote. How can I first check to see if the value of that variable exists in the VOTE_TYPE tuple? Thank you.

like image 292
Justin Lucas Avatar asked Mar 26 '10 01:03

Justin Lucas


2 Answers

  1. Are you sure you want to set value to "Up"? that is a display value, you might be interested in setting value to -1 or 1

  2. You can easily check if a choice is in list by converting choice list to dict e.g.

    my_choice in dict(VOTE_TYPE)

  3. If possible you can use choiceField or TypedChoiceField , they will validate that the given value exists in the list of choices.

like image 189
Anurag Uniyal Avatar answered Sep 29 '22 12:09

Anurag Uniyal


First off, models.SmallIntegerField cannot have a max_length option. That are used only in CharField and it's derivatives. Secondly, according to the documentation, the choices option is mainly used by Django for the Admin site. That said, one option you have is to import the VOTE_TYPE tuple from you models.py file into your views.py file and then checking the value - something like:

models.py
...
VOTE_TYPE = ( (1, "Up"), (-1, "Down"), )

views.py
...
from app.models import VOTE_TYPE
for k,v in VOTE_TYPE:
    if v == new_value:
        return True

Also, if I remember right, you can create a ModelForm from your model and then pass in values to said form which should validate the data against the VOTE_TYPE tuple.

like image 27
Rishabh Manocha Avatar answered Sep 29 '22 13:09

Rishabh Manocha