Don't close db connections when group function \(cli\) is run

This commit is contained in:
AulonSal 2021-02-27 00:43:55 +05:30
parent 14d20455e6
commit e74fc304a5
No known key found for this signature in database
GPG Key ID: A844B4CD17044BC8

View File

@ -8,7 +8,7 @@ from typing import List
import click
from click import Context, UsageError
from tortoise import Tortoise, generate_schema_for_client, run_async
from tortoise import Tortoise, generate_schema_for_client
from tortoise.exceptions import OperationalError
from tortoise.transactions import in_transaction
from tortoise.utils import get_schema_sql
@ -34,7 +34,14 @@ parser = ConfigParser()
def coro(f):
@wraps(f)
def wrapper(*args, **kwargs):
run_async(f(*args, **kwargs))
loop = asyncio.get_event_loop()
# Close db connections at the end of all all but the cli group function
try:
loop.run_until_complete(f(*args, **kwargs))
finally:
if f.__name__ != "cli":
loop.run_until_complete(Tortoise.close_connections())
return wrapper