Merge pull request #272 from interlegis/265-Refatorar_protocolo

265 refatorar protocolo

protocoloadm/forms.py

@@ -17,6 +17,7 @@ from sapl.utils import RANGE_ANOS
 from .models import (DocumentoAcessorioAdministrativo, DocumentoAdministrativo,
                      Protocolo, TipoDocumentoAdministrativo,
                      TramitacaoAdministrativo)
+from materia.models import Autor
 
 TIPOS_PROTOCOLO = [('0', 'Enviado'), ('1', 'Recebido')]
 
@@ -286,7 +287,7 @@ class ProtocoloDocumentForm(ModelForm):
             *args, **kwargs)
 
 
-class ProtocoloMateriaForm(forms.Form):
+class ProtocoloMateriaForm(ModelForm):
 
     NUMERACAO_CHOICES = [('1', 'Sequencial por Ano'),
                          ('2', 'Sequencial Único')]
@@ -303,22 +304,26 @@ class ProtocoloMateriaForm(forms.Form):
                                        widget=forms.RadioSelect(
                                            renderer=HorizontalRadioRenderer))
 
-    tipo_materia = forms.ModelChoiceField(
+    autor = forms.IntegerField(widget=forms.HiddenInput(), required=False)
-        label='Tipo de Matéria',
-        required=False,
-        queryset=TipoMateriaLegislativa.objects.all(),
-        empty_label='Selecione',
-    )
-
-    num_paginas = forms.CharField(label='Núm. Páginas', required=True)
-    ementa = forms.CharField(
-        widget=forms.Textarea, label='Ementa', required=True)
 
-    autor = forms.CharField(widget=forms.HiddenInput(), required=False)
+    def clean_autor(self):
+        autor_field = self.cleaned_data['autor']
+        try:
+            autor = Autor.objects.get(id=autor_field)
+        except ObjectDoesNotExist:
+            autor_field = None
+        else:
+            autor_field = autor
+        return autor_field
 
-    observacao = forms.CharField(required=True,
+    class Meta:
-                                 widget=forms.Textarea,
+        model = Protocolo
-                                 label='Observação')
+        fields = ['numeracao',
+                  'tipo_protocolo',
+                  'tipo_materia',
+                  'numero_paginas',
+                  'autor',
+                  'observacao']
 
     def __init__(self, *args, **kwargs):
 
@@ -327,7 +332,7 @@ class ProtocoloMateriaForm(forms.Form):
         row2 = crispy_layout_mixin.to_row(
             [('tipo_materia', 4),
              ('tipo_protocolo', 4),
-             ('num_paginas', 4)])
+             ('numero_paginas', 4)])
         row3 = crispy_layout_mixin.to_row(
             [('autor', 0),
              (Button('pesquisar',
@@ -335,11 +340,9 @@ class ProtocoloMateriaForm(forms.Form):
                      css_class='btn btn-primary btn-sm'), 2),
              (Button('limpar',
                      'limpar Autor',
-                     css_class='btn btn-primary btn-sm'), 2)])
+                     css_class='btn btn-primary btn-sm'), 10)])
         row4 = crispy_layout_mixin.to_row(
             [('observacao', 12)])
-        row5 = crispy_layout_mixin.to_row(
-            [('ementa', 12)])
 
         self.helper = FormHelper()
         self.helper.layout = Layout(
@@ -350,7 +353,6 @@ class ProtocoloMateriaForm(forms.Form):
                      HTML(sapl.utils.autor_modal),
                      row3,
                      row4,
-                     row5,
                      form_actions(save_label='Protocolar Matéria')
                      )
         )

protocoloadm/views.py

@@ -115,87 +115,62 @@ class ProtocoloListView(ListView):
         return context
 
 
-class AnularProtocoloAdmView(FormView):
+class AnularProtocoloAdmView(CreateView):
     template_name = 'protocoloadm/anular_protocoloadm.html'
     form_class = AnularProcoloAdmForm
-    success_url = reverse_lazy('anular_protocolo')
     form_valid_message = _('Protocolo anulado com sucesso!')
 
