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

Decorator for closing db connection after func run

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

11
aerich/cli.py
View File

@ -1,3 +1,4 @@
import functools
import json import json
import os import os
import sys import sys
@ -27,6 +28,16 @@ class Color(str, Enum):
parser = ConfigParser() parser = ConfigParser()
def close_db(func):
@functools.wraps(func)
async def close_db_inner(*args, **kwargs):
result = await func(*args, **kwargs)
await Tortoise.close_connections()
return result
return close_db_inner
@click.group(context_settings={"help_option_names": ["-h", "--help"]}) @click.group(context_settings={"help_option_names": ["-h", "--help"]})
@click.version_option(__version__, "-V", "--version") @click.version_option(__version__, "-V", "--version")
@click.option( @click.option(