Close connections even if command raises exception

This commit is contained in:
AulonSal 2021-02-22 07:37:40 +05:30
parent 02fe5a9d31
commit 90b47c5af7
No known key found for this signature in database
GPG Key ID: A844B4CD17044BC8

View File

@ -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