Fix sqlite stuck. (#90)

tortoise · Nov 26, 2020 · 0bbc471 · 0bbc471
1 parent fb6cc62
commit 0bbc471
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 43 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@
 - Use `.sql` instead of `.json` to store version file.
 - Add `rename` column support MySQL5.
 - Remove callable detection for defaults. (#87)
+- Fix `sqlite` stuck. (#90)
 
 ## 0.3
 

diff --git a/aerich/cli.py b/aerich/cli.py
@@ -33,6 +33,7 @@ def wrapper(*args, **kwargs):
         loop = asyncio.get_event_loop()
         ctx = args[0]
         loop.run_until_complete(f(*args, **kwargs))
+        loop.run_until_complete(Tortoise.close_connections())
         app = ctx.obj.get("app")
         if app:
             Migrate.remove_old_model_file(app, ctx.obj["location"])
@@ -282,7 +283,7 @@ async def init_db(ctx: Context, safe):
         "upgrade": [schema],
     }
     write_version_file(os.path.join(dirname, version), content)
-    return click.secho(f'Success generate schema for app "{app}"', fg=Color.green)
+    click.secho(f'Success generate schema for app "{app}"', fg=Color.green)
 
 
 def main():

diff --git a/poetry.lock b/poetry.lock