2 from django import forms
3 from django.core.validators import RegexValidator
4 from django.template.defaultfilters import capfirst
5 from pyqcg.utils import TaskStatus
7 from qcg.fields import TimeRangeField
8 from qcg.models import Task, Allocation, JobTemplate
11 date_range_validator = RegexValidator(r'[0-9]{2}\.[0-9]{2}\.[0-9]{4} - [0-9]{2}\.[0-9]{2}\.[0-9]{4}')
12 nodes_validator = RegexValidator(r'^[0-9]{1,3}:[0-9]{1,2}(:[0-9]{1,2})?$')
13 env_name_validator = RegexValidator(r'^[a-zA-Z_][a-zA-Z0-9_]*$')
15 CHOICES_PLACEHOLDER = (None, '')
18 class FiltersForm(forms.Form):
19 ACTIVE, FINISHED, FAILED = range(3)
22 (FINISHED, u"Zakończone"),
23 (FAILED, u"Niepowodzenia"),
28 Task.STATUS_CHOICES_REVERSED[TaskStatus.UNSUBMITTED],
29 Task.STATUS_CHOICES_REVERSED[TaskStatus.UNCOMMITTED],
30 Task.STATUS_CHOICES_REVERSED[TaskStatus.QUEUED],
31 Task.STATUS_CHOICES_REVERSED[TaskStatus.PREPROCESSING],
32 Task.STATUS_CHOICES_REVERSED[TaskStatus.PENDING],
33 Task.STATUS_CHOICES_REVERSED[TaskStatus.RUNNING],
34 Task.STATUS_CHOICES_REVERSED[TaskStatus.STOPPED],
35 Task.STATUS_CHOICES_REVERSED[TaskStatus.POSTPROCESSING],
38 Task.STATUS_CHOICES_REVERSED[TaskStatus.FINISHED],
41 Task.STATUS_CHOICES_REVERSED[TaskStatus.FAILED],
42 Task.STATUS_CHOICES_REVERSED[TaskStatus.CANCELED],
45 STATUS_CHOICES_DICT = dict(STATUS_CHOICES)
47 keywords = forms.CharField(max_length=100, label=u"Wyszukaj frazę", required=False)
48 status = forms.MultipleChoiceField(choices=STATUS_CHOICES, label=u"Status", required=False,
49 widget=forms.CheckboxSelectMultiple)
52 host = forms.MultipleChoiceField(label=u"Host", required=False, widget=forms.CheckboxSelectMultiple)
53 purged = forms.TypedChoiceField(label=u"Istniejący katalog roboczy?", required=False, coerce=lambda x: bool(int(x)),
54 choices=((0, 'Tak'), (1, 'Nie')), widget=forms.RadioSelect)
55 submission = forms.CharField(label=u"Data zlecenia", validators=[date_range_validator], required=False)
56 finish = forms.CharField(label=u"Data zakończenia", validators=[date_range_validator], required=False)
58 def __init__(self, *args, **kwargs):
59 super(FiltersForm, self).__init__(*args, **kwargs)
61 self.fields['host'].choices = tuple(
62 (host, capfirst(host.split('.', 1)[0]))
63 for host in Allocation.objects.values_list('host_name', flat=True).order_by('host_name').distinct())
66 class JobDescriptionForm(forms.Form):
67 class Process(object):
79 APPLICATION_CHOICES = (
81 ('unres-gab', 'UNRES GAB'),
82 ('unres-e0ll2y', 'UNRES E0LL2Y'),
89 ('plgrid-long', 'plgrid-long'),
90 ('plgrid-testing', 'plgrid-testing'),
95 ('mailto', u'E-mail'),
100 FORCE_FIELD_CHOICES = (
102 ('E0LL2Y', u'E0LL2Y'),
105 nstep = forms.IntegerField(label=u"NSTEP", help_text=u"Liczba kroków w trajektorii", min_value=1, initial=500000, required=False)
106 ntwe = forms.IntegerField(label=u"NTWE", help_text=u"Częstość zapisu energii w krokach", min_value=0, initial=100, required=False)
107 ntwx = forms.IntegerField(label=u"NTWX", help_text=u"Częstość zapisu współrzędnych w krokach", min_value=0, initial=1000, required=False)
108 dt = forms.DecimalField(label=u"DT", help_text=u"Krok czasowy. Wartość kroku równa jedności to 48.9 fs", max_digits=5, decimal_places=2, min_value=0.01, initial=0.1, required=False)
109 damax = forms.DecimalField(label=u"DAMAX", help_text=u"Maksymalna dopuszczalna zmiana przyspieszenia podczas jednego kroku czasowego", max_digits=5, decimal_places=2, min_value=0.01, initial=1.0, required=False)
110 force_field = forms.ChoiceField(choices=FORCE_FIELD_CHOICES, label=u"Pole siłowe", required=False, initial='GAB')
112 pdb_file = forms.CharField(label=u"Plik PDB", max_length=500, required=False)
113 #retmin = forms.IntegerField(label=u"RETMIN", help_text=u"Dolny zakres temparatur dla wymiany replik", min_value=2, initial=250, required=False)
114 #retmax = forms.IntegerField(label=u"RETMAX", help_text=u"Górny zakres temperatur dla wymiany replik", min_value=2, initial=450, required=False)
115 sequence = forms.CharField(label=u"Sekwencja", help_text=u"Sekwencja aminokwasów w zapisie jednoliterowym", widget=forms.Textarea(attrs={'rows': 2, 'cols': 40}), required=False)
118 application = forms.ChoiceField(choices=APPLICATION_CHOICES, label=u"Aplikacja", required=False, initial='unres-gab')
119 #master_file = forms.CharField(label=u"Plik główny", max_length=500, required=False)
121 executable = forms.CharField(label=u"Plik wykonywalny", max_length=500, required=False)
122 script = forms.CharField(label=u"Skrypt", widget=forms.Textarea(attrs={'rows': 2, 'cols': 40}), required=False)
123 arguments = forms.MultipleChoiceField(label=u"Argumenty", required=False)
124 note = forms.CharField(label=u"Opis", widget=forms.Textarea(attrs={'rows': 2, 'cols': 40}), required=False)
125 grant = forms.CharField(label=u"Grant", max_length=100, required=False)
127 hosts = forms.MultipleChoiceField(label=u"Host", required=False)
128 properties = forms.CharField(label=u"Właściwości węzłów", required=False)
129 queue = forms.ChoiceField(choices=QUEUE_CHOICES, label=u"Kolejka", required=False)
130 procs = forms.IntegerField(label=u"Liczba procesów", min_value=0, required=False)
131 nodes = forms.CharField(label=u"Topologia węzłów", max_length=10, validators=[nodes_validator], required=False)
132 wall_time = TimeRangeField(label=u"Wall time", required=False)
133 memory = forms.IntegerField(label=u"Pamięć (MB)", min_value=0, required=False)
134 memory_per_slot = forms.IntegerField(label=u"Pamięci per proces (MB)", min_value=0, required=False)
135 modules = forms.MultipleChoiceField(label=u"Moduły", required=False)
136 reservation = forms.CharField(label=u"Rezerwacja", max_length=100, required=False)
138 input = forms.CharField(label=u"Standardowe wejście", max_length=500, required=False)
139 stage_in = forms.MultipleChoiceField(label=u"Stage in", required=False)
141 monitoring = forms.BooleanField(label=u"Portal QCG-Monitoring", required=False)
142 notify_type = forms.ChoiceField(label=u"Monitorowanie stanu", choices=PROTOCOL_CHOICES, required=False, initial='',
143 widget=forms.RadioSelect)
144 notify_address = forms.EmailField(label=u"Adres", required=False)
145 watch_output_type = forms.ChoiceField(label=u"Monitorowanie wyjścia", choices=PROTOCOL_CHOICES, required=False,
146 initial='', widget=forms.RadioSelect)
147 watch_output_address = forms.EmailField(label=u"Adres", required=False)
148 watch_output_pattern = forms.CharField(label=u"Wzorzec", max_length=500, required=False)
150 preprocess_type = forms.ChoiceField(label=u"Preprocessing", choices=Process.CHOICES, required=False,
151 initial=Process.NONE, widget=forms.RadioSelect)
152 preprocess_cmd = forms.CharField(label=u"Polecenie", max_length=1000, required=False)
153 preprocess_script = forms.CharField(label=u"Skrypt", max_length=500, required=False)
154 postprocess_type = forms.ChoiceField(label=u"Postprocessing", choices=Process.CHOICES, required=False,
155 initial=Process.NONE, widget=forms.RadioSelect)
156 postprocess_cmd = forms.CharField(label=u"Polecenie", max_length=1000, required=False)
157 postprocess_script = forms.CharField(label=u"Skrypt", max_length=500, required=False)
158 native = forms.MultipleChoiceField(label=u"Opcje systemu kolejkowego", required=False)
159 persistent = forms.BooleanField(label=u"Trwałe", required=False)
161 def __init__(self, data=None, initial=None, hosts=(), applications=(), modules=(), *args, **kwargs):
162 super(JobDescriptionForm, self).__init__(data, initial=initial, *args, **kwargs)
164 self.fields['hosts'].choices = hosts
165 self.fields['application'].choices = applications
166 self.fields['modules'].choices = modules
169 self._init_user_choices('queue')
170 self._init_user_choices('arguments')
171 self._init_user_choices('native')
172 self._init_user_choices('stage_in')
173 self._init_user_choices('force_field')
176 self._init_user_choices('hosts')
177 self._init_user_choices('application')
178 self._init_user_choices('modules')
181 self.fields['monitoring'].widget.attrs['disabled'] = True
184 data = super(JobDescriptionForm, self).clean()
186 force_field = data.get('force_field')
188 if force_field == u'GAB':
189 data['application'] = [u'unres-gab']
191 data['application'] = [u'unres-e0ll2y']
193 if data['pdb_file']== None:
194 self.add_error('pdb_file', u"Należy podać plik PDB.")
196 if data['sequence']== '':
197 self.add_error('sequence', u"Należy podać sekwencję aminokwasów.")
199 if data['procs'] and data['nodes']:
200 self.add_error(None, u"Zdefiniuj tylko jedno z pól: liczbę procesów lub topologię węzłów")
202 notify_type = data.get('notify_type')
203 data['notify'] = u'{}:{}'.format(notify_type, data['notify_address']) if notify_type else ''
205 wo_type = data.get('watch_output_type')
206 data['watch_output'] = u'{}:{}'.format(wo_type, data['watch_output_address']) if wo_type else ''
208 preprocess_type = data.get('preprocess_type')
209 if preprocess_type == self.Process.CMD:
210 data['preprocess'] = data['preprocess_cmd']
211 elif preprocess_type == self.Process.SCRIPT:
212 data['preprocess'] = data['preprocess_script']
214 data['preprocess'] = ''
216 postprocess_type = data.get('postprocess_type')
217 if postprocess_type == self.Process.CMD:
218 data['postprocess'] = data['postprocess_cmd']
219 elif postprocess_type == self.Process.SCRIPT:
220 data['postprocess'] = data['postprocess_script']
222 data['postprocess'] = ''
226 def clean_application(self):
227 return self.cleaned_data['application'].split('/', 1) if self.cleaned_data['application'] else ''
229 def clean_nodes(self):
230 return map(int, self.cleaned_data['nodes'].split(':', 2)) if self.cleaned_data['nodes'] else ''
232 def clean_executable(self):
233 return self._gsiftp_suffix(self.cleaned_data['executable'])
235 def clean_master_file(self):
236 return self._gsiftp_suffix(self.cleaned_data['master_file'])
238 def clean_input(self):
239 return self._gsiftp_suffix(self.cleaned_data['input'])
241 def clean_stage_in(self):
242 return [self._gsiftp_suffix(item) for item in self.cleaned_data['stage_in']]
244 def clean_preprocess_script(self):
245 return self._gsiftp_suffix(self.cleaned_data['preprocess_script'])
247 def clean_postprocess_script(self):
248 return self._gsiftp_suffix(self.cleaned_data['postprocess_script'])
250 def clean_pdb_file(self):
251 return self._gsiftp_suffix(self.cleaned_data['pdb_file'])
253 def clean_sequence(self):
254 return self.cleaned_data['sequence'].strip(' \t\n\r')
257 def _gsiftp_suffix(url):
259 return url if url.startswith('gsiftp://') else 'gsiftp://' + url
261 def _init_user_choices(self, name):
262 initial = self.initial.get(name)
263 choices = self.data.getlist(name)[:] if self.data else []
266 choices += initial if isinstance(initial, list) else [initial]
269 self.fields[name].choices += ((v, v) for v in choices)
272 class EnvForm(forms.Form):
273 name = forms.CharField(label=u"Nazwa", max_length=100, validators=[env_name_validator],
274 widget=forms.TextInput(attrs={'placeholder': u'Nazwa'}))
275 value = forms.CharField(label=u"Wartość", max_length=500,
276 widget=forms.TextInput(attrs={'placeholder': u'Wartość'}))
279 EnvFormSet = forms.formset_factory(EnvForm, can_delete=True, extra=0)
282 class ColumnsForm(forms.Form):
283 JOB_ID, DESCRIPTION, SUBMISSION, START, END, STATUS, HOST = range(7)
285 (JOB_ID, u"Identyfikator zadania"),
286 (DESCRIPTION, u"Opis"),
287 (SUBMISSION, u"Wysłane"),
294 columns = forms.MultipleChoiceField(choices=COLUMNS_CHOICES, initial=[k for k, v in COLUMNS_CHOICES[1:]],
295 label=u"Kolumny", required=False, widget=forms.CheckboxSelectMultiple)
298 class JobTemplateForm(forms.ModelForm):