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 th开发者_运维问答e VOTE_TYPE tuple?
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
You can easily check if a choice is in list by converting choice list to dict e.g.
my_choice in dict(VOTE_TYPE)
If possible you can use choiceField or TypedChoiceField , they will validate that the given value exists in the list of choices.
All new projects probably use TextChoices
or IntegerChoices
.
As it inherits from a builtin enum.Enum
, you can now check it in a way your intuition tells you.
from django.db.models import TextChoices
class ExampleChoices(TextChoices):
example_a = "a"
example_b = "b", "display_b"
>>> "a" in ExampleChoices
True
>>> "b" in ExampleChoices
True
>>> "display_b" in ExampleChoices
False
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.
精彩评论