From 125389461fdcda4c2f36c9fe965222481a42ef0a Mon Sep 17 00:00:00 2001
From: long2ice <long2ice@gmail.com>
Date: Tue, 26 May 2020 14:44:55 +0800
Subject: [PATCH] check tortoise add aerich.models

---
 CHANGELOG.rst | 1 +
 aerich/cli.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index 8bb539d..bb26271 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -9,6 +9,7 @@ ChangeLog
 - Fix default_connection when upgrade
 - Find default app instead of default.
 - Diff MySQL ddl.
+- Check tortoise config.
 
 0.1.8
 -----
diff --git a/aerich/cli.py b/aerich/cli.py
index 8d9a349..af4e19d 100644
--- a/aerich/cli.py
+++ b/aerich/cli.py
@@ -57,6 +57,8 @@ async def cli(ctx: Context, config, app, name):
 
         tortoise_config = get_tortoise_config(ctx, tortoise_orm)
         app = app or list(tortoise_config.get("apps").keys())[0]
+        if "aerich.models" not in tortoise_config.get("apps").get(app).get("models"):
+            raise UsageError("Check your tortoise config and add aerich.models to it.", ctx=ctx)
         ctx.obj["config"] = tortoise_config
         ctx.obj["location"] = location
         ctx.obj["app"] = app