pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Being compatible with non-django style chioces

Browse Source
This commit is contained in:
Karim Allah 2011-09-18 19:48:33 +02:00
parent 89c44cd14e
commit adb7bbeea0
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mongoengine/base.py
View File

@ -139,9 +139,14 @@ class BaseField(object):
def _validate(self, value): def _validate(self, value):
# check choices # check choices
if self.choices is not None: if self.choices is not None:
if type(choices[0]) is tuple:
option_keys = [option_key for option_key, option_value in self.choices] option_keys = [option_key for option_key, option_value in self.choices]
if value not in option_keys: if value not in option_keys:
raise ValidationError("Value must be one of %s." % unicode(option_keys)) raise ValidationError("Value must be one of %s." % unicode(option_keys))
else:
if value not in self.choices:
raise ValidationError("Value must be one of %s." % unicode(self.choices))
# check validation argument # check validation argument
if self.validation is not None: if self.validation is not None: