Browse Source

Retirada de um laço

pull/2626/head
Vinícius Cantuária 7 years ago
parent
commit
8e55bd1a6c
  1. 14
      scripts/remove_multiplos_parlamentares.py

14
scripts/remove_multiplos_parlamentares.py

@ -13,23 +13,21 @@ def get_multiple():
main_models = {} main_models = {}
for model in models: for model in models:
model_str = re.findall(r'\w+', str(model))[-1] model_name = re.findall(r'\w+', str(model))[-1]
main_models[model_str] = { main_models[model_name] = {
'model': model, 'model': model,
'kwargs': {}, 'kwargs': {},
'pks': [] 'pks': []
} }
for model_name, model_dict in main_models.items(): objs = main_models[model_name]['model'].objects.all()
objs = model_dict['model'].objects.all()
for obj in objs: for obj in objs:
if model_name == 'Autor': if model_name == 'Autor':
model_dict['kwargs']['nome'] = obj.nome main_models[model_name]['kwargs']['nome'] = obj.nome
elif model_name == 'Parlamentar': elif model_name == 'Parlamentar':
model_dict['kwargs']['nome_parlamentar'] = obj.nome_parlamentar main_models[model_name]['kwargs']['nome_parlamentar'] = obj.nome_parlamentar
pesquisa_obj = model_dict['model'].objects.filter(**model_dict['kwargs']) pesquisa_obj = main_models[model_name]['model'].objects.filter(**main_models[model_name]['kwargs'])
if pesquisa_obj.count() > 1: if pesquisa_obj.count() > 1:
multiplos_objs = [o.pk for o in pesquisa_obj] multiplos_objs = [o.pk for o in pesquisa_obj]
multiplos_objs.sort() multiplos_objs.sort()

Loading…
Cancel
Save