How to use values from list as pydantic validator?

Question

I would like to create pydantic model to validate users form. one of my model values should be validated from a list of names. I succeed to create the model using enum as follow:

from enum import Enum
class Fruit(str, Enum):
    APPLE = 'apple'
    BANANA = 'banana'
    MELON = 'melon'

from pydantic import BaseModel
class UserForm(BaseModel):
    fruit: Fruit
    name: str

Now I would like to switch the enum to a list of values in my code:

fruit = ['apple','banana','melon']

How can I manage to do so?

tnx

alex_noname · Accepted Answer · 2020-12-27 19:19:43Z

26

You could do this also by means of a list of Literal type. Like so:

import pydantic
from typing import Literal, List

class M(pydantic.BaseModel):
    fruits: List[Literal["apple", "orange"]]

print(M.parse_obj({"fruits":["apple", "orange"]}))  # OK fruits=['apple', 'orange']
print(M.parse_obj({"fruits":["apple", "orange", "potato"]}))  # Error unexpected value potato

answered Dec 27, 2020 at 19:19

alex_noname

33.2k6 gold badges95 silver badges110 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

snow6oy Nov 9 at 22:02

The question is a bit loose, but I don't think the input is going to be a list. However, using Literal instead of Enum is the answer.

aahnik · Accepted Answer · 2021-05-05 11:31:10Z

26

I am proposing an elegant solution.

from pydantic import BaseModel
from typing import List
from enum import Enum


class Fruit(str, Enum):
    APPLE = 'apple'
    BANANA = 'banana'
    MELON = 'melon'


class UserForm(BaseModel):
    fruits: List[Fruit]
    name: str

And that's it.

you don't need to write your own validator
just tell pydantic you need a list of Fruit object, and it will do it for you

Check the above code:

put the above code in a file main.py.

Run

python -i main.py

>>> uf = UserForm(fruits=['apple','banana'],name='hello')
>>> uf
UserForm(fruits=[<Fruit.APPLE: 'apple'>, <Fruit.BANANA: 'banana'>], name='hello')


>>> af = UserForm(fruits=['monkey','apple'],name='hello')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "pydantic/main.py", line 400, in pydantic.main.BaseModel.__init__
pydantic.error_wrappers.ValidationError: 1 validation error for UserForm
fruits -> 0
  value is not a valid enumeration member; permitted: 'apple', 'banana', 'melon' (type=type_error.enum; enum_values=[<Fruit.APPLE: 'apple'>, <Fruit.BANANA: 'banana'>, <Fruit.MELON: 'melon'>])
>>>

pydantic will raise an error, as monkey is not in fruits.

answered May 5, 2021 at 11:31

aahnik

1,9212 gold badges15 silver badges33 bronze badges

5 Comments

Orsiris de Jong Over a year ago

Indeed, elegant ;)

Dan Lee Over a year ago

One problem with Enum s is that if integer is given, it will always raise value error, not a type error.

Yassine Nacif Over a year ago

I believe he meant to replace the Fruit class with a list of strings instead. I don't think he meant to make fruit plural (accepting multiple values in a list).

Eren Over a year ago

TypeError: Type List cannot be instantiated; use list() instead

Benyamin Jafari Over a year ago

can I use list rather than typing.List in pydantic schema model?

Walid · Accepted Answer · 2020-12-27 13:43:12Z

8

You can use validator in the following way:

 from pydantic import BaseModel, ValidationError, validator
 class UserForm(BaseModel):
    fruit: str
    name: str
    @validator('fruit')
    def fruit_must_be_in_fruits(cls,fruit):
      fruits=['apple','banana','melon']
      if fruit not in fruits:
        raise ValueError(f'must be in {fruits}')
      return fruit
 try:
    UserForm(fruit="apple",name="apple")
 except ValidationError as e:
    print(e)

It will raise a validation error if it doesn't match the criteria.

edited Dec 27, 2020 at 13:43

answered Dec 27, 2020 at 12:26

Walid

7285 silver badges13 bronze badges

1 Comment

OrFeldman Over a year ago

Hi, this is great thank you. But it still not working as I would like. I am using this UserForm in fastapi and that means that while swagger/docs will automatically suggest all the values for the user when I am using the enum solution it will not suggest them in this solution.

Patrick Artner · Accepted Answer · 2020-12-27 12:18:35Z

1

You can get informations about the enum by its .__members__ dictionary - here you can simply iterate it keys though:

from enum import Enum
class Fruit(str, Enum):
    APPLE = 'apple'
    BANANA = 'banana'
    MELON = 'melon'

# only need __members__ if you need more infos about it

print(Fruit.__members__)

# you do not need the __members__ if you just want the keys
print([name.lower() for name in Fruit])

Output:

# enums __members__ dictionary
{'APPLE': <Fruit.APPLE: 'apple'>, 
 'BANANA': <Fruit.BANANA: 'banana'>, 
 'MELON': <Fruit.MELON: 'melon'>} 

# lower keys
['apple', 'banana', 'melon']

answered Dec 27, 2020 at 12:18

Patrick Artner

51.9k10 gold badges50 silver badges79 bronze badges

Comments

snow6oy · Accepted Answer · 2025-11-09 21:33:05Z

0

As Yassine said, the question is looking for an alternative to using Enums. For example, Literal

from pydantic import BaseModel
from typing import Literal

class UserForm(BaseModel):
    fruit: Literal['apple','banana','melon']
    name: str

form_data = {'fruit':'apple', 'name': 'someuser'}
user_form = UserForm(**form_data)
print(user_form.fruit) # apple

answered Nov 9 at 21:33

snow6oy

7162 gold badges7 silver badges17 bronze badges

Collectives™ on Stack Overflow

How to use values from list as pydantic validator?

5 Answers 5

1 Comment

5 Comments

1 Comment

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

5 Answers 5

1 Comment

5 Comments

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related