pydantic-diagram 0.1.0

Generate ERD-style D2 diagrams from Pydantic models

pydantic-diagram

Generate ERD-style diagrams from your Pydantic models using D2.

Takes a set of Pydantic models and outputs D2 code that renders as a nice visual diagram showing model relationships, inheritance, and field types.

Installation

pip install pydantic-diagram

Usage

As a library

from pydantic import BaseModel
from pydantic_diagram import render_d2

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

class Post(BaseModel):
    id: int
    title: str
    author: User

# Generate D2 code
d2_code = render_d2([User, Post])
print(d2_code)

From the command line

# Diagram specific models from a module
pydantic-diagram myapp.models:User,Post

# Diagram all models in a module
pydantic-diagram myapp.models

# Write to a file
pydantic-diagram myapp.models -o schema.d2

# Then render with D2
d2 schema.d2 schema.svg

CLI options

pydantic-diagram <module:models> [options]

Options:
  -o, --output FILE      Write to file instead of stdout
  --direction DIR        Layout direction: right, down, left, up (default: right)
  --no-inherited         Exclude inherited fields from subclasses
  --qualified-names      Use fully qualified names (module.ClassName)
  --no-docstrings        Exclude docstrings and field descriptions

What it diagrams

• Model relationships: composition, collections, unions
• Inheritance: shows parent/child class relationships
• Field types: rendered as D2 sql_table shapes
• Self-references: handles recursive models correctly
• Docstrings: model and field descriptions appear as tooltips/constraints

Requirements

• Python 3.10+
• Pydantic 2.0+
• D2 (for rendering the output)

License

MIT