Correção da classe mesas.Legislatura para registrar as legislaturas e o número de parlamentares por Casa Legislativa

sigi/apps/mesas/admin.py

@@ -6,10 +6,16 @@ from sigi.apps.mesas.models import (Legislatura, Coligacao, ComposicaoColigacao,
 
 class LegislaturaAdmin(admin.ModelAdmin):
     date_hierarchy = 'data_inicio'
-    list_display = ('numero', 'data_inicio', 'data_fim', 'data_eleicao')
+    list_display = ('casa_legislativa', 'numero', 'data_inicio', 'data_fim', 'data_eleicao', 'total_parlamentares')
+    raw_id_fields = ('casa_legislativa',)
     list_display_links = ('numero',)
-    list_filter = ('data_eleicao',)
-    search_fields = ('numero',)
+    list_filter = ('casa_legislativa', 'data_eleicao',)
+    search_fields = ('numero', 'casa_legislativa')
+    
+    def lookup_allowed(self, lookup, value):
+        return super(LegislaturaAdmin, self).lookup_allowed(lookup, value) or \
+            lookup in ['casa_legislativa__municipio__uf__codigo_ibge__exact']
+    
 
 class ColigacaoAdmin(admin.ModelAdmin):
     list_display = ('nome', 'legislatura', 'numero_votos')

sigi/apps/mesas/models.py

@@ -1,11 +1,20 @@
 # -*- coding: utf-8 -*-
 from django.db import models
+from sigi.apps.casas.models import CasaLegislativa
 
 class Legislatura(models.Model):
+    casa_legislativa = models.ForeignKey(CasaLegislativa)
     numero = models.PositiveSmallIntegerField(u'número legislatura')
     data_inicio = models.DateField(u'início')
     data_fim = models.DateField('fim')
     data_eleicao = models.DateField(u'data da eleição')
+    total_parlamentares = models.PositiveIntegerField(u"Total de parlamentares")
+    
+    casa_legislativa.convenio_uf_filter = True
+    casa_legislativa.convenio_cl_tipo_filter = True
+    
+    def meta(self):
+        unique_together = (('casa_legislativa', 'numero'))
 
     def __unicode__(self):
         return str(self.numero)