Skip to content

Instantly share code, notes, and snippets.

@socratescli

socratescli/pydantic_partial_model.py

Created December 9, 2024 15:07
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save socratescli/db6d2da8197fe1a15b6094eb6e3305dd to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save socratescli/db6d2da8197fe1a15b6094eb6e3305dd to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. Li created this gist Dec 9, 2024.
    28 changes: 28 additions & 0 deletions pydantic_partial_model.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,28 @@
    import pydantic

    def partial_model(model: pydantic.BaseModel,
    partial_model_name: str,
    include: list[str] | None = None,
    exclude: list[str] | None= None,
    doc: str | None= None,
    module: str | None = None
    ):
    fields = []
    for name, info in model.model_fields.items():
    if exclude and name in exclude:
    continue
    if include and name not in include:
    continue
    fields.append((name, info))
    new_model = pydantic.create_model(
    partial_model_name,
    __doc__=doc or model.__doc__,
    __base__=model.__mro__[1:],
    __module__=module or model.__module__,
    **{name: (info.annotation, info) for (name, info) in fields}
    )
    for field in model.model_computed_fields:
    if field in model.__dict__:
    setattr(new_model, field, getattr(model, field))
    new_model.__pydantic_validator__ = model.__pydantic_validator__
    return new_model