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

add checkout

Browse source
This commit is contained in:
Waylon Walker 2023-05-22 08:26:44 -05:00
parent 61fd56bf21
commit ad1b6d19b9
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
1 changed files with 17 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
learn_sql_model/cli/model.py
View file

@ -1,3 +1,5 @@
import alembic
from alembic.config import Config
import typer import typer
from learn_sql_model.cli.common import verbose_callback from learn_sql_model.cli.common import verbose_callback
@ -27,9 +29,6 @@ def create_revision(
prompt=True, prompt=True,
), ),
): ):
import alembic
# python -m alembic revision --autogenerate -m "New Attribute"
from alembic.config import Config
alembic_cfg = Config("alembic.ini") alembic_cfg = Config("alembic.ini")
alembic.command.revision( alembic.command.revision(
@ -37,6 +36,21 @@ def create_revision(
message=message, message=message,
autogenerate=True, autogenerate=True,
) )
alembic.command.upgrade(config=alembic_cfg, revision="head")
@model_app.command()
def checkout(
verbose: bool = typer.Option(
False,
callback=verbose_callback,
help="show the log messages",
),
revision: str = typer.Option("head"),
):
alembic_cfg = Config("alembic.ini")
alembic.command.upgrade(config=alembic_cfg, revision="head")
@model_app.command() @model_app.command()