wip
This commit is contained in:
parent
4be274d9e2
commit
c238b9d757
21 changed files with 219 additions and 184 deletions
|
|
@ -2,6 +2,7 @@ import typer
|
|||
import uvicorn
|
||||
|
||||
from learn_sql_model.cli.common import verbose_callback
|
||||
from learn_sql_model.config import config
|
||||
|
||||
api_app = typer.Typer()
|
||||
|
||||
|
|
@ -25,4 +26,4 @@ def run(
|
|||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
uvicorn.run("learn_sql_model.api.app:app", port=5000, log_level="info")
|
||||
uvicorn.run("learn_sql_model.api.app:app", port=config.port, log_level="info")
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@ from learn_sql_model.cli.api import api_app
|
|||
from learn_sql_model.cli.common import verbose_callback
|
||||
from learn_sql_model.cli.config import config_app
|
||||
from learn_sql_model.cli.hero import hero_app
|
||||
from learn_sql_model.cli.model_app import model_app
|
||||
from learn_sql_model.cli.model import model_app
|
||||
from learn_sql_model.cli.tui import tui_app
|
||||
|
||||
app = typer.Typer(
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@ from pydantic_typer import expand_pydantic_args
|
|||
from rich.console import Console
|
||||
import typer
|
||||
|
||||
from learn_sql_model.models import Hero
|
||||
from learn_sql_model.models.hero import Hero
|
||||
|
||||
hero_app = typer.Typer()
|
||||
|
||||
|
|
|
|||
50
learn_sql_model/cli/model.py
Normal file
50
learn_sql_model/cli/model.py
Normal file
|
|
@ -0,0 +1,50 @@
|
|||
import typer
|
||||
|
||||
from learn_sql_model.cli.common import verbose_callback
|
||||
|
||||
model_app = typer.Typer()
|
||||
|
||||
|
||||
@model_app.callback()
|
||||
def model(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
"model cli"
|
||||
|
||||
|
||||
@model_app.command()
|
||||
def create_revision(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
message: str = typer.Option(
|
||||
prompt=True,
|
||||
),
|
||||
):
|
||||
import alembic
|
||||
# python -m alembic revision --autogenerate -m "New Attribute"
|
||||
from alembic.config import Config
|
||||
|
||||
alembic_cfg = Config("alembic.ini")
|
||||
alembic.command.revision(
|
||||
config=alembic_cfg,
|
||||
message=message,
|
||||
autogenerate=True,
|
||||
)
|
||||
|
||||
|
||||
@model_app.command()
|
||||
def populate(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
...
|
||||
|
|
@ -1,119 +0,0 @@
|
|||
from rich.console import Console
|
||||
from sqlmodel import SQLModel, Session
|
||||
import typer
|
||||
|
||||
from learn_sql_model.cli.common import verbose_callback
|
||||
from learn_sql_model.config import config
|
||||
from learn_sql_model.models import Hero, Pet
|
||||
|
||||
model_app = typer.Typer()
|
||||
|
||||
|
||||
@model_app.callback()
|
||||
def model(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
"model cli"
|
||||
|
||||
|
||||
@model_app.command()
|
||||
def create_revision(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
message: str = typer.Option(
|
||||
prompt=True,
|
||||
),
|
||||
):
|
||||
import alembic
|
||||
# python -m alembic revision --autogenerate -m "New Attribute"
|
||||
from alembic.config import Config
|
||||
|
||||
alembic_cfg = Config("alembic.ini")
|
||||
alembic.command.revision(
|
||||
config=alembic_cfg,
|
||||
message=message,
|
||||
autogenerate=True,
|
||||
)
|
||||
|
||||
|
||||
@model_app.command()
|
||||
def show(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
|
||||
SQLModel.metadata.create_all(config.engine)
|
||||
with Session(config.engine) as session:
|
||||
heros = session.exec(select(Hero)).all()
|
||||
Console().print(heros)
|
||||
|
||||
|
||||
@model_app.command()
|
||||
def read(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
from learn_sql_model.api import read_heroes
|
||||
|
||||
Console().print(read_heroes())
|
||||
|
||||
|
||||
# @model_app.command()
|
||||
# @expand_pydantic_args(typer=True)
|
||||
# def create(
|
||||
# hero: Hero,
|
||||
# ):
|
||||
# hero.post()
|
||||
|
||||
# try:
|
||||
# httpx.post("http://localhost:5000/heroes/", json=hero.dict())
|
||||
# except httpx.ConnectError:
|
||||
# console.log("local failover")
|
||||
# with Session(config.engine) as session:
|
||||
# session.add(hero)
|
||||
# session.commit()
|
||||
|
||||
|
||||
@model_app.command()
|
||||
def populate(
|
||||
verbose: bool = typer.Option(
|
||||
False,
|
||||
callback=verbose_callback,
|
||||
help="show the log messages",
|
||||
),
|
||||
):
|
||||
|
||||
pet_1 = Pet(name="Deadpond-Dog")
|
||||
hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson", pets=[pet_1])
|
||||
hero_2 = Hero(
|
||||
name="Spider-Boy",
|
||||
secret_name="Pedro Parqueador",
|
||||
pet=Pet(name="Spider-Boy-Dog"),
|
||||
)
|
||||
hero_3 = Hero(
|
||||
name="Rusty-Man",
|
||||
secret_name="Tommy Sharp",
|
||||
age=48,
|
||||
pet=Pet(name="Rusty-Man-Dog"),
|
||||
)
|
||||
|
||||
SQLModel.metadata.create_all(config.engine)
|
||||
|
||||
with Session(config.engine) as session:
|
||||
session.add(hero_1)
|
||||
session.add(hero_2)
|
||||
session.add(hero_3)
|
||||
session.commit()
|
||||
Loading…
Add table
Add a link
Reference in a new issue