forms.py
1.58 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# coding=utf-8
from django import forms
from django.utils.translation import ugettext_lazy as _
from django.utils.html import strip_tags
from .models import GeneralPost, Comment
class Validation(forms.ModelForm):
MAX_UPLOAD_SIZE = 5*1024*1024
def clean_post(self):
post = self.cleaned_data.get('post', '')
cleaned_post = strip_tags(post)
if cleaned_post == '':
self._errors['post'] = [_('This field is required.')]
return ValueError
return post
def clean_image(self):
image = self.cleaned_data.get('image', False)
if image:
if hasattr(image, '_size'):
if image._size > self.MAX_UPLOAD_SIZE:
self._errors['image'] = [_("The image is too large. It should have less than 5MB.")]
return ValueError
return image
class GeneralPostForm(Validation):
class Meta:
model = GeneralPost
fields = ['action', 'post', 'image']
widgets = {
'action': forms.RadioSelect,
'post': forms.Textarea
}
class CommentForm(forms.ModelForm):
MAX_UPLOAD_SIZE = 5*1024*1024
def clean_comment(self):
comment = self.cleaned_data.get('comment', '')
cleaned_comment = strip_tags(comment)
if cleaned_comment == '':
self._errors['comment'] = [_('This field is required.')]
return ValueError
return comment
def clean_image(self):
image = self.cleaned_data.get('image', False)
if image:
if hasattr(image, '_size'):
if image._size > self.MAX_UPLOAD_SIZE:
self._errors['image'] = [_("The image is too large. It should have less than 5MB.")]
return ValueError
return image
class Meta:
model = Comment
fields = ['comment', 'image']