From f0a7ceed7625f68654453d90052a43dd4cad58f1 Mon Sep 17 00:00:00 2001 From: Edward Ribeiro Date: Fri, 25 Sep 2015 14:57:33 -0300 Subject: [PATCH] Add get_parlamentares to OrdemDia e VotacaoNominal --- sessao/views.py | 39 ++++++++++++++------------------------- 1 file changed, 14 insertions(+), 25 deletions(-) diff --git a/sessao/views.py b/sessao/views.py index 9b9e40cff..69927e716 100644 --- a/sessao/views.py +++ b/sessao/views.py @@ -189,24 +189,19 @@ class PresencaOrdemDiaView(FormMixin, sessao_crud.CrudDetailView): return reverse('sessaoplenaria:presencaordemdia', kwargs={'pk': pk}) def get_parlamentares(self): - self.object = self.get_object() + self.object = self.get_object() presencas = PresencaOrdemDia.objects.filter( sessao_plenaria_id=self.object.id ) - presentes = [] - for p in presencas: - presentes.append(p.parlamentar.id) + presentes = [p.parlamentar for p in presencas] - for parlamentar in Parlamentar.objects.all(): - if parlamentar.ativo: - try: - presentes.index(parlamentar.id) - except ValueError: - yield (parlamentar, False) - else: - yield (parlamentar, True) + for parlamentar in Parlamentar.objects.filter(ativo=True): + if parlamentar in presentes: + yield (parlamentar, True) + else: + yield (parlamentar, False) class ListMateriaOrdemDiaView(sessao_crud.CrudDetailView): @@ -1236,19 +1231,13 @@ class VotacaoNomimalView(FormMixin, sessao_crud.CrudDetailView): sessao_plen_id=self.object.id ) - presentes = [] - for p in presencas: - presentes.append(p.parlamentar.id) - - for parlamentar in Parlamentar.objects.all(): - if parlamentar.ativo: - try: - presentes.index(parlamentar.id) - except ValueError: - pass - else: - yield parlamentar - + presentes = [p.parlamentar for p in presencas] + + for parlamentar in Parlamentar.objects.filter(ativo=True): + if parlamentar in presentes: + yield (parlamentar, True) + else: + yield (parlamentar, False) class VotacaoSecretaView(FormMixin, sessao_crud.CrudDetailView): template_name = 'sessao/votacao/secreta.html'