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

Merge pull request #111 from lqmanh/bugfixes/fix-tortoise-orm-0.16.19

Browse Source 
Fix Aerich b/c of a new feature in Tortoise ORM v0.16.19
This commit is contained in:
long2ice 2021-01-04 14:59:11 +08:00 committed by GitHub
commit cd176c1fd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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,6 +211,9 @@ 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:
if isinstance(model, ModuleType):
module = model
else:
module = import_module(model) 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(