mirror of
https://github.com/kennethreitz/pydantic.git
synced 2026-06-05 23:00:18 +00:00
Samuel Colvin
16263bafea
* tweaks to None behaviour * prevent sub_fields for Optional fields by default * rewrite None validation * rename whole > each_item on validators * cleanup processing of the Json type * fix schema coverage and cleanup * tweak validate_model * change and docs * fix validators on optional fields * coverage * remove is_none_validator * minor performance improvements to ErrorWrapper * fix coverage * fix PaymentCardNumber * undo schema changes, fix * tweak validators
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from typing import List
|
|
from pydantic import BaseModel, ValidationError, validator
|
|
|
|
class DemoModel(BaseModel):
|
|
square_numbers: List[int] = []
|
|
cube_numbers: List[int] = []
|
|
|
|
@validator('*', pre=True) # '*' is same as 'cube_numbers', 'square_numbers' here
|
|
def split_str(cls, v):
|
|
if isinstance(v, str):
|
|
return v.split('|')
|
|
return v
|
|
|
|
@validator('cube_numbers', 'square_numbers')
|
|
def check_sum(cls, v):
|
|
if sum(v) > 42:
|
|
raise ValueError(f'sum of numbers greater than 42')
|
|
return v
|
|
|
|
@validator('square_numbers', each_item=True)
|
|
def check_squares(cls, v):
|
|
assert v ** 0.5 % 1 == 0, f'{v} is not a square number'
|
|
return v
|
|
|
|
@validator('cube_numbers', each_item=True)
|
|
def check_cubes(cls, v):
|
|
# 64 ** (1 / 3) == 3.9999999999999996! this is not a good way of checking cubes
|
|
assert v ** (1 / 3) % 1 == 0, f'{v} is not a cubed number'
|
|
return v
|
|
|
|
print(DemoModel(square_numbers=[1, 4, 9]))
|
|
# > DemoModel square_numbers=[1, 4, 9] cube_numbers=[]
|
|
print(DemoModel(square_numbers='1|4|16'))
|
|
# > DemoModel square_numbers=[1, 4, 16] cube_numbers=[]
|
|
print(DemoModel(square_numbers=[16], cube_numbers=[8, 27]))
|
|
# > DemoModel square_numbers=[16] cube_numbers=[8, 27]
|
|
|
|
try:
|
|
DemoModel(square_numbers=[1, 4, 2])
|
|
except ValidationError as e:
|
|
print(e)
|
|
"""
|
|
1 validation error for DemoModel
|
|
square_numbers -> 2
|
|
2 is not a square number (type=assertion_error)
|
|
"""
|
|
|
|
try:
|
|
DemoModel(cube_numbers=[27, 27])
|
|
except ValidationError as e:
|
|
print(e)
|
|
"""
|
|
1 validation error for DemoModel
|
|
cube_numbers
|
|
sum of numbers greater than 42 (type=value_error)
|
|
"""
|