Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix for #354 - allow None for nullable field with choices

Browse Source
This commit is contained in:
collerek
2021-09-26 12:43:09 +02:00
parent 8376fb023a
commit 67487ebf9c
2 changed files with 35 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ormar/models/helpers/validation.py
View File

@ -52,8 +52,8 @@ def convert_choices_if_needed( # noqa: CCR001
:param field: ormar field to check with choices
:type field: BaseField
:param values: current values of the model to verify
:type values: Dict
:param value: current values of the model to verify
:type value: Dict
:return: value, choices list
:rtype: Tuple[Any, List]
"""
@ -97,6 +97,8 @@ def validate_choices(field: "BaseField", value: Any) -> None:
:type value: Any
"""
value, choices = convert_choices_if_needed(field=field, value=value)
if field.nullable:
choices.append(None)
if value is not ormar.Undefined and value not in choices:
raise ValueError(
f"{field.name}: '{value}' " f"not in allowed choices set:" f" {choices}"