qcg/models.py

   1 # coding=utf-8
   2 from django.contrib.auth.models import AbstractUser
   3 from django.core.urlresolvers import reverse
   4 from django.db import models
   5 from django.utils.timezone import now
   6 from pyqcg.service import Job as QcgJob, Task as QcgTask
   7 from pyqcg.utils import JobStatus, TaskStatus, TaskType, AllocationType, EPRUtils
   8
   9 from qcg.utils import username_from_dn, get_attributes
  10
  11
  12 __all__ = ['User', 'Job', 'Task', 'Allocation', 'NodeInfo', 'JobTemplate']
  13
  14
  15 class User(AbstractUser):
  16     last_update = models.DateTimeField(default=now)
  17
  18     @property
  19     def tasks(self):
  20         return Task.objects.filter(job__owner=self)
  21
  22
  23 class Job(models.Model):
  24     STATUS_CHOICES = list(enumerate(field for field in dir(JobStatus) if not field.startswith('__')))
  25     STATUS_CHOICES_REVERSED = {v: k for k, v in STATUS_CHOICES}
  26
  27     _job = None
  28
  29     epr = models.TextField(u"Endpoint reference")
  30     job_id = models.TextField(u"Identyfikator", unique=True)
  31     status = models.IntegerField(u"Status", choices=STATUS_CHOICES)
  32     note = models.TextField(u"Notatka", blank=True, default='')
  33     description = models.TextField(u"Opis", blank=True, default='')
  34     project = models.TextField(u"Projekt", blank=True, default='')
  35     submission_time = models.DateTimeField(u"Data wysłania")
  36     finish_time = models.DateTimeField(u"Data zakończenia", blank=True, null=True)
  37     proxy_lifetime = models.DateTimeField(u"Czas życia proxy", blank=True, null=True)
  38     purged = models.BooleanField(u"Usunięty katalog roboczy?", default=False)
  39
  40     owner = models.ForeignKey(User, verbose_name=u"Właściciel", related_name='jobs')
  41     created = models.DateTimeField(u"Utworzono", auto_now_add=True)
  42     updated = models.DateTimeField(u"Uaktualniono", auto_now=True)
  43
  44     class Meta:
  45         # verbose_name = u"Zadanie"
  46         # verbose_name_plural = u"Zadania"
  47         ordering = ('-submission_time',)
  48
  49     def __unicode__(self):
  50         return self.job_id
  51
  52     def get_absolute_url(self):
  53         return reverse('job', kwargs={'job_id': self.job_id})
  54
  55     @property
  56     def qcg_job(self):
  57         if self._job is None:
  58             self._job = QcgJob(EPRUtils.deserialize_epr(self.epr))
  59
  60         return self._job
  61
  62     @staticmethod
  63     def qcg_map(qcg_job, user=None):
  64         attrs = get_attributes(qcg_job, ('job_id', 'note', 'description', 'submission_time', 'finish_time',
  65                                          'project', 'purged'))
  66
  67         attrs['epr'] = EPRUtils.serialize_epr(qcg_job.epr)
  68         attrs['status'] = Job.STATUS_CHOICES_REVERSED[qcg_job.status]
  69         attrs['proxy_lifetime'] = now() + qcg_job.proxy_lifetime
  70
  71         username = username_from_dn(qcg_job.user_dn)
  72         if user is not None:
  73             if user.username != username:
  74                 raise ValueError('Username does not match ({} vs. {})!'.format(repr(user.username), repr(username)))
  75             attrs['owner'] = user
  76         else:
  77             attrs['owner'] = User.objects.get(username=username)
  78
  79         return attrs
  80
  81     @property
  82     def terminated(self):
  83         return self.get_status_display() in [JobStatus.FINISHED, JobStatus.FAILED, JobStatus.CANCELED]
  84
  85
  86 class Task(models.Model):
  87     STATUS_CHOICES = list(enumerate(field for field in dir(TaskStatus) if not field.startswith('__')))
  88     STATUS_CHOICES_REVERSED = {v: k for k, v in STATUS_CHOICES}
  89     TYPE_CHOICES = list(enumerate(field for field in dir(TaskType) if not field.startswith('__')))
  90     TYPE_CHOICES_REVERSED = {v: k for k, v in TYPE_CHOICES}
  91
  92     _task = None
  93
  94     job = models.ForeignKey(Job, verbose_name=u"Zadanie", related_name='tasks')
  95
  96     epr = models.TextField(u"Endpoint reference")
  97     task_id = models.TextField(u"Identyfikator", db_index=True)
  98     status = models.IntegerField(u"Status", choices=STATUS_CHOICES)
  99     status_description = models.TextField(u"Opis statusu", blank=True, default='')
 100     type = models.IntegerField(u"Typ", choices=TYPE_CHOICES)
 101     note = models.TextField(u"Notatka", blank=True, default='')
 102     description = models.TextField(u"Opis", blank=True, default='')
 103     submission_time = models.DateTimeField(u"Data wysłania")
 104     start_time = models.DateTimeField(u"Data rozpoczęcia", blank=True, null=True)
 105     finish_time = models.DateTimeField(u"Data zakończenia", blank=True, null=True)
 106     reserved_time_start = models.DateTimeField(u"Data rozpoczęcia rezerwacji", blank=True, null=True)
 107     reserved_time_finish = models.DateTimeField(u"Data zakończenia rezerwacji", blank=True, null=True)
 108     proxy_lifetime = models.DateTimeField(u"Czas życia proxy", blank=True, null=True)
 109     purged = models.BooleanField(u"Usunięty katalog roboczy?", default=False)
 110
 111     created = models.DateTimeField(u"Utworzono", auto_now_add=True)
 112     updated = models.DateTimeField(u"Uaktualniono", auto_now=True)
 113
 114     class Meta:
 115         # verbose_name = u"Zadanie"
 116         # verbose_name_plural = u"Zadania"
 117         ordering = ('-submission_time',)
 118         unique_together = ('job', 'task_id')
 119
 120     def __unicode__(self):
 121         return u'{}/{}'.format(self.job, self.task_id)
 122
 123     def get_absolute_url(self):
 124         return reverse('task', kwargs={'job_id': self.job.job_id, 'task_id': self.task_id})
 125
 126     @property
 127     def qcg_task(self):
 128         if self._task is None:
 129             self._task = QcgTask(EPRUtils.deserialize_epr(self.epr))
 130
 131         return self._task
 132
 133     @staticmethod
 134     def qcg_map(qcg_task, job=None):
 135         attrs = get_attributes(qcg_task, ('task_id', 'status_description', 'note', 'description', 'submission_time',
 136                                           'start_time', 'finish_time', 'reserved_time_slot', 'purged'))
 137
 138         attrs['epr'] = EPRUtils.serialize_epr(qcg_task.epr)
 139         attrs['status'] = Task.STATUS_CHOICES_REVERSED[qcg_task.status]
 140         attrs['type'] = Task.TYPE_CHOICES_REVERSED[qcg_task.type]
 141         attrs['proxy_lifetime'] = now() + qcg_task.proxy_lifetime
 142
 143         if job is not None:
 144             if qcg_task.job_id != job.job_id:
 145                 raise ValueError('Job id does not match ({} vs. {})!'.format(repr(qcg_task.job_id), repr(job.job_id)))
 146             attrs['job'] = job
 147         else:
 148             attrs['job'] = Job.objects.get(job_id=qcg_task.job_id)
 149
 150         return attrs
 151
 152     @property
 153     def reserved_time_slot(self):
 154         if self.reserved_time_start or self.reserved_time_finish:
 155             return self.reserved_time_start, self.reserved_time_finish
 156
 157     @reserved_time_slot.setter
 158     def reserved_time_slot(self, value):
 159         if value is not None:
 160             self.reserved_time_start, self.reserved_time_finish = value
 161
 162     @property
 163     def short_host_names(self):
 164         return {alloc.host_name.split('.', 1)[0] for alloc in self.allocations.all()}
 165
 166     @property
 167     def terminated(self):
 168         return self.get_status_display() in [TaskStatus.FINISHED, TaskStatus.FAILED, TaskStatus.CANCELED]
 169
 170
 171 class Allocation(models.Model):
 172     STATUS_CHOICES = list(enumerate(field for field in dir(AllocationType) if not field.startswith('__')))
 173     STATUS_CHOICES_REVERSED = {v: k for k, v in STATUS_CHOICES}
 174
 175     task = models.ForeignKey(Task, verbose_name='Zadanie', related_name='allocations')
 176
 177     host_name = models.CharField(u"Host", max_length=100)
 178     status = models.IntegerField(u"Status", choices=STATUS_CHOICES)
 179     status_description = models.TextField(u"Opis statusu", blank=True, default='')
 180     processes_count = models.PositiveIntegerField(u"Liczba procesów")
 181     slots_count = models.PositiveIntegerField(u"Liczba rdzeni")
 182     processes_group_id = models.TextField(u"Identyfikator grupy procesów", blank=True, default='')
 183     submission_time = models.DateTimeField(u"Data wysłania")
 184     estimated_start_time = models.DateTimeField(u"Przewidywana data rozpoczęcia", blank=True, null=True)
 185     finish_time = models.DateTimeField(u"Data zakończenia", blank=True, null=True)
 186     local_submission_time = models.DateTimeField(u"Lokalna data wysłania", blank=True, null=True)
 187     local_start_time = models.DateTimeField(u"Lokalna data rozpoczęcia", blank=True, null=True)
 188     local_finish_time = models.DateTimeField(u"Lokalna data zakończenia", blank=True, null=True)
 189     purged = models.BooleanField(u"Usunięty katalog roboczy?", default=False)
 190     efficiency = models.IntegerField(u"Efektywność", blank=True, null=True)  # ??
 191     comment = models.TextField(u"Komentarz", blank=True, default='')
 192     memory_usage = models.PositiveIntegerField(u"Użycie pamięci", blank=True, null=True)
 193     working_directory = models.CharField(u"Katalog roboczy", max_length=1024, blank=True, default='')
 194
 195     class Meta:
 196         verbose_name = u"Alokacja"
 197         verbose_name_plural = u"Alokacje"
 198         ordering = ('-submission_time',)
 199
 200     def __unicode__(self):
 201         return u'{} @ {}'.format(self.task, self.host_name)
 202
 203     @staticmethod
 204     def qcg_map(qcg_allocation):
 205         attrs = get_attributes(qcg_allocation, ('host_name', 'status_description', 'processes_count', 'slots_count',
 206                                                 'processes_group_id', 'submission_time', 'estimated_start_time',
 207                                                 'finish_time', 'local_submission_time', 'local_start_time',
 208                                                 'local_finish_time', 'purged', 'efficiency', 'comment', 'memory_usage',
 209                                                 'working_directory'))
 210
 211         attrs['status'] = Allocation.STATUS_CHOICES_REVERSED[qcg_allocation.status]
 212
 213         return attrs
 214
 215
 216 class NodeInfo(models.Model):
 217     allocation = models.ForeignKey(Allocation, verbose_name=u"Alokacja", related_name='nodes')
 218
 219     name = models.CharField(u"Nazwa", max_length=32)
 220     count = models.PositiveSmallIntegerField(u"Liczba rdzeni", blank=True, null=True)
 221
 222     class Meta:
 223         verbose_name = u"Węzeł"
 224         verbose_name_plural = u"Węzły"
 225
 226     def __unicode__(self):
 227         result = u'{} - {}'.format(self.allocation, self.name)
 228
 229         if self.count is not None:
 230             result += u' ({})'.format(self.count)
 231
 232         return result
 233
 234     @staticmethod
 235     def qcg_map(qcg_node):
 236         return {'name': qcg_node.name, 'count': qcg_node.slots_count}
 237
 238
 239 class JobTemplate(models.Model):
 240     name = models.CharField(u"Nazwa", max_length=100)
 241     description = models.TextField(u"Opis")
 242
 243     owner = models.ForeignKey(User, verbose_name=u"Właściciel", related_name='templates')
 244     created = models.DateTimeField(u"Utworzono", auto_now_add=True)
 245     updated = models.DateTimeField(u"Uaktualniono", auto_now=True)
 246
 247     class Meta:
 248         verbose_name = u"Szablon zadania"
 249         verbose_name_plural = u"Szablony zadania"
 250
 251     def __unicode__(self):
 252         return u"{} ({})".format(self.name, self.owner)
 253
 254     def get_absolute_url(self):
 255         return reverse('template_submit', kwargs={'template_id': self.id})