From 0066bc1c2475a69bc3733e373afc620fa52e174d Mon Sep 17 00:00:00 2001
From: Eduardo Edson Batista Cordeiro Alves <eduardoalves@interlegis.leg.br>
Date: Wed, 16 Mar 2016 10:03:29 -0300
Subject: [PATCH] Fix #220

---
 materia/forms.py | 19 +++++++++++++++++++
 norma/forms.py   | 10 ++++++++++
 sessao/forms.py  | 19 +++++++++++++++++++
 3 files changed, 48 insertions(+)

diff --git a/materia/forms.py b/materia/forms.py
index 6b260d58e..78f581d22 100644
--- a/materia/forms.py
+++ b/materia/forms.py
@@ -1,6 +1,7 @@
 from crispy_forms.helper import FormHelper
 from crispy_forms.layout import HTML, Button, Column, Fieldset, Layout, Submit
 from django import forms
+from django.core.exceptions import ValidationError
 from django.forms import ModelForm
 from django.utils.translation import ugettext_lazy as _
 
@@ -39,6 +40,15 @@ class ProposicaoForm(ModelForm):
     ano_materia = forms.CharField(
         label='Ano', required=False)
 
+    def clean_texto_original(self):
+        texto_original = self.cleaned_data.get('texto_original', False)
+        if texto_original:
+            if texto_original.size > 5*1024*1024:
+                raise ValidationError("Arquivo muito grande. ( > 5mb )")
+            return texto_original
+        else:
+            raise ValidationError("Não foi possível salvar o arquivo.")
+
     class Meta:
         model = Proposicao
         fields = ['tipo', 'data_envio', 'descricao', 'texto_original']
@@ -398,6 +408,15 @@ class FormularioSimplificadoForm(ModelForm):
             'data_apresentacao': forms.DateInput(attrs={'class': 'dateinput'}),
         }
 
+    def clean_texto_original(self):
+        texto_original = self.cleaned_data.get('texto_original', False)
+        if texto_original:
+            if texto_original.size > 5*1024*1024:
+                raise ValidationError("Arquivo muito grande. ( > 5mb )")
+            return texto_original
+        else:
+            raise ValidationError("Não foi possível salvar o arquivo.")
+
     def __init__(self, *args, **kwargs):
 
         row1 = crispy_layout_mixin.to_row(
diff --git a/norma/forms.py b/norma/forms.py
index c7565d382..4b4c850d3 100644
--- a/norma/forms.py
+++ b/norma/forms.py
@@ -1,6 +1,7 @@
 from crispy_forms.helper import FormHelper
 from crispy_forms.layout import Fieldset, Layout
 from django import forms
+from django.core.exceptions import ValidationError
 from django.forms import ModelForm
 from django.utils.safestring import mark_safe
 
@@ -99,6 +100,15 @@ class NormaJuridicaForm(ModelForm):
 
     ano_materia = forms.CharField(label='Ano', required=False)
 
+    def clean_texto_integral(self):
+        texto_integral = self.cleaned_data.get('texto_integral', False)
+        if texto_integral:
+            if texto_integral.size > 5*1024*1024:
+                raise ValidationError("Arquivo muito grande. ( > 5mb )")
+            return texto_integral
+        else:
+            raise ValidationError("Não foi possível salvar o arquivo.")
+
     class Meta:
         model = NormaJuridica
         fields = ['tipo',
diff --git a/sessao/forms.py b/sessao/forms.py
index 4b2367e52..79d8c458c 100644
--- a/sessao/forms.py
+++ b/sessao/forms.py
@@ -1,6 +1,7 @@
 from crispy_forms.helper import FormHelper
 from crispy_forms.layout import Fieldset, Layout
 from django import forms
+from django.core.exceptions import ValidationError
 from django.forms import ModelForm
 from django.utils.translation import ugettext_lazy as _
 
@@ -92,6 +93,24 @@ class SessaoForm(ModelForm):
             'hora_fim': forms.TextInput(attrs={'class': 'hora'}),
         }
 
+    def clean_url_audio(self):
+        url_audio = self.cleaned_data.get('url_audio', False)
+        if url_audio:
+            if url_audio.size > 5*1024*1024:
+                raise ValidationError("Arquivo muito grande. ( > 5mb )")
+            return url_audio
+        else:
+            raise ValidationError("Não foi possível salvar o arquivo.")
+
+    def clean_url_video(self):
+        url_video = self.cleaned_data.get('url_video', False)
+        if url_video:
+            if url_video.size > 5*1024*1024:
+                raise ValidationError("Arquivo muito grande. ( > 5mb )")
+            return url_video
+        else:
+            raise ValidationError("Não foi possível salvar o arquivo.")
+
     def __init__(self, *args, **kwargs):
 
         row1 = crispy_layout_mixin.to_row(