create all on startup

learn_sql_model/cli/hero.py

@@ -2,9 +2,14 @@ from typing import List, Union
 
 from pydantic_typer import expand_pydantic_args
 from rich.console import Console
+from sqlmodel import SQLModel
 import typer
 
+from learn_sql_model.config import Config, get_config
+from learn_sql_model.factories.hero import HeroFactory
+from learn_sql_model.factories.pet import PetFactory
 from learn_sql_model.models.hero import Hero
+from learn_sql_model.models.pet import Pet
 
 hero_app = typer.Typer()
 
@@ -12,19 +17,45 @@ hero_app = typer.Typer()
 @hero_app.callback()
 def hero():
     "model cli"
+    SQLModel.metadata.create_all(get_config().database.engine)
 
 
 @hero_app.command()
 def get(id: int = None) -> Union[Hero, List[Hero]]:
     "get one hero"
-    hero = Hero().get(item_id=id)
+    hero = Hero().get(id=id)
     Console().print(hero)
     return hero
 
 
 @hero_app.command()
 @expand_pydantic_args(typer=True)
-def create(hero: Hero) -> Hero:
+def create(
+    hero: Hero,
+    pet: Pet = None,
+    config: Config = None,
+) -> Hero:
     "read all the heros"
-    hero = hero.post()
+    hero.pet = pet
+    hero = hero.post(config=config)
     Console().print(hero)
+
+
+@hero_app.command()
+@expand_pydantic_args(typer=True)
+def populate(
+    n: int = 10,
+    config: Config = None,
+) -> Hero:
+    "read all the heros"
+    if config is None:
+        config = Config()
+    if config.env == "prod":
+        Console().print("populate is not supported in production")
+        return
+
+    for hero in HeroFactory().batch(n):
+        pet = PetFactory().build()
+        hero.pet = pet
+        Console().print(hero)
+        hero.post(config=config)