fix init KeyError (#61)

2020-10-12 20:59:13 +08:00 · 2020-10-12 20:59:13 +08:00 · 1fde3cd04e
commit 1fde3cd04e
parent d0ce545ff5
1 changed files with 3 additions and 1 deletions
--- a/aerich/cli.py
+++ b/aerich/cli.py
@ -28,7 +28,9 @@ def coro(f):
        loop = asyncio.get_event_loop()
        ctx = args[0]
        loop.run_until_complete(f(*args, **kwargs))
-        Migrate.remove_old_model_file(ctx.obj["app"], ctx.obj["location"])
+        app = ctx.obj.get("app")
+        if app:
+            Migrate.remove_old_model_file(app, ctx.obj["location"])

    return wrapper