pub/aerich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use close_db decorator on command functions

Browse Source
This commit is contained in:
Patrick Schneeweis 2020-07-24 09:22:57 -04:00
parent 5bdaa32a9e
commit 9bd96a9487
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
aerich/cli.py
View File

@ -81,6 +81,7 @@ async def cli(ctx: Context, config, app, name):
@cli.command(help="Generate migrate changes file.") @cli.command(help="Generate migrate changes file.")
@click.option("--name", default="update", show_default=True, help="Migrate name.") @click.option("--name", default="update", show_default=True, help="Migrate name.")
@click.pass_context @click.pass_context
@close_db
async def migrate(ctx: Context, name): async def migrate(ctx: Context, name):
config = ctx.obj["config"] config = ctx.obj["config"]
location = ctx.obj["location"] location = ctx.obj["location"]
@ -95,6 +96,7 @@ async def migrate(ctx: Context, name):
@cli.command(help="Upgrade to latest version.") @cli.command(help="Upgrade to latest version.")
@click.pass_context @click.pass_context
@close_db
async def upgrade(ctx: Context): async def upgrade(ctx: Context):
config = ctx.obj["config"] config = ctx.obj["config"]
app = ctx.obj["app"] app = ctx.obj["app"]
@ -121,6 +123,7 @@ async def upgrade(ctx: Context):
@cli.command(help="Downgrade to previous version.") @cli.command(help="Downgrade to previous version.")
@click.pass_context @click.pass_context
@close_db
async def downgrade(ctx: Context): async def downgrade(ctx: Context):
app = ctx.obj["app"] app = ctx.obj["app"]
config = ctx.obj["config"] config = ctx.obj["config"]
@ -143,6 +146,7 @@ async def downgrade(ctx: Context):
@cli.command(help="Show current available heads in migrate location.") @cli.command(help="Show current available heads in migrate location.")
@click.pass_context @click.pass_context
@close_db
async def heads(ctx: Context): async def heads(ctx: Context):
app = ctx.obj["app"] app = ctx.obj["app"]
versions = Migrate.get_all_version_files() versions = Migrate.get_all_version_files()
@ -207,6 +211,7 @@ async def init(
show_default=True, show_default=True,
) )
@click.pass_context @click.pass_context
@close_db
async def init_db(ctx: Context, safe): async def init_db(ctx: Context, safe):
config = ctx.obj["config"] config = ctx.obj["config"]
location = ctx.obj["location"] location = ctx.obj["location"]