gault-millau/apps/transfer/utils.py
littlewolf 673acb4bb9 Add sort
Move dependencies to model
Add data_type to model
2019-10-03 23:19:40 +03:00

56 lines
1.9 KiB
Python

from os.path import exists
from importlib.machinery import SourceFileLoader
from django.apps import apps
from collections import OrderedDict
def transfer_objects(data_type):
models_list = {}
for app in apps.get_app_configs():
if exists(f"{app.path}/transfer.py"):
card_module = SourceFileLoader("transfer", f"{app.path}/transfer.py").load_module()
if not hasattr(card_module, "card") or len(card_module.card) < 1:
continue
for model, card in card_module.card.items():
if "data_type" in card and data_type == card["data_type"]:
card['name'] = model
models_list[model] = card
if len(models_list) < 1:
print(f"Models with data type {data_type} not found in structure")
exit(1)
models_list = sort_by_dependencies(models_list)
print(f"==========================\r\n{models_list}\r\n==========================\r\n")
for model, card in models_list.items():
print(model)
print(card)
def sort_by_dependencies(data):
"""Сначала мы сортируем модели по зависимостям в обратном порядке, используя сортировку вставкой"""
result = []
for model, card in data.items():
if "dependencies" in card and isinstance(card['dependencies'], tuple):
for model_dependency in result:
if model_dependency in card['dependencies']:
result.insert(result.index(model_dependency), model)
break
else:
result.append(model)
"""Затем мы создаём сортированный словарь из реверса получившегося результата"""
print(result)
result = OrderedDict(
[(model, data[model]) for model in reversed(result)]
)
return result