sapl/sapl/comissoes/views.py

from django.contrib.auth.mixins import PermissionRequiredMixin
from django.core.urlresolvers import reverse
from django.views.generic import ListView

from sapl.crud.base import (Crud, CrudBaseMixin, CrudCreateView,
                            CrudDeleteView, CrudListView, CrudUpdateView)
from sapl.crud.masterdetail import MasterDetailCrud
from sapl.materia.models import Tramitacao
from sapl.utils import permissao_tb_aux, permissoes_comissoes

from .models import (CargoComissao, Comissao, Composicao, Participacao,
                     Periodo, TipoComissao)


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


class CargoCrud(Crud):
    model = CargoComissao
    help_path = 'cargo_comissao'

    class BaseMixin(PermissionRequiredMixin, CrudBaseMixin):
        list_field_names = ['nome', 'unico']

        def has_permission(self):
            return permissao_tb_aux(self)


class PeriodoComposicaoCrud(Crud):
    model = Periodo
    help_path = 'periodo_composicao_comissao'

    class BaseMixin(PermissionRequiredMixin, CrudBaseMixin):
        list_field_names = ['data_inicio', 'data_fim']

        def has_permission(self):
            return permissao_tb_aux(self)


class TipoComissaoCrud(Crud):
    model = TipoComissao
    help_path = 'tipo_comissao'

    class BaseMixin(PermissionRequiredMixin, CrudBaseMixin):
        list_field_names = ['sigla', 'nome', 'natureza',
                            'dispositivo_regimental']

        def has_permission(self):
            return permissao_tb_aux(self)


class ParticipacaoCrud(MasterDetailCrud):
    model = Participacao
    parent_field = 'composicao'
    help_path = ''

    class DetailView(MasterDetailCrud.DetailView):
        def get(self, request, *args, **kwargs):
            self.object = self.get_object()
            context = self.get_context_data(object=self.object)
            context['root_pk'] = self.object.composicao.comissao.pk
            return self.render_to_response(context)

    class CreateView(MasterDetailCrud.CreateView):

        def get_success_url(self):
            return reverse(
                'sapl.comissoes:composicao_detail',
                kwargs={'pk': self.kwargs['pk']}
            )

        def cancel_url(self):
            return reverse(
                'sapl.comissoes:composicao_detail',
                kwargs={'pk': self.kwargs['pk']}
            )

    class UpdateView(MasterDetailCrud.UpdateView):

        def get_success_url(self):
            return pegar_url_composicao(self.kwargs['pk'])

        def cancel_url(self):
            return pegar_url_composicao(self.kwargs['pk'])

    class DeleteView(MasterDetailCrud.DeleteView):

        def get_success_url(self):
            return pegar_url_composicao(self.kwargs['pk'])

        def cancel_url(self):
            return pegar_url_composicao(self.kwargs['pk'])

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


class ComposicaoCrud(MasterDetailCrud):
    model = Composicao
    parent_field = 'comissao'
    help_path = ''

    class DetailView(MasterDetailCrud.DetailView):

        def get(self, request, *args, **kwargs):
            self.object = self.get_object()
            context = self.get_context_data(object=self.object)
            composicao = Composicao.objects.get(id=self.kwargs['pk'])
            context['participacoes'] = composicao.participacao_set.all()
            return self.render_to_response(context)

    class CreateView(PermissionRequiredMixin, MasterDetailCrud.CreateView):
        permission_required = permissoes_comissoes()

    class UpdateView(PermissionRequiredMixin, MasterDetailCrud.UpdateView):
        permission_required = permissoes_comissoes()

    class DeleteView(PermissionRequiredMixin, MasterDetailCrud.DeleteView):
        permission_required = permissoes_comissoes()


class ComissaoCrud(Crud):
    model = Comissao
    help_path = 'modulo_comissoes'

    class CreateView(PermissionRequiredMixin, CrudCreateView):
        permission_required = permissoes_comissoes()

    class UpdateView(PermissionRequiredMixin, CrudUpdateView):
        permission_required = permissoes_comissoes()

    class DeleteView(PermissionRequiredMixin, CrudDeleteView):
        permission_required = permissoes_comissoes()

    class ListView(CrudListView):
        ordering = ['-ativa', 'sigla']

    class BaseMixin(CrudBaseMixin):
        list_field_names = ['nome', 'sigla', 'tipo', 'data_criacao', 'ativa']


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

    def get_queryset(self):
        pk = self.kwargs['pk']
        tramitacoes = Tramitacao.objects.filter(
            unidade_tramitacao_local__comissao=pk)
        return tramitacoes

    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