pub/aerich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Aerich b/c of Tortoise ORM v0.16.19

Browse Source
This commit is contained in:
Lương Quang Mạnh 2020-12-26 10:31:10 +07:00
parent 78a15f9f19
commit 47824a100b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
aerich/migrate.py
View File

@ -5,6 +5,7 @@ from datetime import datetime
from importlib import import_module from importlib import import_module
from io import StringIO from io import StringIO
from pathlib import Path from pathlib import Path
from types import ModuleType
from typing import Dict, List, Optional, Tuple, Type from typing import Dict, List, Optional, Tuple, Type
import click import click
@ -210,7 +211,10 @@ class Migrate:
old_model_files = [] old_model_files = []
models = config.get("apps").get(app).get("models") models = config.get("apps").get(app).get("models")
for model in 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)] possible_models = [getattr(module, attr_name) for attr_name in dir(module)]
for attr in filter( for attr in filter(
lambda x: inspect.isclass(x) and issubclass(x, Model) and x is not Model, lambda x: inspect.isclass(x) and issubclass(x, Model) and x is not Model,