fix init KeyError (#61)

This commit is contained in:
long2ice 2020-10-12 20:59:13 +08:00
parent d0ce545ff5
commit 1fde3cd04e

View File

@ -28,7 +28,9 @@ def coro(f):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
ctx = args[0] ctx = args[0]
loop.run_until_complete(f(*args, **kwargs)) 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 return wrapper