+    def get_success_url(self):
+        return reverse('protocolo')
+
     def get_initial(self):
         initial_data = {}
         initial_data['user_anulacao'] = self.request.user.username
         initial_data['ip_anulacao'] = get_client_ip(self.request)
-        initial_data['anulado'] = True
         return initial_data
 
-    def post(self, request, *args, **kwargs):
+    def form_valid(self, form):
-
+        protocolo = Protocolo.objects.get(numero=form.cleaned_data['numero'],
-        form = AnularProcoloAdmForm(request.POST)
+                                          ano=form.cleaned_data['ano'])
-
+        protocolo.anulado = True
-        if form.is_valid():
+        protocolo.justificativa_anulacao = (
-
+            form.cleaned_data['justificativa_anulacao'])
-            numero = form.cleaned_data['numero']
+        protocolo.user_anulacao = form.cleaned_data['user_anulacao']
-            ano = form.cleaned_data['ano']
+        protocolo.ip_anulacao = form.cleaned_data['ip_anulacao']
+        protocolo.save()
+        return redirect(self.get_success_url())
 
-            protocolo = Protocolo.objects.get(numero=numero, ano=ano)
-            protocolo.anulado = True
-            protocolo.justificativa_anulacao = sub(' ', ' ', strip_tags(
-                form.cleaned_data['justificativa_anulacao']))
-            protocolo.user_anulacao = form.cleaned_data['user_anulacao']
-            protocolo.ip_anulacao = form.cleaned_data['ip_anulacao']
-            protocolo.save()
-
-            return self.form_valid(form)
-        else:
-            return self.form_invalid(form)
-
-
-class ProtocoloDocumentoView(FormValidMessageMixin, FormView):
 
+class ProtocoloDocumentoView(FormValidMessageMixin, CreateView):
     template_name = "protocoloadm/protocolar_documento.html"
     form_class = ProtocoloDocumentForm
-    success_url = reverse_lazy('protocolo')
     form_valid_message = _('Protocolo cadastrado com sucesso!')
 
-    def post(self, request, *args, **kwargs):
+    def get_success_url(self):
+        return reverse('protocolo')
 
-        form = ProtocoloDocumentForm(request.POST)
+    def form_valid(self, form):
+        f = form.save(commit=False)
 
-        if form.is_valid():
+        if form.cleaned_data['numeracao'] == '1':
-            if form.cleaned_data['numeracao'] == '1':
+            numeracao = Protocolo.objects.filter(
-                numeracao = Protocolo.objects.filter(
+                ano=date.today().year).aggregate(Max('numero'))
-                    ano=date.today().year).aggregate(Max('numero'))
+        elif form.cleaned_data['numeracao'] == '2':
-            elif form.cleaned_data['numeracao'] == '2':
+            numeracao = Protocolo.objects.all().aggregate(Max('numero'))
-                numeracao = Protocolo.objects.all().aggregate(Max('numero'))
-            # else:
-            #     raise ValidationError(_("Campo numeração é obrigatório"))
-
-            if numeracao['numero__max'] is None:
-                numeracao['numero__max'] = 0
-
-            protocolo = form.save(commit=False)
-            protocolo.numero = numeracao['numero__max'] + 1
-            protocolo.ano = datetime.now().year
-            protocolo.data = datetime.now().strftime("%Y-%m-%d")
-            protocolo.hora = datetime.now().strftime("%H:%M")
-            protocolo.timestamp = datetime.now().strftime("%Y-%m-%d %H:%M")
-            protocolo.tipo_protocolo = request.POST['tipo_protocolo']
-            protocolo.tipo_processo = '0'  # TODO validar o significado
-            protocolo.interessado = request.POST['interessado']
-            protocolo.anulado = False
-            protocolo.tipo_documento = TipoDocumentoAdministrativo.objects.get(
-                id=request.POST['tipo_documento'])
-            protocolo.assunto_ementa = sub(
-                ' ', ' ', strip_tags(request.POST['assunto']))
-            protocolo.numero_paginas = request.POST['num_paginas']
-            protocolo.observacao = sub(
-                ' ', ' ', strip_tags(request.POST['observacao']))
-
-            protocolo.save()
 
