Refactor parlamentar list

parlamentares/views.py

@@ -158,24 +158,63 @@ class ParlamentaresView(GenericView):
 
     def get(self, request, *args, **kwargs):
         form = ParlamentaresListForm()
+
         legislaturas = Legislatura.objects.all().order_by(
             '-data_inicio', '-data_fim')
+
+        mandatos = Mandato.objects.filter(
+            legislatura_id=legislaturas.first().id)
+
+        parlamentares = []
+        dict_parlamentar = {}
+        for m in mandatos:
+
+            if m.parlamentar.filiacao_set.last():
+                partido = m.parlamentar.filiacao_set.last().partido.sigla
+            else:
+                partido = 'Sem Registro'
+
+            dict_parlamentar = {
+                'id': m.parlamentar.id,
+                'nome': m.parlamentar.nome_parlamentar,
+                'partido': partido,
+                'ativo': m.parlamentar.ativo}
+            parlamentares.append(dict_parlamentar)
+
         return self.render_to_response(
             {'legislaturas': legislaturas,
              'legislatura_id': legislaturas.first().id,
-             'mandatos': Mandato.objects.all(),
              'form': form,
-             'filiacao': Filiacao.objects.all()})
+             'parlamentares': parlamentares})
 
     def post(self, request, *args, **kwargs):
         form = ParlamentaresListForm(request.POST)
+
+        mandatos = Mandato.objects.filter(
+            legislatura_id=int(form.data['periodo']))
+
+        parlamentares = []
+        dict_parlamentar = {}
+        for m in mandatos:
+
+            if m.parlamentar.filiacao_set.last():
+                partido = m.parlamentar.filiacao_set.last().partido.sigla
+            else:
+                partido = 'Sem Registro'
+
+            dict_parlamentar = {
+                'id': m.parlamentar.id,
+                'nome': m.parlamentar.nome_parlamentar,
+                'partido': partido,
+                'ativo': m.parlamentar.ativo}
+            parlamentares.append(dict_parlamentar)
+
         return self.render_to_response(
             {'legislaturas': Legislatura.objects.all().order_by(
                 '-data_inicio', '-data_fim'),
              'legislatura_id': int(form.data['periodo']),
-             'mandatos': Mandato.objects.all(),
              'form': form,
-             'filiacao': Filiacao.objects.all()})
+             'parlamentares': parlamentares})
 
 
 class HorizontalRadioRenderer(forms.RadioSelect.renderer):

templates/parlamentares/parlamentares_list.html

@@ -26,20 +26,14 @@
 				<th>Ativo?</th>
 			</tr>
 
-			{% for m in mandatos %}
-				{% if m.legislatura_id == legislatura_id %}
+			{% for p in parlamentares %}
 				<tr>
-						<td><a href="{% url 'parlamentares_editar' legislatura_id m.parlamentar.id %}">{{m.parlamentar.nome_parlamentar}}</a></td>
-						<td>
-							{% for f in filiacao %}
-								{% if f.parlamentar.id == m.parlamentar.id %}
-									{{f.partido}}
-								{% endif %}
-							{% endfor %}
-						</td>
-						<td>{{m.parlamentar.ativo|yesno:"Sim,Não"}}</td>
+					<td><a href="{% url 'parlamentares_editar' legislatura_id p.id %}">
+						{{p.nome}}
+					</a></td>	
+					<td>{{p.partido}}</td>					
+					<td>{{p.ativo|yesno:"Sim,Não"}}</td>
 				</tr>
-				{% endif %}
 			{% endfor %}
 		</table>
 	</fieldset>