Close connections in the init_db wrapper

This commit is contained in:
Konstantin Mikhailov
2022-09-29 08:09:59 +10:00
committed by GitHub
parent 20aebc4413
commit d7ecd97e88

View File

@@ -30,7 +30,7 @@ def coro(f):
try: try:
loop.run_until_complete(f(*args, **kwargs)) loop.run_until_complete(f(*args, **kwargs))
finally: finally:
if f.__name__ not in ["cli", "init_db", "init"]: if f.__name__ not in ["cli", "init"]:
loop.run_until_complete(Tortoise.close_connections()) loop.run_until_complete(Tortoise.close_connections())
return wrapper return wrapper