From 14d20455e613e99cfc27ecbdde7c3ba720ef6358 Mon Sep 17 00:00:00 2001 From: AulonSal Date: Tue, 23 Feb 2021 13:06:40 +0530 Subject: [PATCH] Replace coro logic with tortoise.run_async --- aerich/cli.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/aerich/cli.py b/aerich/cli.py index 9bc8b6d..26a0164 100644 --- a/aerich/cli.py +++ b/aerich/cli.py @@ -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