From 8e55bd1a6cdab656e3b4af313b0a53deffd4111d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Vin=C3=ADcius=20Cantu=C3=A1ria?= Date: Wed, 3 Apr 2019 14:38:58 -0300 Subject: [PATCH] =?UTF-8?q?Retirada=20de=20um=20la=C3=A7o?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/remove_multiplos_parlamentares.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/scripts/remove_multiplos_parlamentares.py b/scripts/remove_multiplos_parlamentares.py index 20fed7650..3d7450295 100644 --- a/scripts/remove_multiplos_parlamentares.py +++ b/scripts/remove_multiplos_parlamentares.py @@ -13,23 +13,21 @@ def get_multiple(): main_models = {} for model in models: - model_str = re.findall(r'\w+', str(model))[-1] - main_models[model_str] = { + model_name = re.findall(r'\w+', str(model))[-1] + main_models[model_name] = { 'model': model, 'kwargs': {}, 'pks': [] } - for model_name, model_dict in main_models.items(): - objs = model_dict['model'].objects.all() - + objs = main_models[model_name]['model'].objects.all() for obj in objs: if model_name == 'Autor': - model_dict['kwargs']['nome'] = obj.nome + main_models[model_name]['kwargs']['nome'] = obj.nome 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: multiplos_objs = [o.pk for o in pesquisa_obj] multiplos_objs.sort()