models.py
756 lines
| 23.9 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r185 | |||
Fiorella Quino
|
r266 | import os | ||
| import json | ||||
|
Juan C. Espinoza
|
r236 | import requests | ||
|
Fiorella Quino
|
r266 | import time | ||
|
Juan C. Espinoza
|
r85 | from datetime import datetime | ||
|
Juan C. Espinoza
|
r185 | try: | ||
| from polymorphic.models import PolymorphicModel | ||||
| except: | ||||
| from polymorphic import PolymorphicModel | ||||
|
Juan C. Espinoza
|
r0 | |||
|
Fiorella Quino
|
r266 | from django.template.base import kwarg_re | ||
|
Juan C. Espinoza
|
r185 | from django.db import models | ||
| r335 | from django.urls import reverse | |||
|
Fiorella Quino
|
r266 | from django.core.validators import MinValueValidator, MaxValueValidator | ||
|
Juan C. Espinoza
|
r185 | from django.shortcuts import get_object_or_404 | ||
|
Juan C. Espinoza
|
r316 | from django.contrib.auth.models import User | ||
|
Juan C. Espinoza
|
r319 | from django.db.models.signals import post_save | ||
| from django.dispatch import receiver | ||||
|
Fiorella Quino
|
r108 | |||
|
Fiorella Quino
|
r266 | from apps.main.utils import Params | ||
|
Fiorella Quino
|
r72 | |||
|
Fiorella Quino
|
r266 | DEV_PORTS = { | ||
| r350 | 'pedestal' : 80, | |||
| 'pedestal_dev' : 80, | ||||
| 'generator' : 80, | ||||
| 'usrp_rx' : 2000, | ||||
| 'usrp_tx' : 2000, | ||||
|
Fiorella Quino
|
r266 | } | ||
| RADAR_STATES = ( | ||||
| (0, 'No connected'), | ||||
| (1, 'Connected'), | ||||
| (2, 'Configured'), | ||||
| (3, 'Running'), | ||||
| (4, 'Scheduled'), | ||||
Miguel Urco
|
r47 | ) | ||
|
Miguel Urco
|
r41 | |||
|
Fiorella Quino
|
r266 | EXPERIMENT_TYPE = ( | ||
| (0, 'RAW_DATA'), | ||||
| (1, 'PDATA'), | ||||
| ) | ||||
| DECODE_TYPE = ( | ||||
| (0, 'None'), | ||||
| (1, 'TimeDomain'), | ||||
| (2, 'FreqDomain'), | ||||
| (3, 'InvFreqDomain'), | ||||
|
Miguel Urco
|
r47 | ) | ||
|
Miguel Urco
|
r16 | |||
|
Fiorella Quino
|
r226 | DEV_STATES = ( | ||
| r350 | (0, 'Unknown'), | |||
|
Miguel Urco
|
r47 | (1, 'Connected'), | ||
| (2, 'Configured'), | ||||
| (3, 'Running'), | ||||
| r350 | (4, 'Offline'), | |||
|
Miguel Urco
|
r47 | ) | ||
|
Juan C. Espinoza
|
r2 | |||
|
Miguel Urco
|
r13 | DEV_TYPES = ( | ||
| r345 | ('', 'Select a device type'), | |||
| ('pedestal', 'Pedestal Controller'), | ||||
| r350 | ('pedestal_dev', 'Pedestal Controller Dev Mode'), | |||
| r345 | ('generator', 'Pulse Generator'), | |||
| ('usrp_rx', 'Universal Software Radio Peripheral Rx'), | ||||
| ('usrp_tx', 'Universal Software Radio Peripheral Tx'), | ||||
|
Miguel Urco
|
r47 | ) | ||
|
Fiorella Quino
|
r266 | EXP_STATES = ( | ||
| (0,'Error'), #RED | ||||
|
Juan C. Espinoza
|
r329 | (1,'Cancelled'), #YELLOW | ||
|
Fiorella Quino
|
r266 | (2,'Running'), #GREEN | ||
|
Juan C. Espinoza
|
r329 | (3,'Scheduled'), #BLUE | ||
| (4,'Unknown'), #WHITE | ||||
|
Fiorella Quino
|
r49 | ) | ||
|
Juan C. Espinoza
|
r216 | |||
|
Fiorella Quino
|
r266 | CONF_TYPES = ( | ||
| (0, 'Active'), | ||||
| (1, 'Historical'), | ||||
| ) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r319 | class Profile(models.Model): | ||
| user = models.OneToOneField(User, on_delete=models.CASCADE) | ||||
|
Juan C. Espinoza
|
r326 | theme = models.CharField(max_length=30, default='spacelab') | ||
|
Juan C. Espinoza
|
r319 | |||
| @receiver(post_save, sender=User) | ||||
| def create_user_profile(sender, instance, created, **kwargs): | ||||
| if created: | ||||
| Profile.objects.create(user=instance) | ||||
| @receiver(post_save, sender=User) | ||||
| def save_user_profile(sender, instance, **kwargs): | ||||
| instance.profile.save() | ||||
|
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' | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
|
Miguel Urco
|
r53 | return u'%s' % self.name | ||
|
Juan C. Espinoza
|
r89 | def get_absolute_url(self): | ||
|
Juan C. Espinoza
|
r138 | return reverse('url_location', args=[str(self.id)]) | ||
|
Juan C. Espinoza
|
r89 | |||
|
Juan C. Espinoza
|
r2 | class DeviceType(models.Model): | ||
| r350 | name = models.CharField(max_length = 15, choices = DEV_TYPES, default = 'pedestal') | |||
| r335 | sequence = models.PositiveSmallIntegerField(default=55) | |||
|
Miguel Urco
|
r13 | description = models.TextField(blank=True, null=True) | ||
|
Juan C. Espinoza
|
r2 | |||
| class Meta: | ||||
|
Miguel Urco
|
r13 | db_table = 'db_device_types' | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
| r349 | return u'%s' % self.name.title() | |||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r2 | class Device(models.Model): | ||
gonzalesluisfrancisco
|
r343 | device_type = models.ForeignKey('DeviceType', on_delete=models.CASCADE) | ||
| location = models.ForeignKey('Location', on_delete=models.CASCADE) | ||||
| r346 | ip_address = models.GenericIPAddressField(verbose_name = 'IP address', protocol='IPv4', default='0.0.0.0') | |||
|
Miguel Urco
|
r13 | port_address = models.PositiveSmallIntegerField(default=2000) | ||
|
gonzalesluisfrancisco
|
r343 | description = models.TextField(blank=True, null=True) | ||
| status = models.PositiveSmallIntegerField(default=4, choices=DEV_STATES) | ||||
| conf_active = models.PositiveIntegerField(default=0, verbose_name='Current configuration') | ||||
|
Juan C. Espinoza
|
r2 | |||
| class Meta: | ||||
|
Miguel Urco
|
r13 | db_table = 'db_devices' | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
| r349 | ret = self.device_type | |||
| r346 | #ret = u'{} [{}]'.format(self.device_type.name.upper(), self.location.name) | |||
| r349 | return str(ret) | |||
|
Juan C. Espinoza
|
r316 | |||
| @property | ||||
| def name(self): | ||||
| return str(self) | ||||
|
Fiorella Quino
|
r162 | |||
| def get_status(self): | ||||
|
Miguel Urco
|
r47 | return self.status | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r120 | @property | ||
| def status_color(self): | ||||
| color = 'muted' | ||||
| if self.status == 0: | ||||
| color = "danger" | ||||
| elif self.status == 1: | ||||
| r360 | color = "primary" | |||
|
Juan C. Espinoza
|
r120 | elif self.status == 2: | ||
| color = "info" | ||||
| elif self.status == 3: | ||||
| color = "success" | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r120 | return color | ||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r185 | def url(self, path=None): | ||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r185 | if path: | ||
| return 'http://{}:{}/{}/'.format(self.ip_address, self.port_address, path) | ||||
| else: | ||||
| return 'http://{}:{}/'.format(self.ip_address, self.port_address) | ||||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r89 | def get_absolute_url(self): | ||
| return reverse('url_device', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r316 | def get_absolute_url_edit(self): | ||
| return reverse('url_edit_device', args=[str(self.id)]) | ||||
| def get_absolute_url_delete(self): | ||||
| return reverse('url_delete_device', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r266 | def change_ip(self, ip_address, mask, gateway, dns, **kwargs): | ||
|
Fiorella Quino
|
r226 | |||
| r346 | if self.device_type.name=='pedestal': | |||
|
Fiorella Quino
|
r266 | headers = {'content-type': "application/json", | ||
| 'cache-control': "no-cache"} | ||||
|
gonzalesluisfrancisco
|
r343 | ip = [int(x) for x in ip_address.split('.')] | ||
| dns = [int(x) for x in dns.split('.')] | ||||
|
Fiorella Quino
|
r266 | gateway = [int(x) for x in gateway.split('.')] | ||
|
gonzalesluisfrancisco
|
r343 | subnet = [int(x) for x in mask.split('.')] | ||
|
Fiorella Quino
|
r266 | |||
| payload = { | ||||
| "ip": ip, | ||||
| "dns": dns, | ||||
| "gateway": gateway, | ||||
| "subnet": subnet | ||||
| } | ||||
| req = requests.post(self.url('changeip'), data=json.dumps(payload), headers=headers) | ||||
|
Juan C. Espinoza
|
r236 | try: | ||
| answer = req.json() | ||||
|
Fiorella Quino
|
r243 | if answer['changeip']=='ok': | ||
|
gonzalesluisfrancisco
|
r343 | self.message = '25|IP succesfully changed' | ||
|
Juan C. Espinoza
|
r236 | self.ip_address = ip_address | ||
| self.save() | ||||
| else: | ||||
| self.message = '30|An error ocuur when changing IP' | ||||
| except Exception as e: | ||||
| self.message = '40|{}'.format(str(e)) | ||||
|
Juan C. Espinoza
|
r219 | else: | ||
| self.message = 'Not implemented' | ||||
| return False | ||||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r219 | return True | ||
|
Miguel Urco
|
r13 | |||
| class Campaign(models.Model): | ||||
|
gonzalesluisfrancisco
|
r343 | template = models.BooleanField(default=False) | ||
| name = models.CharField(max_length=60, unique=True) | ||||
| start_date = models.DateTimeField(blank=True, null=True) | ||||
| end_date = models.DateTimeField(blank=True, null=True) | ||||
| tags = models.CharField(max_length=40, blank=True, null=True) | ||||
|
Miguel Urco
|
r13 | description = models.TextField(blank=True, null=True) | ||
|
Juan C. Espinoza
|
r85 | experiments = models.ManyToManyField('Experiment', blank=True) | ||
|
gonzalesluisfrancisco
|
r343 | author = models.ForeignKey(User, null=True, blank=True,on_delete=models.CASCADE) | ||
| r335 | ||||
|
Miguel Urco
|
r13 | class Meta: | ||
| db_table = 'db_campaigns' | ||||
|
Juan C. Espinoza
|
r85 | ordering = ('name',) | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
|
Juan C. Espinoza
|
r138 | if self.template: | ||
| return u'{} (template)'.format(self.name) | ||||
| else: | ||||
| return u'{}'.format(self.name) | ||||
|
Fiorella Quino
|
r72 | |||
|
Fiorella Quino
|
r266 | def jsonify(self): | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | data = {} | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | ignored = ('template') | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | for field in self._meta.fields: | ||
| if field.name in ignored: | ||||
| continue | ||||
| data[field.name] = field.value_from_object(self) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | data['start_date'] = data['start_date'].strftime('%Y-%m-%d') | ||
|
gonzalesluisfrancisco
|
r343 | data['end_date'] = data['end_date'].strftime('%Y-%m-%d') | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | return data | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | def parms_to_dict(self): | ||
|
Fiorella Quino
|
r162 | |||
| r311 | params = Params({}) | |||
|
Fiorella Quino
|
r266 | params.add(self.jsonify(), 'campaigns') | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | for exp in Experiment.objects.filter(campaign = self): | ||
| params.add(exp.jsonify(), 'experiments') | ||||
| configurations = Configuration.objects.filter(experiment=exp, type=0) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | for conf in configurations: | ||
| params.add(conf.jsonify(), 'configurations') | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | return params.data | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r109 | def dict_to_parms(self, parms, CONF_MODELS): | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r109 | experiments = Experiment.objects.filter(campaign = self) | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r109 | if experiments: | ||
| for experiment in experiments: | ||||
| experiment.delete() | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | for id_exp in parms['experiments']['allIds']: | ||
| exp_parms = parms['experiments']['byId'][id_exp] | ||||
| dum = (datetime.now() - datetime(1970, 1, 1)).total_seconds() | ||||
| exp = Experiment(name='{}'.format(dum)) | ||||
| exp.save() | ||||
| exp.dict_to_parms(parms, CONF_MODELS, id_exp=id_exp) | ||||
| self.experiments.add(exp) | ||||
| camp_parms = parms['campaigns']['byId'][parms['campaigns']['allIds'][0]] | ||||
|
gonzalesluisfrancisco
|
r343 | self.name = '{}-{}'.format(camp_parms['name'], datetime.now().strftime('%y%m%d')) | ||
|
Fiorella Quino
|
r266 | self.start_date = camp_parms['start_date'] | ||
|
gonzalesluisfrancisco
|
r343 | self.end_date = camp_parms['end_date'] | ||
| self.tags = camp_parms['tags'] | ||||
|
Fiorella Quino
|
r266 | self.save() | ||
|
Fiorella Quino
|
r162 | |||
| return self | ||||
|
Fiorella Quino
|
r267 | def get_experiments_by_radar(self, radar=None): | ||
|
Juan C. Espinoza
|
r172 | |||
| ret = [] | ||||
|
Juan C. Espinoza
|
r196 | if radar: | ||
| locations = Location.objects.filter(pk=radar) | ||||
| else: | ||||
| locations = set([e.location for e in self.experiments.all()]) | ||||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r172 | for loc in locations: | ||
| dum = {} | ||||
|
gonzalesluisfrancisco
|
r343 | dum['name'] = loc.name | ||
| dum['id'] = loc.pk | ||||
|
Juan C. Espinoza
|
r172 | dum['experiments'] = [e for e in self.experiments.all() if e.location==loc] | ||
| ret.append(dum) | ||||
| return ret | ||||
|
Fiorella Quino
|
r108 | def get_absolute_url(self): | ||
| return reverse('url_campaign', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r108 | def get_absolute_url_edit(self): | ||
| return reverse('url_edit_campaign', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r316 | def get_absolute_url_delete(self): | ||
| return reverse('url_delete_campaign', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r100 | def get_absolute_url_export(self): | ||
| return reverse('url_export_campaign', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r108 | def get_absolute_url_import(self): | ||
| return reverse('url_import_campaign', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r84 | class RunningExperiment(models.Model): | ||
|
gonzalesluisfrancisco
|
r343 | radar = models.OneToOneField('Location', on_delete=models.CASCADE) | ||
|
Fiorella Quino
|
r92 | running_experiment = models.ManyToManyField('Experiment', blank = True) | ||
|
gonzalesluisfrancisco
|
r343 | status = models.PositiveSmallIntegerField(default=0, choices=RADAR_STATES) | ||
|
Fiorella Quino
|
r162 | |||
| r360 | class Experiment(PolymorphicModel): | |||
| r358 | ||||
|
gonzalesluisfrancisco
|
r343 | template = models.BooleanField(default=False) | ||
| name = models.CharField(max_length=40, default='', unique=True) | ||||
| r358 | location = models.ForeignKey('Location', null=False, blank=False, on_delete=models.CASCADE, default=None) | |||
| r346 | #freq = models.FloatField(verbose_name='Operating Freq. (MHz)', validators=[MinValueValidator(1), MaxValueValidator(10000)], default=49.9200) | |||
| r358 | #start_time = models.TimeField(default='00:00:00') | |||
| #end_time = models.TimeField(default='23:59:59') | ||||
| r360 | pedestal = models.ForeignKey('pedestal.PedestalConfiguration', null=False, blank=False, on_delete=models.PROTECT, default=None, related_name = "pedestal_conf") | |||
| generator = models.ForeignKey('Device', null=False, blank=False, on_delete=models.PROTECT, default=None, limit_choices_to={'device_type__name': 'generator'}, related_name = "generator_conf") | ||||
| reception_rx = models.ForeignKey('usrp_rx.USRPRXConfiguration', null=False, blank=False, on_delete=models.PROTECT, default=None, related_name = "usrp_rx_CONF") | ||||
| transmission_tx = models.ForeignKey('usrp_tx.USRPTXConfiguration', null=False, blank=False, on_delete=models.PROTECT, default=None, related_name = "usrp_tx") | ||||
|
gonzalesluisfrancisco
|
r343 | task = models.CharField(max_length=36, default='', blank=True, null=True) | ||
| status = models.PositiveSmallIntegerField(default=4, choices=EXP_STATES) | ||||
| r360 | author = models.ForeignKey(User, null=True, blank=True, on_delete=models.CASCADE) | |||
|
gonzalesluisfrancisco
|
r343 | hash = models.CharField(default='', max_length=64, null=True, blank=True) | ||
|
Juan C. Espinoza
|
r2 | |||
| class Meta: | ||||
|
Miguel Urco
|
r13 | db_table = 'db_experiments' | ||
|
Juan C. Espinoza
|
r91 | ordering = ('template', 'name') | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
|
Juan C. Espinoza
|
r91 | if self.template: | ||
|
Juan C. Espinoza
|
r319 | return u'%s (template)' % (self.name) | ||
|
Juan C. Espinoza
|
r91 | else: | ||
|
Juan C. Espinoza
|
r319 | return u'%s' % (self.name) | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | def jsonify(self): | ||
| data = {} | ||||
| ignored = ('template') | ||||
| for field in self._meta.fields: | ||||
| if field.name in ignored: | ||||
| continue | ||||
| data[field.name] = field.value_from_object(self) | ||||
| r358 | ||||
|
Fiorella Quino
|
r266 | data['start_time'] = data['start_time'].strftime('%H:%M:%S') | ||
|
gonzalesluisfrancisco
|
r343 | data['end_time'] = data['end_time'].strftime('%H:%M:%S') | ||
| data['location'] = self.location.name | ||||
|
Fiorella Quino
|
r266 | data['configurations'] = ['{}'.format(conf.pk) for | ||
Juan C. Espinoza
|
r306 | conf in Configuration.objects.filter(experiment=self, type=0)] | ||
|
Fiorella Quino
|
r266 | |||
| return data | ||||
|
Juan C. Espinoza
|
r89 | @property | ||
|
Juan C. Espinoza
|
r138 | def radar_system(self): | ||
|
Juan C. Espinoza
|
r89 | return self.location | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r85 | def clone(self, **kwargs): | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r85 | confs = Configuration.objects.filter(experiment=self, type=0) | ||
| self.pk = None | ||||
|
Juan C. Espinoza
|
r316 | self.name = '{}_{:%y%m%d}'.format(self.name, datetime.now()) | ||
|
Juan C. Espinoza
|
r85 | for attr, value in kwargs.items(): | ||
| setattr(self, attr, value) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r85 | self.save() | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r85 | for conf in confs: | ||
| conf.clone(experiment=self, template=False) | ||||
|
Fiorella Quino
|
r162 | |||
| return self | ||||
|
Juan C. Espinoza
|
r196 | def start(self): | ||
| ''' | ||||
|
Fiorella Quino
|
r226 | Configure and start experiments's devices | ||
|
Juan C. Espinoza
|
r196 | ''' | ||
| r360 | #confs = [] | |||
| #allconfs = Configuration.objects.filter(experiment=self, type = 0).order_by('-device__device_type__sequence') | ||||
| #confs = allconfs | ||||
| experiment = get_object_or_404(Experiment, pk=self.id) | ||||
| id_p = experiment.pedestal_id | ||||
| id_rx = experiment.reception_rx_id | ||||
| id_tx = experiment.transmission_tx_id | ||||
| generator_periode = 1 /experiment.transmission_tx.frequency | ||||
| generator_delay = experiment.transmission_tx.delay_1 | ||||
| generator_width = int((experiment.transmission_tx.ipp * 2 / 300000) * 1000000) * experiment.transmission_tx.pulse_1 / 100 | ||||
| print(generator_width) | ||||
| generator_selector = 1 | ||||
| r335 | ||||
|
Juan C. Espinoza
|
r329 | try: | ||
| r360 | print("Hola") | |||
| #Configuration.objects.get(id = id_rx).start_device() | ||||
| #Configuration.objects.get(id = id_p).start_device() | ||||
| #Configuration.objects.get(id = id_tx).start_device() | ||||
| # for conf in confs: | ||||
| # conf.stop_device() | ||||
| # print("OK") | ||||
| # #conf.write_device() | ||||
| # conf.device.conf_active = conf.pk | ||||
| # conf.device.save() | ||||
| # conf.start_device() | ||||
| # print("OK") | ||||
| # time.sleep(1) | ||||
|
Juan C. Espinoza
|
r329 | except: | ||
| return 0 | ||||
| return 2 | ||||
|
Fiorella Quino
|
r226 | |||
|
Fiorella Quino
|
r250 | |||
|
Fiorella Quino
|
r226 | def stop(self): | ||
|
Juan C. Espinoza
|
r196 | ''' | ||
|
Fiorella Quino
|
r226 | Stop experiments's devices | ||
| r346 | PEDESTAL, PULSE GENERATOR & USRP's | |||
|
Juan C. Espinoza
|
r196 | ''' | ||
|
Fiorella Quino
|
r226 | |||
| r360 | #confs = Configuration.objects.filter(experiment=self, type = 0).order_by('device__device_type__sequence') | |||
| experiment = get_object_or_404(Experiment, pk=self.id) | ||||
| id_p = experiment.pedestal_id | ||||
| id_rx = experiment.reception_rx_id | ||||
| id_tx = experiment.transmission_tx_id | ||||
|
Juan C. Espinoza
|
r329 | try: | ||
| r360 | Configuration.objects.get(id = id_rx).stop_device() | |||
| Configuration.objects.get(id = id_p).stop_device() | ||||
| Configuration.objects.get(id = id_tx).stop_device() | ||||
| # for conf in confs: | ||||
| # conf.stop_device() | ||||
|
Juan C. Espinoza
|
r329 | except: | ||
| return 0 | ||||
| return 1 | ||||
|
Fiorella Quino
|
r250 | |||
|
Juan C. Espinoza
|
r196 | def get_status(self): | ||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r306 | if self.status == 3: | ||
| return | ||||
| confs = Configuration.objects.filter(experiment=self, type=0) | ||||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r240 | for conf in confs: | ||
| conf.status_device() | ||||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r216 | total = confs.aggregate(models.Sum('device__status'))['device__status__sum'] | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r196 | if total==2*confs.count(): | ||
| status = 1 | ||||
| elif total == 3*confs.count(): | ||||
|
Fiorella Quino
|
r84 | status = 2 | ||
| else: | ||||
|
Juan C. Espinoza
|
r196 | status = 0 | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r306 | self.status = status | ||
| self.save() | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r84 | def status_color(self): | ||
|
Juan C. Espinoza
|
r121 | color = 'muted' | ||
|
Fiorella Quino
|
r84 | if self.status == 0: | ||
| color = "danger" | ||||
| elif self.status == 1: | ||||
|
Juan C. Espinoza
|
r329 | color = "warning" | ||
|
Fiorella Quino
|
r84 | elif self.status == 2: | ||
|
Juan C. Espinoza
|
r91 | color = "success" | ||
|
Fiorella Quino
|
r84 | elif self.status == 3: | ||
|
Juan C. Espinoza
|
r329 | color = "info" | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r226 | return color | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r100 | def parms_to_dict(self): | ||
|
Fiorella Quino
|
r162 | |||
| r311 | params = Params({}) | |||
|
Fiorella Quino
|
r266 | params.add(self.jsonify(), 'experiments') | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | configurations = Configuration.objects.filter(experiment=self, type=0) | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | for conf in configurations: | ||
| params.add(conf.jsonify(), 'configurations') | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | return params.data | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | def dict_to_parms(self, parms, CONF_MODELS, id_exp=None): | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r108 | configurations = Configuration.objects.filter(experiment=self) | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | if id_exp is not None: | ||
| exp_parms = parms['experiments']['byId'][id_exp] | ||||
| else: | ||||
| exp_parms = parms['experiments']['byId'][parms['experiments']['allIds'][0]] | ||||
|
Fiorella Quino
|
r108 | if configurations: | ||
|
Fiorella Quino
|
r243 | for configuration in configurations: | ||
| configuration.delete() | ||||
|
Fiorella Quino
|
r266 | |||
| for id_conf in exp_parms['configurations']: | ||||
| conf_parms = parms['configurations']['byId'][id_conf] | ||||
|
gonzalesluisfrancisco
|
r343 | device = Device.objects.filter(device_type__name=conf_parms['device_type'])[0] | ||
| model = CONF_MODELS[conf_parms['device_type']] | ||||
|
Fiorella Quino
|
r266 | conf = model( | ||
| experiment = self, | ||||
| device = device, | ||||
| ) | ||||
| conf.dict_to_parms(parms, id=id_conf) | ||||
| r358 | ||||
|
Fiorella Quino
|
r266 | location, created = Location.objects.get_or_create(name=exp_parms['location']) | ||
| self.name = '{}-{}'.format(exp_parms['name'], datetime.now().strftime('%y%m%d')) | ||||
|
gonzalesluisfrancisco
|
r343 | self.location = location | ||
|
Fiorella Quino
|
r266 | self.start_time = exp_parms['start_time'] | ||
|
gonzalesluisfrancisco
|
r343 | self.end_time = exp_parms['end_time'] | ||
|
Fiorella Quino
|
r110 | self.save() | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r110 | return self | ||
|
Fiorella Quino
|
r226 | |||
|
Juan C. Espinoza
|
r196 | def get_absolute_url(self): | ||
| return reverse('url_experiment', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r108 | def get_absolute_url_edit(self): | ||
| return reverse('url_edit_experiment', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r316 | def get_absolute_url_delete(self): | ||
| return reverse('url_delete_experiment', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r108 | def get_absolute_url_import(self): | ||
| return reverse('url_import_experiment', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r100 | def get_absolute_url_export(self): | ||
| return reverse('url_export_experiment', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r240 | def get_absolute_url_start(self): | ||
| return reverse('url_start_experiment', args=[str(self.id)]) | ||||
| def get_absolute_url_stop(self): | ||||
| return reverse('url_stop_experiment', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r6 | class Configuration(PolymorphicModel): | ||
|
Juan C. Espinoza
|
r2 | |||
| r349 | id = models.AutoField(primary_key=True) | |||
|
gonzalesluisfrancisco
|
r343 | template = models.BooleanField(default=False) | ||
| # name = models.CharField(verbose_name="Configuration Name", max_length=40, default='') | ||||
| device = models.ForeignKey('Device', verbose_name='Device', null=True, on_delete=models.CASCADE) | ||||
| r352 | label = models.CharField(verbose_name="Label", max_length=40, default='', blank=False, null=False) | |||
|
gonzalesluisfrancisco
|
r343 | type = models.PositiveSmallIntegerField(default=0, choices=CONF_TYPES) | ||
| created_date = models.DateTimeField(auto_now_add=True) | ||||
|
Miguel Urco
|
r22 | programmed_date = models.DateTimeField(auto_now=True) | ||
|
gonzalesluisfrancisco
|
r343 | parameters = models.TextField(default='{}') | ||
| author = models.ForeignKey(User, null=True, blank=True,on_delete=models.CASCADE) | ||||
| hash = models.CharField(default='', max_length=64, null=True, blank=True) | ||||
|
Miguel Urco
|
r53 | message = "" | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r2 | class Meta: | ||
|
Miguel Urco
|
r13 | db_table = 'db_configurations' | ||
|
Juan C. Espinoza
|
r316 | ordering = ('device__device_type__name',) | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | def __str__(self): | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r316 | ret = u'{} '.format(self.device.device_type.name.upper()) | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | if 'mix' in [f.name for f in self._meta.get_fields()]: | ||
|
Juan C. Espinoza
|
r138 | if self.mix: | ||
|
Juan C. Espinoza
|
r316 | ret = '{} MIX '.format(self.device.device_type.name.upper()) | ||
| r335 | ||||
|
Juan C. Espinoza
|
r316 | if 'label' in [f.name for f in self._meta.get_fields()]: | ||
|
Juan C. Espinoza
|
r319 | ret += '{}'.format(self.label) | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r138 | if self.template: | ||
|
Juan C. Espinoza
|
r316 | ret += ' (template)' | ||
| r335 | ||||
|
Juan C. Espinoza
|
r316 | return ret | ||
| @property | ||||
| def name(self): | ||||
| return str(self) | ||||
|
Juan C. Espinoza
|
r79 | |||
|
Fiorella Quino
|
r266 | def jsonify(self): | ||
| data = {} | ||||
| ignored = ('type', 'polymorphic_ctype', 'configuration_ptr', | ||||
| 'created_date', 'programmed_date', 'template', 'device', | ||||
| r360 | 'experiment', 'author') | |||
|
Fiorella Quino
|
r266 | |||
| for field in self._meta.fields: | ||||
| if field.name in ignored: | ||||
| continue | ||||
| data[field.name] = field.value_from_object(self) | ||||
| data['device_type'] = self.device.device_type.name | ||||
| return data | ||||
|
Juan C. Espinoza
|
r85 | def clone(self, **kwargs): | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r85 | self.pk = None | ||
| self.id = None | ||||
| for attr, value in kwargs.items(): | ||||
| setattr(self, attr, value) | ||||
|
Fiorella Quino
|
r162 | |||
| self.save() | ||||
|
Juan C. Espinoza
|
r85 | |||
| return self | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def parms_to_dict(self): | ||
|
Fiorella Quino
|
r162 | |||
| r311 | params = Params({}) | |||
|
Fiorella Quino
|
r266 | params.add(self.jsonify(), 'configurations') | ||
| return params.data | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | def parms_to_text(self): | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | raise NotImplementedError("This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper()) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | def parms_to_binary(self): | ||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r172 | raise NotImplementedError("This method should be implemented in %s Configuration model" %str(self.device.device_type.name).upper()) | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | def dict_to_parms(self, parameters, id=None): | ||
| params = Params(parameters) | ||||
| if id: | ||||
| data = params.get_conf(id_conf=id) | ||||
| else: | ||||
| data = params.get_conf(dtype=self.device.device_type.name) | ||||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r266 | for key, value in data.items(): | ||
| if key not in ('id', 'device_type'): | ||||
| setattr(self, key, value) | ||||
| r346 | ||||
|
Fiorella Quino
|
r266 | self.save() | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r267 | |||
| def export_to_file(self, format="json"): | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | content_type = '' | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r276 | if format == 'racp': | ||
| content_type = 'text/plain' | ||||
| filename = '%s_%s.%s' %(self.device.device_type.name, self.name, 'racp') | ||||
| content = self.parms_to_text(file_format = 'racp') | ||||
|
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() | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | if format == 'binary': | ||
| content_type = 'application/octet-stream' | ||||
|
Fiorella Quino
|
r266 | filename = '%s_%s.bin' %(self.device.device_type.name, self.name) | ||
|
Miguel Urco
|
r57 | content = self.parms_to_binary() | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | if not content_type: | ||
| content_type = 'application/json' | ||||
| filename = '%s_%s.json' %(self.device.device_type.name, self.name) | ||||
|
Juan C. Espinoza
|
r79 | content = json.dumps(self.parms_to_dict(), indent=2) | ||
| r360 | print(content) | |||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | fields = {'content_type':content_type, | ||
| 'filename':filename, | ||||
| 'content':content | ||||
| } | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | return fields | ||
|
Fiorella Quino
|
r162 | |||
|
Fiorella Quino
|
r267 | def import_from_file(self, fp): | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | parms = {} | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | path, ext = os.path.splitext(fp.name) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | if ext == '.json': | ||
| parms = json.load(fp) | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r57 | return parms | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def status_device(self): | ||
|
Fiorella Quino
|
r162 | |||
| r360 | self.message = 'Function not supported' | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def stop_device(self): | ||
|
Fiorella Quino
|
r162 | |||
| r360 | self.message = 'Function not supported' | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def start_device(self): | ||
| r360 | ||||
| self.message = 'Function not supported' | ||||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r162 | |||
| r346 | def write_device(self): | |||
|
Fiorella Quino
|
r162 | |||
| r360 | self.message = 'Function not supported' | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def read_device(self): | ||
|
Fiorella Quino
|
r162 | |||
| r360 | self.message = 'Function not supported' | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r162 | |||
|
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)]) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r30 | def get_absolute_url_edit(self): | ||
| return reverse('url_edit_%s_conf' % self.device.device_type.name, args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Juan C. Espinoza
|
r316 | def get_absolute_url_delete(self): | ||
| return reverse('url_delete_dev_conf', args=[str(self.id)]) | ||||
| r335 | ||||
|
Miguel Urco
|
r30 | def get_absolute_url_import(self): | ||
|
Miguel Urco
|
r53 | return reverse('url_import_dev_conf', args=[str(self.id)]) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r30 | def get_absolute_url_export(self): | ||
|
Miguel Urco
|
r53 | return reverse('url_export_dev_conf', args=[str(self.id)]) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r30 | def get_absolute_url_write(self): | ||
|
Miguel Urco
|
r53 | return reverse('url_write_dev_conf', args=[str(self.id)]) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r30 | def get_absolute_url_read(self): | ||
|
Miguel Urco
|
r53 | return reverse('url_read_dev_conf', args=[str(self.id)]) | ||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def get_absolute_url_start(self): | ||
| return reverse('url_start_dev_conf', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def get_absolute_url_stop(self): | ||
| return reverse('url_stop_dev_conf', args=[str(self.id)]) | ||||
|
Fiorella Quino
|
r162 | |||
|
Miguel Urco
|
r53 | def get_absolute_url_status(self): | ||
|
Fiorella Quino
|
r162 | return reverse('url_status_dev_conf', args=[str(self.id)]) | ||
