From 881f70f74881133c236b234a7bf52825f47f2a86 Mon Sep 17 00:00:00 2001
From: long2ice <long2ice@prismslight.com>
Date: Mon, 25 May 2020 23:44:42 +0800
Subject: [PATCH] Fix default_connection when upgrade

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

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index fc0c877..f01ad17 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -4,6 +4,9 @@ ChangeLog
 
 0.1
 ===
+0.1.9
+-----
+- Fix default_connection when upgrade
 
 0.1.8
 -----
diff --git a/aerich/migrate.py b/aerich/migrate.py
index b8365de..9f6fddd 100644
--- a/aerich/migrate.py
+++ b/aerich/migrate.py
@@ -174,7 +174,10 @@ class Migrate:
         temp_config = deepcopy(config)
         path = os.path.join(location, app, cls.old_models)
         path = path.replace("/", ".").lstrip(".")
-        temp_config["apps"][cls.diff_app] = {"models": [path]}
+        temp_config["apps"][cls.diff_app] = {
+            "models": [path],
+            "default_connection": config.get("apps").get(app).get("default_connection","default"),
+        }
         return temp_config
 
     @classmethod