From bf194ca8ce9b08cb91e502b30bf99247f8e4bb2c Mon Sep 17 00:00:00 2001
From: long2ice <long2ice@gmail.com>
Date: Wed, 3 Jun 2020 18:42:35 +0800
Subject: [PATCH] Update model file find method

---
 CHANGELOG.rst     | 4 ++++
 aerich/migrate.py | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index bb26271..2eb02fe 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -4,6 +4,10 @@ ChangeLog
 
 0.1
 ===
+0.2.0
+-----
+- Update model file find method.
+
 0.1.9
 -----
 - Fix default_connection when upgrade
diff --git a/aerich/migrate.py b/aerich/migrate.py
index 891217e..ad545a7 100644
--- a/aerich/migrate.py
+++ b/aerich/migrate.py
@@ -3,6 +3,7 @@ import os
 import re
 from copy import deepcopy
 from datetime import datetime
+from importlib import import_module
 from typing import Dict, List, Tuple, Type
 
 from tortoise import (
@@ -195,7 +196,7 @@ class Migrate:
         models = config.get("apps").get(app).get("models")
         for model in models:
             if model != "aerich.models":
-                old_model_files.append(model.replace(".", "/") + ".py")
+                old_model_files.append(import_module(model).__file__)
 
         cls.cp_models(app, old_model_files, os.path.join(location, app, cls.get_old_model_file()))