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

Replace coro logic with tortoise.run_async

Browse Source
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
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
aerich/cli.py
View File

@ -8,7 +8,7 @@ from typing import List
import click import click
from click import Context, UsageError 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.exceptions import OperationalError
from tortoise.transactions import in_transaction from tortoise.transactions import in_transaction
from tortoise.utils import get_schema_sql from tortoise.utils import get_schema_sql
@ -34,11 +34,7 @@ parser = ConfigParser()
def coro(f): def coro(f):
@wraps(f) @wraps(f)
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
loop = asyncio.get_event_loop() run_async(f(*args, **kwargs))
try:
loop.run_until_complete(f(*args, **kwargs))
finally:
loop.run_until_complete(Tortoise.close_connections())
return wrapper return wrapper