1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
from typing import Any, Type
import pydantic
from pydantic import BaseModel
IS_PYDANTIC_V2 = int(pydantic.VERSION.split(".")[0]) >= 2
IS_PYDANTIC_V2_10 = (
IS_PYDANTIC_V2 and int(pydantic.VERSION.split(".")[1]) >= 10
)
if IS_PYDANTIC_V2:
from pydantic import TypeAdapter
else:
from pydantic import parse_obj_as
def parse_object_as(object_type: Type, data: Any):
if IS_PYDANTIC_V2:
return TypeAdapter(object_type).validate_python(data)
else:
return parse_obj_as(object_type, data)
def get_field_type(field):
if IS_PYDANTIC_V2:
return field.annotation
else:
return field.outer_type_
def get_model_fields(model):
if IS_PYDANTIC_V2:
return model.model_fields
else:
return model.__fields__
def parse_model(model_type: Type[BaseModel], data: Any):
if IS_PYDANTIC_V2:
return model_type.model_validate(data)
else:
return model_type.parse_obj(data)
def get_extra_field_info(field, parameter: str):
if IS_PYDANTIC_V2:
if isinstance(field.json_schema_extra, dict):
return field.json_schema_extra.get(parameter)
return None
else:
return field.field_info.extra.get(parameter)
def get_config_value(model, parameter: str):
if IS_PYDANTIC_V2:
return model.model_config.get(parameter)
else:
return getattr(model.Config, parameter, None)
def get_model_dump(model, *args: Any, **kwargs: Any):
if IS_PYDANTIC_V2:
return model.model_dump(*args, **kwargs)
else:
return model.dict(*args, **kwargs)
|
