Skip to main content

Typer extension to enable pydantic support

Project description

Pydantic Typer

PyPI - Version PyPI - Python Version

Typer extension to enable pydantic support


Table of Contents

Installation

pip install pydantic-typer

Note: pydantic-typer comes with pydantic and typer as dependencies, so you don't need to install anything else.

Usage

All the code blocks below can be copied and used directly (they are tested Python files). To run any of the examples, copy the code to a file main.py, and run it:

python main.py

Basic Usage

Simply add the @enable_pydantic decorator to any function that you use with typer.run:

import pydantic
import typer

from pydantic_typer import enable_pydantic


class User(pydantic.BaseModel):
    id: int
    name: str = "Jane Doe"


@enable_pydantic
def main(num: int, user: User):
    print(num, type(num))
    print(user, type(user))


if __name__ == "__main__":
    typer.run(main)

Usage with nested models

@enable_pydantic also works with nested pydantic models:

from __future__ import annotations

from typing import Optional

import pydantic
import typer

from pydantic_typer import enable_pydantic


class Pet(pydantic.BaseModel):
    name: str
    species: str


class Person(pydantic.BaseModel):
    name: str
    age: Optional[float] = None  # noqa: UP007 typer does not support float | None yet, see https://github.com/tiangolo/typer/pull/548
    pet: Pet


@enable_pydantic
def main(person: Person):
    print(person, type(person))


if __name__ == "__main__":
    typer.run(main)

Use pydantic models with typer.Argument

You can annotate the parameters with typer.Argument to make all model fields CLI arguments:

from __future__ import annotations

import pydantic
import typer
from typing_extensions import Annotated

from pydantic_typer import enable_pydantic


class User(pydantic.BaseModel):
    id: int
    name: str


@enable_pydantic
def main(num: Annotated[int, typer.Option()], user: Annotated[User, typer.Argument()]):
    print(num, type(num))
    print(user, type(user))


if __name__ == "__main__":
    typer.run(main)
*Note: You can also override annotations directly on the pydantic model fields:*
from __future__ import annotations

import pydantic
import typer
from typing_extensions import Annotated

from pydantic_typer import enable_pydantic


class User(pydantic.BaseModel):
    id: Annotated[int, typer.Argument(metavar="THE_ID")]
    name: Annotated[str, typer.Option()]


@enable_pydantic
def main(num: Annotated[int, typer.Option()], user: Annotated[User, typer.Argument()]):
    print(num, type(num))
    print(user, type(user))


if __name__ == "__main__":
    typer.run(main)

Here, User is a typer.Argument, but we manually override the fields again:

  • We override the metavar of to User.id be THE_ID
  • And User.name to be a typer.Option

License

pydantic-typer is distributed under the terms of the MIT license.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pydantic_typer-0.0.2.tar.gz (39.1 kB view hashes)

Uploaded Source

Built Distribution

pydantic_typer-0.0.2-py3-none-any.whl (5.6 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page