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
|
import re
from typing import Any, Dict, Union
import srsly
from pydantic import BaseModel
from pydantic.fields import Field as ModelField
from ._config import (
ARGS_FIELD,
ARGS_FIELD_ALIAS,
JSON_EXCEPTIONS,
RESERVED_FIELDS,
SECTION_PREFIX,
VARIABLE_RE,
Config,
try_dump_json,
try_load_json,
)
from ._errors import ConfigValidationError
from ._registry import registry
from .util import SimpleFrozenDict, SimpleFrozenList # noqa: F401
def alias_generator(name: str) -> str:
"""Generate field aliases in promise schema."""
# Underscore fields are not allowed in model, so use alias
if name == ARGS_FIELD_ALIAS:
return ARGS_FIELD
# Auto-alias fields that shadow base model attributes
if name in RESERVED_FIELDS:
return RESERVED_FIELDS[name]
return name
def copy_model_field(field: ModelField, type_: Any) -> ModelField:
"""Copy a model field and assign a new type, e.g. to accept an Any type
even though the original value is typed differently.
"""
return ModelField(
name=field.name,
type_=type_,
class_validators=field.class_validators,
model_config=field.model_config,
default=field.default,
default_factory=field.default_factory,
required=field.required,
alias=field.alias,
)
class EmptySchema(BaseModel):
model_config = {"extra": "allow", "arbitrary_types_allowed": True}
__all__ = [
"Config",
"registry",
"ConfigValidationError",
"SimpleFrozenDict",
"SimpleFrozenList",
]
|
