qcg/forms.py

   1 # coding=utf-8
   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
   6
   7 from qcg.models import Task, Allocation
   8
   9
  10 date_range_validator = RegexValidator(r'[0-9]{2}\.[0-9]{2}\.[0-9]{4} - [0-9]{2}\.[0-9]{2}\.[0-9]{4}')
  11 nodes_validator = RegexValidator(r'^[0-9]{1,3}(:[0-9]{1,2}){0,2}$')
  12 env_name_validator = RegexValidator(r'^[a-zA-Z_][a-zA-Z0-9_]*$')
  13
  14 CHOICES_PLACEHOLDER = (None, '')
  15
  16
  17 class FiltersForm(forms.Form):
  18     ACTIVE, FINISHED, FAILED = range(3)
  19     STATUS_CHOICES = (
  20         (ACTIVE, u"Aktywne"),
  21         (FINISHED, u"Zakończone"),
  22         (FAILED, u"Niepowodzenia"),
  23     )
  24
  25     STATUS_MAP = {
  26         ACTIVE: (
  27             Task.STATUS_CHOICES_REVERSED[TaskStatus.UNSUBMITTED],
  28             Task.STATUS_CHOICES_REVERSED[TaskStatus.UNCOMMITTED],
  29             Task.STATUS_CHOICES_REVERSED[TaskStatus.QUEUED],
  30             Task.STATUS_CHOICES_REVERSED[TaskStatus.PREPROCESSING],
  31             Task.STATUS_CHOICES_REVERSED[TaskStatus.PENDING],
  32             Task.STATUS_CHOICES_REVERSED[TaskStatus.RUNNING],
  33             Task.STATUS_CHOICES_REVERSED[TaskStatus.STOPPED],
  34             Task.STATUS_CHOICES_REVERSED[TaskStatus.POSTPROCESSING],
  35         ),
  36         FINISHED: (
  37             Task.STATUS_CHOICES_REVERSED[TaskStatus.FINISHED],
  38         ),
  39         FAILED: (
  40             Task.STATUS_CHOICES_REVERSED[TaskStatus.FAILED],
  41             Task.STATUS_CHOICES_REVERSED[TaskStatus.CANCELED],
  42         ),
  43     }
  44     STATUS_CHOICES_DICT = dict(STATUS_CHOICES)
  45
  46     status = forms.MultipleChoiceField(choices=STATUS_CHOICES, label=u"Status", required=False,
  47                                        widget=forms.CheckboxSelectMultiple)
  48     host = forms.MultipleChoiceField(label=u"Host", required=False, widget=forms.CheckboxSelectMultiple)
  49
  50     # advanced
  51     keywords = forms.CharField(max_length=100, label=u"Wyszukaj frazę", required=False)
  52     submission = forms.CharField(label=u"Data zlecenia", validators=[date_range_validator], required=False)
  53     finish = forms.CharField(label=u"Data zakończenia", validators=[date_range_validator], required=False)
  54
  55     def __init__(self, *args, **kwargs):
  56         super(FiltersForm, self).__init__(*args, **kwargs)
  57
  58         self.fields['host'].choices = tuple(
  59             (host, capfirst(host.split('.', 1)[0]))
  60             for host in Allocation.objects.values_list('host_name', flat=True).order_by('host_name').distinct())
  61
  62
  63 class JobDescriptionForm(forms.Form):
  64     class Host(object):
  65         GALERA = 'galera.task.gda.pl'
  66         HYDRA = 'hydra.icm.edu.pl'
  67         INULA = 'inula.man.poznan.pl'
  68         MOSS = 'moss.man.poznan.pl'
  69         NOVA = 'nova.wcss.wroc.pl'
  70         REEF = 'reef.man.poznan.pl'
  71         ZEUS = 'zeus.cyfronet.pl'
  72
  73         CHOICES = (
  74             CHOICES_PLACEHOLDER,
  75             (GALERA, u'Galera'),
  76             (HYDRA, u'Hydra'),
  77             (INULA, u'Inula'),
  78             (MOSS, u'Moss'),
  79             (NOVA, u'Supernova'),
  80             (REEF, u'Reef'),
  81             (ZEUS, u'Zeus'),
  82         )
  83     APPLICATION_CHOICES = (
  84         CHOICES_PLACEHOLDER,
  85         ('bash', 'BASH'),
  86         ('gromacs/4.6.3', 'GROMACS 4.6.3'),
  87         ('matlab', 'MATLAB'),
  88         ('python', 'Python'),
  89     )
  90     QUEUE_CHOICES = (
  91         CHOICES_PLACEHOLDER,
  92         ('plgid', 'plgrid'),
  93         ('plgid-long', 'plgrid-long'),
  94         ('plgid-testing', 'plgrid-testing'),
  95     )
  96     MODULES_CHOICES = (
  97         ('plgrid/apps/python', 'plgrid/apps/python'),
  98         ('plgrid/apps/matlab', 'plgrid/apps/matlab'),
  99     )
 100     NOTIFY_CHOICES = (
 101         (0, u'Brak'),
 102         (1, u'E-mail'),
 103         (2, u'XMPP'),
 104     )
 105     PROCESS_CHOICES = (
 106         (0, u'Brak'),
 107         (1, u'Polecenie'),
 108         (2, u'Skrypt'),
 109     )
 110
 111     application = forms.ChoiceField(choices=APPLICATION_CHOICES, label=u"Aplikacja", required=False)  # TODO choices
 112     executable = forms.CharField(label=u"Plik wykonywalny", max_length=500, required=False)  # TODO grid ftp
 113     script = forms.CharField(label=u"Skrypt", widget=forms.Textarea(attrs={'rows': 2, 'cols': 40}), required=False)  # TODO saving to grid ftp
 114     arguments = forms.MultipleChoiceField(label=u"Argumenty", required=False)
 115     note = forms.CharField(label=u"Opis", widget=forms.Textarea(attrs={'rows': 2, 'cols': 40}), required=False)
 116     grant = forms.CharField(label=u"Grant", max_length=100, required=False)
 117
 118     host = forms.ChoiceField(label=u"Host", choices=Host.CHOICES, required=False)
 119     properties = forms.MultipleChoiceField(label=u"Właściwości węzłów", required=False)
 120     queue = forms.ChoiceField(choices=QUEUE_CHOICES, label=u"Kolejka", required=False)
 121     procs = forms.IntegerField(label=u"Liczba procesów", min_value=0, required=False)
 122     nodes = forms.CharField(label=u"Topologia węzłów", max_length=10, validators=[nodes_validator], required=False)
 123     wall_time = forms.IntegerField(label=u"Wall time (s)", min_value=0, required=False)  # TODO duration field
 124     memory = forms.IntegerField(label=u"Pamięć (MB)", min_value=0, required=False)
 125     memory_per_slot = forms.IntegerField(label=u"Pamięci per proces (MB)", min_value=0, required=False)
 126     modules = forms.MultipleChoiceField(label=u"Moduły", choices=MODULES_CHOICES, required=False)  # TODO choices
 127     reservation = forms.CharField(label=u"Rezerwacja", max_length=100, required=False)
 128
 129     # TODO grid ftp
 130     input = forms.CharField(label=u"Standardowe wejście", max_length=500, required=False)
 131     stage_in = forms.CharField(label=u"Stage in", max_length=500, required=False)
 132     stage_out = forms.CharField(label=u"Stage out", max_length=500, required=False)
 133
 134     monitoring = forms.BooleanField(label=u"Portal QCG-Monitoring", required=False)
 135     notify_type = forms.ChoiceField(label=u"Monitorowanie stanu", choices=NOTIFY_CHOICES, required=False, initial=0,
 136                                     widget=forms.RadioSelect)
 137     notify_address = forms.EmailField(label=u"Adres", required=False)
 138     watch_output_type = forms.ChoiceField(label=u"Monitorowanie wyjścia", choices=NOTIFY_CHOICES, required=False,
 139                                           initial=0, widget=forms.RadioSelect)
 140     watch_output_address = forms.EmailField(label=u"Adres", required=False)
 141     watch_output_pattern = forms.CharField(label=u"Wzorzec", max_length=500, required=False)
 142
 143     preprocess_type = forms.ChoiceField(label=u"Preprocessing", choices=PROCESS_CHOICES, required=False, initial=0,
 144                                         widget=forms.RadioSelect)
 145     preprocess_cmd = forms.CharField(label=u"Polecenie", max_length=1000, required=False)
 146     preprocess_script = forms.CharField(label=u"Skrypt", max_length=500, required=False)  # TODO grid ftp
 147     postprocess_type = forms.ChoiceField(label=u"Postprocessing", choices=PROCESS_CHOICES, required=False, initial=0,
 148                                          widget=forms.RadioSelect)
 149     postprocess_cmd = forms.CharField(label=u"Polecenie", max_length=1000, required=False)
 150     postprocess_script = forms.CharField(label=u"Skrypt", max_length=500, required=False)  # TODO grid ftp
 151     native = forms.MultipleChoiceField(label=u"Opcje systemu kolejkowego", required=False)
 152     persistent = forms.BooleanField(label=u"Trwałe", required=False)
 153
 154     def __init__(self, data=None, *args, **kwargs):
 155         super(JobDescriptionForm, self).__init__(data, *args, **kwargs)
 156
 157         if data is not None:
 158             # accept user defined choices
 159             self.fields['queue'].choices += ((data.get('queue'), data.get('queue')), )
 160             self.fields['arguments'].choices += ((v, v) for v in data.getlist('arguments'))
 161             self.fields['properties'].choices += ((v, v) for v in data.getlist('properties'))
 162             self.fields['native'].choices += ((v, v) for v in data.getlist('native'))
 163
 164
 165 class EnvForm(forms.Form):
 166     name = forms.CharField(label=u"Nazwa", max_length=100, validators=[env_name_validator],
 167                            widget=forms.TextInput(attrs={'placeholder': u'Nazwa'}))
 168     value = forms.CharField(label=u"Wartość", max_length=500,
 169                             widget=forms.TextInput(attrs={'placeholder': u'Wartość'}))
 170
 171
 172 EnvFormSet = forms.formset_factory(EnvForm, can_delete=True, extra=0)
 173
 174
 175 class ColumnsForm(forms.Form):
 176     JOB_ID, DESCRIPTION, SUBMISSION, START, END, STATUS, HOST = range(7)
 177     COLUMNS_CHOICES = (
 178         (JOB_ID, u"Identyfikator zadania"),
 179         (DESCRIPTION, u"Opis"),
 180         (SUBMISSION, u"Wysłane"),
 181         (START, u"Start"),
 182         (END, u"Koniec"),
 183         (STATUS, u"Status"),
 184         (HOST, u"Host"),
 185     )
 186
 187     columns = forms.MultipleChoiceField(choices=COLUMNS_CHOICES, initial=[k for k, v in COLUMNS_CHOICES[1:]],
 188                                         label=u"Kolumny", required=False, widget=forms.CheckboxSelectMultiple)