!58 update backend/dvadmin/utils/models.py.

Merge pull request !58 from 木子-李/N/A
This commit is contained in:
dvadmin
2024-07-05 00:47:11 +00:00
committed by Gitee

View File

@@ -216,9 +216,13 @@ def get_all_models_objects(model_name=None):
def get_model_from_app(app_name): def get_model_from_app(app_name):
"""获取模型里的字段""" """获取模型里的字段"""
model_module = import_module(app_name + '.models') model_module = import_module(app_name + '.models')
exclude_models = getattr(model_module, 'exclude_models', [])
filter_model = [ filter_model = [
getattr(model_module, item) for item in dir(model_module) value for key, value in model_module.__dict__.items()
if item != 'CoreModel' and issubclass(getattr(model_module, item).__class__, models.base.ModelBase) if key != 'CoreModel'
and isinstance(value, type)
and issubclass(value, models.Model)
and key not in exclude_models
] ]
model_list = [] model_list = []
for model in filter_model: for model in filter_model: