alihan/datamodel-code-generator
1
0
Fork 0
mirror of https://github.com/koxudaxi/datamodel-code-generator.git synced 2024-03-18 14:54:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix original_name validation (#1755)

Browse Source
This commit is contained in:
Koudai Aono
2023-12-04 22:58:21 +09:00
committed by GitHub
parent 6cd17e195f
commit b2b57e728c
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
datamodel_code_generator/reference.py
View File

@@ -38,7 +38,7 @@ from datamodel_code_generator.util import (
PYDANTIC_V2,
ConfigDict,
cached_property,
field_validator,
model_validator,
)
if TYPE_CHECKING:
@@ -121,14 +121,19 @@ class Reference(_BaseModel):
children: List[Any] = []
_exclude_fields: ClassVar[Set[str]] = {'children'}
@field_validator('original_name')
def validate_original_name(cls, v: Any, values: Dict[str, Any]) -> str:
@model_validator(mode='before')
def validate_original_name(cls, values: Any) -> Any:
"""
If original_name is empty then, `original_name` is assigned `name`
"""
if v: # pragma: no cover
return v
return values.get('name', v) # pragma: no cover
if not isinstance(values, dict): # pragma: no cover
return values
original_name = values.get('original_name')
if original_name:
return values
values['original_name'] = values.get('name', original_name)
return values
if PYDANTIC_V2:
# TODO[pydantic]: The following keys were removed: `copy_on_model_validation`.