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
|
from __future__ import annotations
from typing import Any, ClassVar, Literal, Optional
from datamodel_code_generator.model.pydantic_v2.base_model import BaseModel
class RootModel(BaseModel):
TEMPLATE_FILE_PATH: ClassVar[str] = 'pydantic_v2/RootModel.jinja2'
BASE_CLASS: ClassVar[str] = 'pydantic.RootModel'
def __init__(
self,
**kwargs: Any,
) -> None:
# Remove custom_base_class for Pydantic V2 models; behaviour is different from Pydantic V1 as it will not
# be treated as a root model. custom_base_class cannot both implement BaseModel and RootModel!
if 'custom_base_class' in kwargs:
kwargs.pop('custom_base_class')
super().__init__(**kwargs)
def _get_config_extra(self) -> Optional[Literal["'allow'", "'forbid'"]]:
# PydanticV2 RootModels cannot have extra fields
return None
|
