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.fields import PredefinedChoiceField, MultiplePredefinedChoiceField
   8 from qcg.models import Task, Allocation
   9
  10
  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,2}$')
  13 env_name_validator = RegexValidator(r'^[a-zA-Z_][a-zA-Z0-9_]*$')
  14
  15 CHOICES_PLACEHOLDER = (None, '')
  16
  17
  18 class FiltersForm(forms.Form):
  19     ACTIVE, FINISHED, FAILED = range(3)
  20     STATUS_CHOICES = (
  21         (ACTIVE, u"Aktywne"),
  22         (FINISHED, u"Zakończone"),
  23         (FAILED, u"Niepowodzenia"),
  24     )
  25
  26     STATUS_MAP = {
  27         ACTIVE: (
  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],
  36         ),
  37         FINISHED: (
  38             Task.STATUS_CHOICES_REVERSED[TaskStatus.FINISHED],
  39         ),
  40         FAILED: (
  41             Task.STATUS_CHOICES_REVERSED[TaskStatus.FAILED],
  42             Task.STATUS_CHOICES_REVERSED[TaskStatus.CANCELED],
  43         ),
  44     }
  45     STATUS_CHOICES_DICT = dict(STATUS_CHOICES)
  46
  47     status = forms.MultipleChoiceField(choices=STATUS_CHOICES, label=u"Status", required=False,
  48                                        widget=forms.CheckboxSelectMultiple)
  49     host = forms.MultipleChoiceField(label=u"Host", required=False, widget=forms.CheckboxSelectMultiple)
  50
  51     # advanced
  52     keywords = forms.CharField(max_length=100, label=u"Wyszukaj frazę", required=False)
  53     submission = forms.CharField(label=u"Data zlecenia", validators=[date_range_validator], required=False)
  54     finish = forms.CharField(label=u"Data zakończenia", validators=[date_range_validator], required=False)
  55
  56     def __init__(self, *args, **kwargs):
  57         super(FiltersForm, self).__init__(*args, **kwargs)
  58
  59         self.fields['host'].choices = tuple(
  60             (host, capfirst(host.split('.', 1)[0]))
  61             for host in Allocation.objects.values_list('host_name', flat=True).order_by('host_name').distinct())
  62
  63
  64 class JobDescriptionForm(forms.Form):
  65     class Host(object):
  66         GALERA = 'galera.task.gda.pl'
  67         HYDRA = 'hydra.icm.edu.pl'
  68         INULA = 'inula.man.poznan.pl'
  69         MOSS = 'moss.man.poznan.pl'
  70         NOVA = 'nova.wcss.wroc.pl'
  71         REEF = 'reef.man.poznan.pl'
  72         ZEUS = 'zeus.cyfronet.pl'
  73
  74         CHOICES = (
  75             CHOICES_PLACEHOLDER,
  76             (GALERA, u'Galera'),
  77             (HYDRA, u'Hydra'),
  78             (INULA, u'Inula'),
  79             (MOSS, u'Moss'),
  80             (NOVA, u'Supernova'),
  81             (REEF, u'Reef'),
  82             (ZEUS, u'Zeus'),
  83         )
  84     APPLICATION_CHOICES = (
  85         CHOICES_PLACEHOLDER,
  86         ('bash', 'BASH'),
  87         ('python', 'Python'),
  88         ('matlab', 'MATLAB'),
  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"Polecenie", max_length=500, required=False)  # TODO grid ftp
 113     arguments = forms.CharField(label=u"Argumenty", max_length=1000, required=False)
 114
 115     name = forms.CharField(label=u"Nazwa", max_length=100, required=False)
 116     note = forms.CharField(label=u"Notatka", widget=forms.Textarea(attrs={'rows': 2, 'cols': 40}), required=False)
 117     grant = forms.CharField(label=u"Grant", max_length=100, required=False)
 118
 119     host = forms.ChoiceField(label=u"Host", choices=Host.CHOICES, required=False)
 120     queue = PredefinedChoiceField(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
 127     # TODO grid ftp
 128     input = forms.CharField(label=u"Standardowe wejście", max_length=500, required=False)
 129     output = forms.CharField(label=u"Standardowe wyjście", max_length=500, required=False)
 130     error = forms.CharField(label=u"Standardowe wyjście błędów", 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     properties = MultiplePredefinedChoiceField(label=u"Właściwości węzłów", required=False)
 135     modules = forms.MultipleChoiceField(label=u"Moduły", choices=MODULES_CHOICES, required=False)  # TODO choices
 136
 137     not_after = forms.DateTimeField(label=u"Nie później niż", required=False)
 138     not_before = forms.DateTimeField(label=u"Nie wcześniej niż", required=False)
 139     deadline = forms.IntegerField(label=u"Deadline (s)", min_value=0, required=False)  # TODO duration field
 140     reservation = forms.CharField(label=u"Rezerwacja", max_length=100, required=False)
 141
 142     monitoring = forms.BooleanField(label=u"Portal QCG-Monitoring", required=False)
 143     notify_type = forms.ChoiceField(label=u"Monitorowanie stanu", choices=NOTIFY_CHOICES, required=False, initial=0,
 144                                     widget=forms.RadioSelect)
 145     notify_address = forms.EmailField(label=u"Adres", required=False)
 146     watch_output_type = forms.ChoiceField(label=u"Monitorowanie wyjścia", choices=NOTIFY_CHOICES, required=False,
 147                                           initial=0, widget=forms.RadioSelect)
 148     watch_output_address = forms.EmailField(label=u"Adres", required=False)
 149     watch_output_pattern = forms.CharField(label=u"Wzorzec", max_length=500, required=False)
 150
 151     preprocess_type = forms.ChoiceField(label=u"Preprocessing", choices=PROCESS_CHOICES, required=False, initial=0,
 152                                         widget=forms.RadioSelect)
 153     preprocess_cmd = forms.CharField(label=u"Polecenie", max_length=1000, required=False)
 154     preprocess_script = forms.CharField(label=u"Skrypt", max_length=500, required=False)  # TODO grid ftp
 155     postprocess_type = forms.ChoiceField(label=u"Postprocessing", choices=PROCESS_CHOICES, required=False, initial=0,
 156                                          widget=forms.RadioSelect)
 157     postprocess_cmd = forms.CharField(label=u"Polecenie", max_length=1000, required=False)
 158     postprocess_script = forms.CharField(label=u"Skrypt", max_length=500, required=False)  # TODO grid ftp
 159     native = MultiplePredefinedChoiceField(label=u"Parametry natywne", required=False)
 160     persistent = forms.BooleanField(label=u"Trwałe", required=False)
 161     use_scratch = forms.BooleanField(label=u"Scratch", required=False)
 162
 163
 164 class EnvForm(forms.Form):
 165     name = forms.CharField(label=u"Nazwa", max_length=100, validators=[env_name_validator],
 166                            widget=forms.TextInput(attrs={'placeholder': u'Nazwa'}))
 167     value = forms.CharField(label=u"Wartość", max_length=500,
 168                             widget=forms.TextInput(attrs={'placeholder': u'Wartość'}))
 169
 170
 171 EnvFormSet = forms.formset_factory(EnvForm, can_delete=True, extra=0)
 172
 173
 174 class ColumnsForm(forms.Form):
 175     JOB_ID, DESCRIPTION, SUBMISSION, START, END, STATUS, HOST = range(7)
 176     COLUMNS_CHOICES = (
 177         (JOB_ID, u"Identyfikator zadania"),
 178         (DESCRIPTION, u"Opis"),
 179         (SUBMISSION, u"Wysłane"),
 180         (START, u"Start"),
 181         (END, u"Koniec"),
 182         (STATUS, u"Status"),
 183         (HOST, u"Host"),
 184     )
 185
 186     columns = forms.MultipleChoiceField(choices=COLUMNS_CHOICES, label=u"Kolumny", required=False,
 187                                         widget=forms.CheckboxSelectMultiple)