From 47824a100bef8ea25889f6c4a47d3e879d57f376 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?L=C6=B0=C6=A1ng=20Quang=20M=E1=BA=A1nh?= Date: Sat, 26 Dec 2020 10:31:10 +0700 Subject: [PATCH] Fix Aerich b/c of Tortoise ORM v0.16.19 --- aerich/migrate.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/aerich/migrate.py b/aerich/migrate.py index 2b51b28..9f6903b 100644 --- a/aerich/migrate.py +++ b/aerich/migrate.py @@ -5,6 +5,7 @@ from datetime import datetime from importlib import import_module from io import StringIO from pathlib import Path +from types import ModuleType from typing import Dict, List, Optional, Tuple, Type import click @@ -210,7 +211,10 @@ class Migrate: old_model_files = [] models = config.get("apps").get(app).get("models") for model in models: - module = import_module(model) + if isinstance(model, ModuleType): + module = model + else: + module = import_module(model) possible_models = [getattr(module, attr_name) for attr_name in dir(module)] for attr in filter( lambda x: inspect.isclass(x) and issubclass(x, Model) and x is not Model,