Replace coro logic with tortoise.run_async

This commit is contained in:
AulonSal 2021-02-23 13:06:40 +05:30
parent 90b47c5af7
commit 14d20455e6
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
from tortoise import Tortoise, generate_schema_for_client, run_async
from tortoise.exceptions import OperationalError
from tortoise.transactions import in_transaction
from tortoise.utils import get_schema_sql
@ -34,11 +34,7 @@ parser = ConfigParser()
def coro(f):
@wraps(f)
def wrapper(*args, **kwargs):
loop = asyncio.get_event_loop()
try:
loop.run_until_complete(f(*args, **kwargs))
finally:
loop.run_until_complete(Tortoise.close_connections())
run_async(f(*args, **kwargs))
return wrapper