From 90b47c5af7029403e5d8ccccf61f88e88b52c562 Mon Sep 17 00:00:00 2001
From: AulonSal <aulonsal@gmail.com>
Date: Mon, 22 Feb 2021 07:37:40 +0530
Subject: [PATCH] Close connections even if command raises exception

---
 aerich/cli.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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