forms.py
5.09 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
# coding=utf-8
from django import forms
from django.utils.translation import ugettext_lazy as _
from django.utils.html import strip_tags
from django.forms.models import inlineformset_factory
from subjects.models import Tag
from .models import Bulletin
from resubmit.widgets import ResubmitFileWidget
from pendencies.forms import PendenciesLimitedForm
from pendencies.models import Pendencies
class BulletinForm(forms.ModelForm):
subject = None
MAX_UPLOAD_SIZE = 1024*1024
def __init__(self, *args, **kwargs):
super(BulletinForm, self).__init__(*args, **kwargs)
self.subject = kwargs['initial'].get('subject', None)
if self.instance.id:
self.subject = self.instance.topic.subject
self.initial['tags'] = ", ".join(self.instance.tags.all().values_list("name", flat = True))
tags = forms.CharField(label = _('Tags'), required = False)
class Meta:
model = Bulletin
fields = ['name', 'content', 'brief_description', 'show_window', 'visible','file_content','indicators']
labels = {
'name': _('Bulletin name'),
'content': _('Bulletin content'),
}
widgets = {
'content': forms.Textarea,
'brief_description': forms.Textarea,
'file_content': ResubmitFileWidget(attrs={'accept':'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet,application/vnd.ms-excel,application/vnd.oasis.opendocument.spreadsheet,text/csv'}),
'indicators': ResubmitFileWidget(attrs={'accept':'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet,application/vnd.ms-excel,application/vnd.oasis.opendocument.spreadsheet,text/csv'}),
}
def clean_name(self):
name = self.cleaned_data.get('name', '')
topics = self.subject.topic_subject.all()
for topic in topics:
if self.instance.id:
same_name = topic.resource_topic.filter(name__unaccent__iexact = name).exclude(id = self.instance.id).count()
else:
same_name = topic.resource_topic.filter(name__unaccent__iexact = name).count()
if same_name > 0:
self._errors['name'] = [_('This subject already has a bulletin with this name')]
return ValueError
return name
def clean_file_content(self):
file_content = self.cleaned_data.get('file_content', False)
if file_content:
if hasattr(file_content, '_size'):
if file_content._size > self.MAX_UPLOAD_SIZE:
self._errors['file_content'] = [_("The file is too large. It should have less than 1MB.")]
return ValueError
elif not self.instance.pk:
self._errors['file_content'] = [_('This field is required.')]
return ValueError
return file_content
def clean_indicators(self):
indicators = self.cleaned_data.get('indicators', False)
if indicators:
if hasattr(indicators, '_size'):
if indicators._size > self.MAX_UPLOAD_SIZE:
self._errors['indicators'] = [_("The file is too large. It should have less than 1MB.")]
return ValueError
elif not self.instance.pk:
self._errors['indicators'] = [_('This field is required.')]
return ValueError
return indicators
def save(self, commit = True):
super(BulletinForm, self).save(commit = True)
self.instance.save()
previous_tags = self.instance.tags.all()
tags = self.cleaned_data['tags'].split(",")
#Excluding unwanted tags
for prev in previous_tags:
if not prev.name in tags:
self.instance.tags.remove(prev)
for tag in tags:
tag = tag.strip()
exist = Tag.objects.filter(name = tag).exists()
if exist:
new_tag = Tag.objects.get(name = tag)
else:
new_tag = Tag.objects.create(name = tag)
if not new_tag in self.instance.tags.all():
self.instance.tags.add(new_tag)
return self.instance
class FormModalMessage(forms.Form):
MAX_UPLOAD_SIZE = 5*1024*1024
comment = forms.CharField(widget=forms.Textarea,label=_("Message"))
image = forms.FileField(widget=ResubmitFileWidget(attrs={'accept':'image/*'}),required=False)
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
InlinePendenciesFormset = inlineformset_factory(Bulletin, Pendencies, form = PendenciesLimitedForm, extra = 1, max_num = 3, validate_max = True, can_delete = True)