waylon/learn-sql-model
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test em

Browse source
This commit is contained in:
Waylon Walker 2023-05-23 08:55:35 -05:00
parent daf81343bf
commit a2b33b25f8
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
11 changed files with 479 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

8
learn_sql_model/cli/hero.py
View file

@ -1,4 +1,4 @@
from typing import List, Union
from typing import List, Optional, Union
from pydantic_typer import expand_pydantic_args
from rich.console import Console
@ -9,6 +9,7 @@ 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
import sys
hero_app = typer.Typer()
@ -21,7 +22,7 @@ def hero():
@hero_app.command()
@expand_pydantic_args(typer=True)
def get(
id: int = None,
id: Optional[int] = None,
config: Config = None,
) -> Union[Hero, List[Hero]]:
"get one hero"
@ -52,12 +53,11 @@ def populate(
config: Config = None,
) -> Hero:
"read all the heros"
config.init()
if config is None:
config = Config()
if config.env == "prod":
Console().print("populate is not supported in production")
return
sys.exit(1)
for hero in HeroFactory().batch(n):
pet = PetFactory().build()