erdantic.pydantic

PydanticField

Bases: Field[FieldInfo]

Concrete field adapter class for Pydantic fields.

Attributes:

Name	Type	Description
field	ModelField	The Pydantic field object that is associated with this adapter instance.

Source code in erdantic/pydantic.py

class PydanticField(Field[pydantic.fields.FieldInfo]):
    """Concrete field adapter class for Pydantic fields.

    Attributes:
        field (pydantic.fields.ModelField): The Pydantic field object that is associated with this
            adapter instance.
    """

    def __init__(self, name: str, field_info: pydantic.fields.FieldInfo):
        self._name = name
        if not isinstance(field_info, pydantic.fields.FieldInfo):
            raise InvalidFieldError(
                f"field_info must be of type pydantic.fields.FieldInfo. Got: {type(field_info)}"
            )
        super().__init__(field=field_info)

    @property
    def name(self) -> str:
        return self._name

    @property
    def type_obj(self) -> Union[type, GenericAlias]:
        return self.field.annotation

    def is_many(self) -> bool:
        return is_many(self.type_obj)

    def is_nullable(self) -> bool:
        return is_nullable(self.type_obj)

PydanticModel

Bases: Model[Type[BaseModel]]

Concrete model adapter class for a Pydantic BaseModel.

Attributes:

Name	Type	Description
model	Type[BaseModel]	The Pydantic model class that is associated with this adapter instance.
forward_ref_help	Optional[str]	Instructions for how to resolve an unevaluated forward reference in a field's type declaration.

Source code in erdantic/pydantic.py

@register_model_adapter("pydantic")
class PydanticModel(Model[Type[pydantic.BaseModel]]):
    """Concrete model adapter class for a Pydantic
    [`BaseModel`](https://pydantic-docs.helpmanual.io/usage/models/).

    Attributes:
        model (Type[pydantic.BaseModel]): The Pydantic model class that is associated with this
            adapter instance.
        forward_ref_help (Optional[str]): Instructions for how to resolve an unevaluated forward
            reference in a field's type declaration.
    """

    forward_ref_help = (
        "Call 'update_forward_refs' after model is created to resolve. "
        "See: https://pydantic-docs.helpmanual.io/usage/postponed_annotations/"
    )

    def __init__(self, model: Type[pydantic.BaseModel]):
        if not self.is_model_type(model):
            raise InvalidModelError(
                "Argument model must be a subclass of pydantic.BaseModel. "
                f"Got {repr_type_with_mro(model)}"
            )
        super().__init__(model=model)

    @staticmethod
    def is_model_type(obj: Any) -> bool:
        return isinstance(obj, type) and issubclass(obj, pydantic.BaseModel)

    @property
    def fields(self) -> List[Field]:
        return [
            PydanticField(name=name, field_info=field_info)
            for name, field_info in self.model.model_fields.items()
        ]

    @property
    def docstring(self) -> str:
        out = super().docstring
        field_descriptions = [field.field.description for field in self.fields]
        if any(descr is not None for descr in field_descriptions):
            # Sometimes Pydantic models have field documentation as descriptions as metadata on the
            # field instead of in the docstring. If detected, construct docstring and add.
            out += "\nAttributes:\n"
            field_defaults = [field.field.default for field in self.fields]
            for field, descr, default in zip(self.fields, field_descriptions, field_defaults):
                if descr is not None:
                    line = f"{field.name} ({field.type_name}): {descr}"
                    if (
                        not isinstance(default, pydantic_core.PydanticUndefinedType)
                        and default is not ...
                    ):
                        if not line.strip().endswith("."):
                            line = line.rstrip() + ". "
                        else:
                            line = line.rstrip() + " "
                        if isinstance(default, str):
                            line += f"Default is '{default}'."
                        else:
                            line += f"Default is {default}."
                    out += "    " + line.strip() + "\n"

        return out