sapl/sapl/comissoes/views.py

from django.core.urlresolvers import reverse
from django.db.models import F
from django.views.decorators.clickjacking import xframe_options_exempt
from django.views.generic import ListView
from django.views.generic.base import RedirectView
from django.views.generic.detail import DetailView
from django.views.generic.edit import FormMixin


from sapl.base.models import AppConfig as AppsAppConfig
from sapl.crud.base import (RP_DETAIL, RP_LIST, Crud,
                           CrudAux, MasterDetailCrud,
                           PermissionRequiredForAppCrudMixin)
from sapl.comissoes.forms import ParticipacaoCreateForm, ParticipacaoEditForm
from sapl.materia.models import MateriaLegislativa, Tramitacao

from .forms import ReuniaoForm, ComissaoForm

from .models import (CargoComissao, Comissao, Composicao, Participacao,
                     Periodo, TipoComissao, Reuniao)
from sapl.comissoes.apps import AppConfig


def pegar_url_composicao(pk):
    participacao = Participacao.objects.get(id=pk)
    comp_pk = participacao.composicao.pk
    url = reverse('sapl.comissoes:composicao_detail', kwargs={'pk': comp_pk})
    return url


CargoCrud = CrudAux.build(CargoComissao, 'cargo_comissao')
PeriodoComposicaoCrud = CrudAux.build(Periodo, 'periodo_composicao_comissao')

TipoComissaoCrud = CrudAux.build(
    TipoComissao, 'tipo_comissao', list_field_names=[
        'sigla', 'nome', 'natureza', 'dispositivo_regimental'])


class ParticipacaoCrud(MasterDetailCrud):
    model = Participacao
    parent_field = 'composicao__comissao'
    public = [RP_DETAIL, ]
    ListView = None
    link_return_to_parent_field = True

    class BaseMixin(MasterDetailCrud.BaseMixin):
        list_field_names = ['composicao', 'parlamentar', 'cargo']

    class CreateView(MasterDetailCrud.CreateView):
        form_class = ParticipacaoCreateForm

        def get_initial(self):
            initial = super().get_initial()
            initial['parent_pk'] = self.kwargs['pk']
            return initial

    class UpdateView(MasterDetailCrud.UpdateView):
        layout_key = 'ParticipacaoEdit'
        form_class = ParticipacaoEditForm

    class DeleteView(MasterDetailCrud.DeleteView):
        def get_success_url(self):
            composicao_comissao_pk = self.object.composicao.comissao.pk
            composicao_pk = self.object.composicao.pk
            return '{}?pk={}'.format(reverse('sapl.comissoes:composicao_list',
                                             args=[composicao_comissao_pk]),
                                     )


class ComposicaoCrud(MasterDetailCrud):
    model = Composicao
    parent_field = 'comissao'
    model_set = 'participacao_set'
    public = [RP_LIST, RP_DETAIL, ]

    class ListView(MasterDetailCrud.ListView):
        template_name = "comissoes/composicao_list.html"
        paginate_by = None

        def take_composicao_pk(self):
            try:
                return int(self.request.GET['pk'])
            except:
                return 0

        def get_context_data(self, **kwargs):
            context = super().get_context_data(**kwargs)

            composicao_pk = self.take_composicao_pk()

            if composicao_pk == 0:
                ultima_composicao = context['composicao_list'].last()
                if ultima_composicao:
                    context['composicao_pk'] = ultima_composicao.pk
                else:
                    context['composicao_pk'] = 0
            else:
                context['composicao_pk'] = composicao_pk

            context['participacao_set'] = Participacao.objects.filter(
                composicao__pk=context['composicao_pk']
            ).order_by('id')
            return context


class ComissaoCrud(Crud):
    model = Comissao
    help_topic = 'modulo_comissoes'
    public = [RP_LIST, RP_DETAIL, ]

    class BaseMixin(Crud.BaseMixin):
        list_field_names = ['nome', 'sigla', 'tipo',
                            'data_criacao', 'data_extincao', 'ativa']
        ordering = '-ativa', 'sigla'

    class CreateView(Crud.CreateView):
        form_class = ComissaoForm

        def form_valid(self, form):
            return super(Crud.CreateView, self).form_valid(form)


class MateriasTramitacaoListView(ListView):
    template_name = "comissoes/materias_em_tramitacao.html"
    paginate_by = 10

    def get_queryset(self):
        # FIXME: Otimizar consulta
        ts = Tramitacao.objects.order_by(
            'materia', '-data_tramitacao', '-id').annotate(
            comissao=F('unidade_tramitacao_destino__comissao')).distinct(
                'materia').values_list('materia', 'comissao')

        ts = list(filter(lambda x: x[1] == int(self.kwargs['pk']), ts))
        ts = list(zip(*ts))
        ts = ts[0] if ts else []

        materias = MateriaLegislativa.objects.filter(
            pk__in=ts).order_by('tipo', '-ano', '-numero')

        return materias

    def get_context_data(self, **kwargs):
        context = super(
            MateriasTramitacaoListView, self).get_context_data(**kwargs)
        context['object'] = Comissao.objects.get(id=self.kwargs['pk'])
        return context

class ReuniaoCrud(MasterDetailCrud):
    model = Reuniao
    parent_field = 'comissao'
    public = [RP_LIST, RP_DETAIL, ]

    class BaseMixin(MasterDetailCrud.BaseMixin):
        list_field_names = ['nome', 'tema', 'comissao']

        @property
        def list_url(self):
            return ''

    class ListView(MasterDetailCrud.ListView):
        paginate_by = 10

    class UpdateView(MasterDetailCrud.UpdateView):
        form_class = ReuniaoForm

        def get_initial(self):
            return {'comissao': self.object.comissao}


    class CreateView(MasterDetailCrud.CreateView):
        form_class = ReuniaoForm

        def get_initial(self):
          comissao = Comissao.objects.get(id=self.kwargs['pk'])

          return {'comissao': comissao}


    class DeleteView(MasterDetailCrud.DeleteView):
        pass


    class DetailView(MasterDetailCrud.DetailView):

        @xframe_options_exempt
        def get(self, request, *args, **kwargs):
            return super().get(request, *args, **kwargs)