models.py
294 lines
| 9.0 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r0 | from django.db import models | |
|
Juan C. Espinoza
|
r6 | from polymorphic import PolymorphicModel | |
|
Juan C. Espinoza
|
r0 | ||
Miguel Urco
|
r32 | from django.core.urlresolvers import reverse | |
|
Miguel Urco
|
r41 | CONF_STATES = ( | |
|
Miguel Urco
|
r47 | (0, 'Disconnected'), | |
| (1, 'Connected'), | |||
| (1, 'Running'), | |||
| ) | |||
|
Miguel Urco
|
r41 | ||
|
Miguel Urco
|
r21 | CONF_TYPES = ( | |
|
Miguel Urco
|
r47 | (0, 'Active'), | |
| (1, 'Historical'), | |||
| ) | |||
|
Miguel Urco
|
r16 | ||
| DEV_STATES = ( | |||
|
Miguel Urco
|
r47 | (0, 'No connected'), | |
| (1, 'Connected'), | |||
| (2, 'Configured'), | |||
| (3, 'Running'), | |||
| ) | |||
|
Juan C. Espinoza
|
r2 | ||
|
Miguel Urco
|
r13 | DEV_TYPES = ( | |
|
Miguel Urco
|
r47 | ('', 'Select a device type'), | |
| ('rc', 'Radar Controller'), | |||
| ('dds', 'Direct Digital Synthesizer'), | |||
| ('jars', 'Jicamarca Radar Acquisition System'), | |||
| ('usrp', 'Universal Software Radio Peripheral'), | |||
| ('cgs', 'Clock Generator System'), | |||
| ('abs', 'Automatic Beam Switching'), | |||
| ) | |||
| DEV_PORTS = { | |||
| 'rc' : 2000, | |||
| 'dds' : 2000, | |||
| 'jars' : 2000, | |||
| 'usrp' : 2000, | |||
| 'cgs' : 8080, | |||
| 'abs' : 8080 | |||
| } | |||
|
Juan C. Espinoza
|
r2 | ||
Fiorella Quino
|
r49 | RADAR_STATES = ( | |
| (0, 'No connected'), | |||
| (1, 'Connnected'), | |||
| (2, 'Configured'), | |||
| (3, 'Running'), | |||
| (4, 'Scheduled'), | |||
| ) | |||
|
Miguel Urco
|
r13 | # Create your models here. | |
|
Miguel Urco
|
r41 | ||
|
Miguel Urco
|
r53 | ||
| class Location(models.Model): | |||
| name = models.CharField(max_length = 30) | |||
| description = models.TextField(blank=True, null=True) | |||
| class Meta: | |||
| db_table = 'db_location' | |||
| def __unicode__(self): | |||
| return u'%s' % self.name | |||
|
Juan C. Espinoza
|
r2 | class DeviceType(models.Model): | |
|
Miguel Urco
|
r13 | name = models.CharField(max_length = 10, choices = DEV_TYPES, default = 'rc') | |
| description = models.TextField(blank=True, null=True) | |||
|
Juan C. Espinoza
|
r2 | ||
| class Meta: | |||
|
Miguel Urco
|
r13 | db_table = 'db_device_types' | |
|
Juan C. Espinoza
|
r2 | ||
| def __unicode__(self): | |||
|
Miguel Urco
|
r22 | return u'%s' % self.get_name_display() | |
|
Juan C. Espinoza
|
r2 | ||
| class Device(models.Model): | |||
|
Miguel Urco
|
r53 | device_type = models.ForeignKey(DeviceType, on_delete=models.CASCADE) | |
| location = models.ForeignKey(Location, on_delete=models.CASCADE) | |||
|
Miguel Urco
|
r41 | ||
|
Miguel Urco
|
r9 | name = models.CharField(max_length=40, default='') | |
|
Juan C. Espinoza
|
r6 | ip_address = models.GenericIPAddressField(protocol='IPv4', default='0.0.0.0') | |
|
Miguel Urco
|
r13 | port_address = models.PositiveSmallIntegerField(default=2000) | |
|
Miguel Urco
|
r9 | description = models.TextField(blank=True, null=True) | |
|
Miguel Urco
|
r16 | status = models.PositiveSmallIntegerField(default=0, choices=DEV_STATES) | |
|
Juan C. Espinoza
|
r2 | ||
| class Meta: | |||
|
Miguel Urco
|
r13 | db_table = 'db_devices' | |
|
Juan C. Espinoza
|
r2 | ||
| def __unicode__(self): | |||
|
Miguel Urco
|
r22 | return u'%s | %s' % (self.name, self.ip_address) | |
|
Miguel Urco
|
r47 | ||
| def get_status(self): | |||
| return self.status | |||
|
Miguel Urco
|
r13 | ||
| class Campaign(models.Model): | |||
|
Miguel Urco
|
r47 | template = models.BooleanField(default=False) | |
|
Miguel Urco
|
r13 | name = models.CharField(max_length=40, unique=True) | |
| start_date = models.DateTimeField(blank=True, null=True) | |||
| end_date = models.DateTimeField(blank=True, null=True) | |||
| tags = models.CharField(max_length=40) | |||
| description = models.TextField(blank=True, null=True) | |||
| class Meta: | |||
| db_table = 'db_campaigns' | |||
| def __unicode__(self): | |||
|
Miguel Urco
|
r22 | return u'%s' % (self.name) | |
|
Juan C. Espinoza
|
r2 | ||
|
Miguel Urco
|
r53 | # class Radar(models.Model): | |
| # | |||
| # # name = models.CharField(max_length = 30) | |||
| # experiment = models.ForeignKey('Experiment', on_delete=models.CASCADE) | |||
| # location = models.OneToOneField('Location', on_delete=models.CASCADE) | |||
| # status = models.PositiveSmallIntegerField(default=0, choices=RADAR_STATES) | |||
| # | |||
| # class Meta: | |||
| # db_table = 'db_radar' | |||
| # | |||
| # def __unicode__(self): | |||
| # return u'%s' % self.location | |||
|
Fiorella Quino
|
r48 | ||
|
Fiorella Quino
|
r49 | ||
|
Juan C. Espinoza
|
r2 | class Experiment(models.Model): | |
|
Miguel Urco
|
r47 | template = models.BooleanField(default=False) | |
|
Miguel Urco
|
r53 | campaign = models.ForeignKey('Campaign', null=True, blank=True, on_delete=models.CASCADE) | |
| location = models.ForeignKey('Location', null=True, blank=True, on_delete=models.CASCADE) | |||
|
Miguel Urco
|
r13 | name = models.CharField(max_length=40, default='') | |
| start_time = models.TimeField(default='00:00:00') | |||
| end_time = models.TimeField(default='23:59:59') | |||
|
Juan C. Espinoza
|
r2 | ||
| class Meta: | |||
|
Miguel Urco
|
r13 | db_table = 'db_experiments' | |
|
Juan C. Espinoza
|
r2 | ||
| def __unicode__(self): | |||
|
Miguel Urco
|
r53 | return u'%s' % (self.name) | |
|
Juan C. Espinoza
|
r6 | ||
| class Configuration(PolymorphicModel): | |||
|
Juan C. Espinoza
|
r2 | ||
|
Miguel Urco
|
r47 | template = models.BooleanField(default=False) | |
| name = models.CharField(verbose_name="Configuration Name", max_length=40, default='') | |||
|
Miguel Urco
|
r53 | experiment = models.ForeignKey('Experiment', null=True, blank=True, on_delete=models.CASCADE) | |
| device = models.ForeignKey(Device, on_delete=models.CASCADE) | |||
|
Miguel Urco
|
r41 | ||
|
Miguel Urco
|
r21 | type = models.PositiveSmallIntegerField(default=0, choices=CONF_TYPES) | |
|
Miguel Urco
|
r16 | ||
|
Miguel Urco
|
r22 | created_date = models.DateTimeField(auto_now_add=True) | |
| programmed_date = models.DateTimeField(auto_now=True) | |||
| parameters = models.TextField(default='{}') | |||
|
Miguel Urco
|
r53 | message = "" | |
|
Juan C. Espinoza
|
r2 | class Meta: | |
|
Miguel Urco
|
r13 | db_table = 'db_configurations' | |
|
Juan C. Espinoza
|
r2 | ||
| def __unicode__(self): | |||
|
Miguel Urco
|
r57 | return u'[%s, %s]: %s' % (self.experiment.name, | |
| self.device.name, | |||
| self.name) | |||
|
Miguel Urco
|
r53 | ||
| def parms_to_dict(self): | |||
| parameters = {} | |||
| for key in self.__dict__.keys(): | |||
| parameters[key] = getattr(self, key) | |||
| return parameters | |||
|
Miguel Urco
|
r57 | def parms_to_text(self): | |
| raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |||
| return '' | |||
| def parms_to_binary(self): | |||
| raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |||
| return '' | |||
|
Miguel Urco
|
r53 | def dict_to_parms(self, parameters): | |
| if type(parameters) != type({}): | |||
| return | |||
| for key in parameters.keys(): | |||
| setattr(self, key, parameters[key]) | |||
| def export_to_file(self, format="json"): | |||
| import json | |||
|
Miguel Urco
|
r32 | ||
|
Miguel Urco
|
r57 | content_type = '' | |
|
Miguel Urco
|
r53 | ||
| if format == 'text': | |||
| content_type = 'text/plain' | |||
|
Miguel Urco
|
r57 | filename = '%s_%s.%s' %(self.device.device_type.name, self.name, self.device.device_type.name) | |
| content = self.parms_to_text() | |||
|
Miguel Urco
|
r53 | ||
| if format == 'binary': | |||
| content_type = 'application/octet-stream' | |||
|
Miguel Urco
|
r57 | filename = '%s_%s.bin' %(self.device.device_type.name, self.name) | |
| content = self.parms_to_binary() | |||
| if not content_type: | |||
| content_type = 'application/json' | |||
| filename = '%s_%s.json' %(self.device.device_type.name, self.name) | |||
| content = json.dumps(self.parms_to_dict()) | |||
|
Miguel Urco
|
r53 | ||
| fields = {'content_type':content_type, | |||
| 'filename':filename, | |||
| 'content':content | |||
| } | |||
| return fields | |||
|
Miguel Urco
|
r57 | def import_from_file(self, fp): | |
| import os, json | |||
| parms = {} | |||
| path, ext = os.path.splitext(fp.name) | |||
|
Miguel Urco
|
r53 | ||
|
Miguel Urco
|
r57 | if ext == '.json': | |
| parms = json.load(fp) | |||
|
Miguel Urco
|
r53 | ||
|
Miguel Urco
|
r57 | return parms | |
|
Miguel Urco
|
r53 | ||
| def status_device(self): | |||
|
Miguel Urco
|
r57 | raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |
|
Miguel Urco
|
r53 | ||
| return None | |||
| def stop_device(self): | |||
|
Miguel Urco
|
r57 | raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |
|
Miguel Urco
|
r53 | ||
| return None | |||
| def start_device(self): | |||
|
Miguel Urco
|
r57 | raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |
|
Miguel Urco
|
r53 | ||
| return None | |||
| def write_device(self, parms): | |||
|
Miguel Urco
|
r57 | raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |
|
Miguel Urco
|
r53 | ||
| return None | |||
| def read_device(self): | |||
|
Miguel Urco
|
r57 | raise NotImplementedError, "This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper() | |
|
Miguel Urco
|
r53 | ||
| return None | |||
|
Miguel Urco
|
r47 | def get_absolute_url(self): | |
|
Juan C. Espinoza
|
r23 | return reverse('url_%s_conf' % self.device.device_type.name, args=[str(self.id)]) | |
|
Miguel Urco
|
r30 | ||
| def get_absolute_url_edit(self): | |||
| return reverse('url_edit_%s_conf' % self.device.device_type.name, args=[str(self.id)]) | |||
| def get_absolute_url_import(self): | |||
|
Miguel Urco
|
r53 | return reverse('url_import_dev_conf', args=[str(self.id)]) | |
|
Miguel Urco
|
r30 | ||
| def get_absolute_url_export(self): | |||
|
Miguel Urco
|
r53 | return reverse('url_export_dev_conf', args=[str(self.id)]) | |
|
Miguel Urco
|
r30 | ||
| def get_absolute_url_write(self): | |||
|
Miguel Urco
|
r53 | return reverse('url_write_dev_conf', args=[str(self.id)]) | |
|
Miguel Urco
|
r30 | ||
| def get_absolute_url_read(self): | |||
|
Miguel Urco
|
r53 | return reverse('url_read_dev_conf', args=[str(self.id)]) | |
|
Fiorella Quino
|
r49 | ||
|
Miguel Urco
|
r53 | def get_absolute_url_start(self): | |
| return reverse('url_start_dev_conf', args=[str(self.id)]) | |||
|
Fiorella Quino
|
r49 | ||
|
Miguel Urco
|
r53 | def get_absolute_url_stop(self): | |
| return reverse('url_stop_dev_conf', args=[str(self.id)]) | |||
| def get_absolute_url_status(self): | |||
| return reverse('url_status_dev_conf', args=[str(self.id)]) |