from django.db import models from apps.main.models import Configuration from apps.main.utils import Params from django.core.validators import MinValueValidator, MaxValueValidator from .files import read_json_file import requests class ATRADData(models.Model): datetime = models.DateTimeField() nstx = models.SmallIntegerField() status = models.SmallIntegerField() temp_cll = models.SmallIntegerField() nboards = models.SmallIntegerField() tempdvr = models.SmallIntegerField() potincdvr = models.SmallIntegerField() potretdvr = models.SmallIntegerField() temp1 = models.SmallIntegerField() potinc1 = models.SmallIntegerField() potret1 = models.SmallIntegerField() temp2 = models.SmallIntegerField() potinc2 = models.SmallIntegerField() potret2 = models.SmallIntegerField() temp3 = models.SmallIntegerField() potinc3 = models.SmallIntegerField() potret3 = models.SmallIntegerField() temp4 = models.SmallIntegerField() potinc4 = models.SmallIntegerField() potret4 = models.SmallIntegerField() temp5 = models.SmallIntegerField() potinc5 = models.SmallIntegerField() potret5 = models.SmallIntegerField() temp6 = models.SmallIntegerField() potinc6 = models.SmallIntegerField() potret6 = models.SmallIntegerField() class Meta: db_table = 'atrad_datas' def __unicode__(self): return u'%s' % (self.name) class ATRADConfiguration(Configuration): topic = models.PositiveIntegerField(verbose_name='Topic',validators=[MaxValueValidator(10)], default = 0) def status_device(self): ip=self.device.ip_address port=self.device.port_address route = "http://" + str(ip) + ":" + str(port) + "/status/" try: r = requests.get(route, timeout=0.7) except Exception as e: self.device.status = 0 self.device.save() self.message = 'Could not read TX status: ' + str(e) return False response = r.json() self.device.status = response['status'] self.message = response['message'] self.device.save() if response['components_status']==0: return False return True def start_device(self): ip=self.device.ip_address port=self.device.port_address #---Device must be configured if not self.device.status == 2: self.message = 'TX Device must be configured.' return False #---Frequencies from form post_data = self.parms_to_dict() route = "http://" + str(ip) + ":" + str(port) + "/write/" try: r = requests.post(route, post_data, timeout=0.7) except Exception as e: self.message = "Could not start TX device. "+str(e) return False response = r.json() if response['status']==1: self.device.status = 1 self.device.save() self.message = response['message'] return False self.device.status = response['status'] self.device.save() self.message = response['message'] return True def stop_device(self): ip=self.device.ip_address port=self.device.port_address if self.device.status == 2: #Configured self.message = 'TX device is already stopped.' return False # Se crea el modo ocupado para una vez inicia el STOP self.device.status = 5 self.device.save() # Por si se demora deteniendo, que su estado sea busy post_data = {"topic":0} route = "http://" + str(ip) + ":" + str(port) + "/write/" try: r = requests.post(route, post_data, timeout=0.7) except Exception as e: self.message = "Could not write TX parameters. "+str(e) self.device.status = 0 self.device.save() return False response = r.json() status = response['status'] if status == 1: self.device.status = status self.device.save() self.message = 'Could not stop TX device.' return False self.message = 'TX device has been stopped successfully.' self.device.status = 2 self.device.save() return True def read_device(self): ip=self.device.ip_address port=self.device.port_address route = "http://" + str(ip) + ":" + str(port) + "/read/" try: frequencies = requests.get(route,timeout=0.7) except: self.message = "Could not read TX parameters from this device" return None frequencies = frequencies.json() if frequencies: frequencies = frequencies.get("Frequencies") topic = frequencies.get("topic") parms = {'topic': topic} self.message = "TX parameters have been successfully read" return parms else: self.message = "Error reading TX parameters" return None def write_device(self): ip=self.device.ip_address port=self.device.port_address #---Frequencies from form parms = self.parms_to_dict()['configurations'] for parm in parms['allIds']: byid = parm frequencies = parms['byId'][byid] post_data = {} for data in frequencies: if data in ['topic']: post_data[data] = frequencies[data] route = "http://" + str(ip) + ":" + str(port) + "/write/" print (post_data) try: r = requests.post(route, post_data, timeout=0.7) except: self.message = "Could not write TX parameters" self.device.status = 0 self.device.save() return False response = r.json() self.message = response['message'] self.device.status = response['status'] self.device.save() if self.device.status==1: return False return True class Meta: db_table = 'atrad_configurations'