diff --git a/aerich/cli.py b/aerich/cli.py index 20fa4a6..9bc8b6d 100644 --- a/aerich/cli.py +++ b/aerich/cli.py @@ -35,8 +35,10 @@ def coro(f): @wraps(f) def wrapper(*args, **kwargs): loop = asyncio.get_event_loop() - loop.run_until_complete(f(*args, **kwargs)) - loop.run_until_complete(Tortoise.close_connections()) + try: + loop.run_until_complete(f(*args, **kwargs)) + finally: + loop.run_until_complete(Tortoise.close_connections()) return wrapper