diff --git a/sigi/apps/mesas/admin.py b/sigi/apps/mesas/admin.py index 2b0f413..5267db0 100644 --- a/sigi/apps/mesas/admin.py +++ b/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') diff --git a/sigi/apps/mesas/models.py b/sigi/apps/mesas/models.py index efe0e1d..c5f65e6 100644 --- a/sigi/apps/mesas/models.py +++ b/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)