ju functionality demos

[thorwhalen]

A demo of ju functionalities.

[thorwhalen]

Visualizing your models as a graph

from ju.viz import model_digraph

from pydantic import BaseModel
from typing import List, Optional

class Address(BaseModel):
    street: str
    city: str
    state: str

class User(BaseModel):
    name: str
    email: str
    age: Optional[int] = None
    addresses: List[Address]

class SomethingElse(BaseModel):
    a: int
    b: str

dot = model_digraph([User, SomethingElse])

## to output in a file and get the path to it:
# dot.render('model_diagram', format='png', cleanup=True) 

# to get the Digraph object (if last line, will display in jupyter)
dot

TODOs:

• Handle ids and labels more carefully (e.g. what happens when the field names appear in different models?)
• Test loops and recursion artifacts (e.g. what happens when, in the above, b: str is replaced with b: Address)