-            return self.form_valid(form)
+        if numeracao['numero__max'] is None:
-        else:
+            numeracao['numero__max'] = 0
-            return self.form_invalid(form)
+
+        f.tipo_processo = '0'  # TODO validar o significado
+        f.anulado = False
+        f.numero = numeracao['numero__max'] + 1
+        f.ano = datetime.now().year
+        f.data = datetime.now().strftime('%Y-%m-%d')
+        f.hora = datetime.now().strftime('%H:%M')
+        f.timestamp = datetime.now().strftime("%Y-%m-%d %H:%M")
+
+        f.save()
+        return redirect(self.get_success_url())
 
 
 class CriarDocumentoProtocolo(CreateView):
@@ -274,7 +249,7 @@ class ComprovanteProtocoloView(TemplateView):
         return context
 
 
-class ProtocoloMateriaView(FormView):
+class ProtocoloMateriaView(CreateView):
 
     template_name = "protocoloadm/protocolar_materia.html"
     form_class = ProtocoloMateriaForm
@@ -283,40 +258,34 @@ class ProtocoloMateriaView(FormView):
     def get_success_url(self):
         return reverse('protocolo')
 
-    def post(self, request, *args, **kwargs):
+    def form_valid(self, form):
-
+        if self.request.POST['numeracao'] == '1':
-        form = ProtocoloMateriaForm(request.POST)
+            numeracao = Protocolo.objects.filter(
-
+                ano=date.today().year).aggregate(Max('numero'))
-        if form.is_valid():
-            if request.POST['numeracao'] == '1':
-                numeracao = Protocolo.objects.filter(
-                    ano=date.today().year).aggregate(Max('numero'))
-            else:
-                numeracao = Protocolo.objects.all().aggregate(Max('numero'))
-
-            if numeracao is None:
-                numeracao['numero__max'] = 0
-
-            protocolo = Protocolo()
-
-            protocolo.numero = numeracao['numero__max'] + 1
-            protocolo.ano = datetime.now().year
-            protocolo.data = datetime.now().strftime("%Y-%m-%d")
-            protocolo.hora = datetime.now().strftime("%H:%M")
-            protocolo.timestamp = datetime.now().strftime("%Y-%m-%d %H:%M")
-            protocolo.tipo_protocolo = request.POST['tipo_protocolo']
-            protocolo.tipo_processo = '0'  # TODO validar o significado
-            protocolo.autor = Autor.objects.get(id=request.POST['autor'])
-            protocolo.anulado = False
-            protocolo.tipo_materia = TipoMateriaLegislativa.objects.get(
-                id=request.POST['tipo_materia'])
-            protocolo.numero_paginas = request.POST['num_paginas']
-            protocolo.observacao = sub(
-                ' ', ' ', strip_tags(request.POST['observacao']))
-            protocolo.save()
-            return self.form_valid(form)
         else:
-            return self.form_invalid(form)
+            numeracao = Protocolo.objects.all().aggregate(Max('numero'))
+
+        if numeracao is None:
+            numeracao['numero__max'] = 0
+
+        protocolo = Protocolo()
+
+        protocolo.numero = numeracao['numero__max'] + 1
+        protocolo.ano = datetime.now().year
+        protocolo.data = datetime.now().strftime("%Y-%m-%d")
+        protocolo.hora = datetime.now().strftime("%H:%M")
+        protocolo.timestamp = datetime.now().strftime("%Y-%m-%d %H:%M")
+        protocolo.tipo_protocolo = self.request.POST['tipo_protocolo']
+        protocolo.tipo_processo = '0'  # TODO validar o significado
+        if form.cleaned_data['autor']:
+            protocolo.autor = form.cleaned_data['autor']
+        protocolo.anulado = False
+        protocolo.tipo_materia = TipoMateriaLegislativa.objects.get(
+            id=self.request.POST['tipo_materia'])
+        protocolo.numero_paginas = self.request.POST['numero_paginas']
+        protocolo.observacao = self.request.POST['observacao']
+        protocolo.save()
+        return redirect(self.get_success_url())
 
 
 # TODO: move to Proposicao app