diff --git a/parlamentares/forms.py b/parlamentares/forms.py index 92bd3ae64..be49aab6a 100644 --- a/parlamentares/forms.py +++ b/parlamentares/forms.py @@ -1,6 +1,7 @@ from crispy_forms.helper import FormHelper from crispy_forms.layout import HTML, Fieldset, Layout, Submit from django import forms +from django.core.exceptions import ValidationError from django.forms import ModelForm from django.utils.safestring import mark_safe from django.utils.translation import ugettext_lazy as _ @@ -30,6 +31,15 @@ class ParlamentaresForm (ModelForm): ) ) + def clean_fotografia(self): + fotografia = self.cleaned_data.get('fotografia', False) + if fotografia: + if fotografia.size > 2*1024*1024: + raise ValidationError('Imagem muito grande. ( > 2mb )') + return fotografia + else: + raise ValidationError('Não foi possível salvar a imagem.') + class Meta: model = Parlamentar fields = ['nome_parlamentar', diff --git a/parlamentares/views.py b/parlamentares/views.py index deacb89b2..1a5ab7c80 100644 --- a/parlamentares/views.py +++ b/parlamentares/views.py @@ -185,6 +185,7 @@ class ParlamentaresCadastroView(CreateView): return context def form_valid(self, form): + import ipdb; ipdb.set_trace() form.save() mandato = Mandato() mandato.parlamentar = form.instance