models.py
232 lines
| 6.6 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r0 | from django.db import models | ||
|
Juan C. Espinoza
|
r6 | from apps.main.models import Configuration | ||
Fiorella Quino
|
r4 | from django.core.validators import MinValueValidator, MaxValueValidator | ||
|
Juan C. Espinoza
|
r172 | from .files import read_json_file | ||
|
Fiorella Quino
|
r208 | import requests | ||
|
Fiorella Quino
|
r43 | # Create your models here. validators=[MinValueValidator(62.5e6), MaxValueValidator(450e6)] | ||
|
Fiorella Quino
|
r4 | |||
|
Juan C. Espinoza
|
r6 | class CGSConfiguration(Configuration): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r245 | freq0 = models.PositiveIntegerField(verbose_name='Frequency 0 (Hz)',validators=[MaxValueValidator(450e6)], default = 60) | ||
| freq1 = models.PositiveIntegerField(verbose_name='Frequency 1 (Hz)',validators=[MaxValueValidator(450e6)], default = 60) | ||||
| freq2 = models.PositiveIntegerField(verbose_name='Frequency 2 (Hz)',validators=[MaxValueValidator(450e6)], default = 60) | ||||
| freq3 = models.PositiveIntegerField(verbose_name='Frequency 3 (Hz)',validators=[MaxValueValidator(450e6)], default = 60) | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r31 | def verify_frequencies(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r31 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r42 | def update_from_file(self, fp): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r42 | kwargs = read_json_file(fp) | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r42 | if not kwargs: | ||
| return False | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r42 | self.freq0 = kwargs['freq0'] | ||
| self.freq1 = kwargs['freq1'] | ||||
| self.freq2 = kwargs['freq2'] | ||||
| self.freq3 = kwargs['freq3'] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r42 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r58 | def parms_to_dict(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r58 | parameters = {} | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r108 | parameters['device_id'] = self.device.id | ||
|
Fiorella Quino
|
r243 | parameters['device_type'] = self.device.device_type.name | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r58 | if self.freq0 == None or self.freq0 == '': | ||
| parameters['freq0'] = 0 | ||||
| else: | ||||
| parameters['freq0'] = self.freq0 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r58 | if self.freq1 == None or self.freq1 == '': | ||
| parameters['freq1'] = 0 | ||||
| else: | ||||
| parameters['freq1'] = self.freq1 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r58 | if self.freq2 == None or self.freq2 == '': | ||
| parameters['freq2'] = 0 | ||||
| else: | ||||
|
Juan C. Espinoza
|
r172 | parameters['freq2'] = self.freq2 | ||
|
Fiorella Quino
|
r58 | if self.freq3 == None or self.freq3 == '': | ||
| parameters['freq3'] = 0 | ||||
| else: | ||||
| parameters['freq3'] = self.freq3 | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r58 | return parameters | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r108 | def dict_to_parms(self, parameters): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r108 | self.freq0 = parameters['freq0'] | ||
| self.freq1 = parameters['freq1'] | ||||
| self.freq2 = parameters['freq2'] | ||||
| self.freq3 = parameters['freq3'] | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | def status_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | ip=self.device.ip_address | ||
| port=self.device.port_address | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r248 | route = "http://" + str(ip) + ":" + str(port) + "/status/" | ||
|
Fiorella Quino
|
r59 | try: | ||
|
Fiorella Quino
|
r248 | r = requests.get(route, timeout=0.7) | ||
|
Fiorella Quino
|
r215 | except Exception as e: | ||
|
Fiorella Quino
|
r59 | self.device.status = 0 | ||
| self.device.save() | ||||
|
Fiorella Quino
|
r215 | self.message = 'Could not read CGS status: ' + str(e) | ||
|
Fiorella Quino
|
r207 | return False | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r248 | response = r.json() | ||
| self.device.status = response['status'] | ||||
| self.message = response['message'] | ||||
| self.device.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r248 | if response['components_status']==0: | ||
|
Fiorella Quino
|
r220 | return False | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r207 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r208 | def start_device(self): | ||
| ip=self.device.ip_address | ||||
| port=self.device.port_address | ||||
|
Fiorella Quino
|
r249 | #---Device must be configured | ||
| if not self.device.status == 2: | ||||
| self.message = 'CGS Device must be configured.' | ||||
| return False | ||||
|
Fiorella Quino
|
r208 | #---Frequencies from form | ||
|
Fiorella Quino
|
r249 | post_data = self.parms_to_dict() | ||
| route = "http://" + str(ip) + ":" + str(port) + "/write/" | ||||
|
Fiorella Quino
|
r208 | |||
| try: | ||||
| r = requests.post(route, post_data, timeout=0.7) | ||||
|
Fiorella Quino
|
r249 | except Exception as e: | ||
| self.message = "Could not start CGS device. "+str(e) | ||||
| return False | ||||
| response = r.json() | ||||
| if response['status']==1: | ||||
| self.device.status = 1 | ||||
| self.device.save() | ||||
| self.message = response['message'] | ||||
|
Fiorella Quino
|
r208 | return False | ||
|
Fiorella Quino
|
r249 | self.device.status = response['status'] | ||
| self.device.save() | ||||
| self.message = response['message'] | ||||
| return True | ||||
|
Fiorella Quino
|
r208 | |||
| def stop_device(self): | ||||
| ip=self.device.ip_address | ||||
| port=self.device.port_address | ||||
|
Fiorella Quino
|
r248 | if self.device.status == 2: #Configured | ||
| self.message = 'CGS device is already stopped.' | ||||
| return False | ||||
| post_data = {"freq0":0, "freq1":0, "freq2":0, "freq3":0} | ||||
| route = "http://" + str(ip) + ":" + str(port) + "/write/" | ||||
|
Fiorella Quino
|
r208 | |||
| try: | ||||
| r = requests.post(route, post_data, timeout=0.7) | ||||
|
Fiorella Quino
|
r248 | except Exception as e: | ||
| self.message = "Could not write CGS parameters. "+str(e) | ||||
| self.device.status = 0 | ||||
| self.device.save() | ||||
|
Fiorella Quino
|
r208 | return False | ||
|
Fiorella Quino
|
r248 | response = r.json() | ||
| status = response['status'] | ||||
| if status == 1: | ||||
| self.device.status = status | ||||
| self.device.save() | ||||
| self.message = 'Could not stop CGS device.' | ||||
| return False | ||||
|
Fiorella Quino
|
r208 | |||
|
Fiorella Quino
|
r248 | self.message = 'CGS device has been stopped successfully.' | ||
| self.device.status = 2 | ||||
| self.device.save() | ||||
|
Fiorella Quino
|
r208 | |||
|
Fiorella Quino
|
r248 | return True | ||
|
Fiorella Quino
|
r208 | |||
|
Fiorella Quino
|
r59 | def read_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | ip=self.device.ip_address | ||
| port=self.device.port_address | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r246 | route = "http://" + str(ip) + ":" + str(port) + "/read/" | ||
|
Fiorella Quino
|
r59 | try: | ||
|
Fiorella Quino
|
r208 | frequencies = requests.get(route,timeout=0.7) | ||
|
Fiorella Quino
|
r59 | except: | ||
| self.message = "Could not read CGS parameters from this device" | ||||
| return None | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | frequencies = frequencies.json() | ||
|
Fiorella Quino
|
r246 | if frequencies: | ||
| frequencies = frequencies.get("Frequencies") | ||||
| freq0 = frequencies.get("freq0") | ||||
| freq1 = frequencies.get("freq1") | ||||
| freq2 = frequencies.get("freq2") | ||||
| freq3 = frequencies.get("freq3") | ||||
| parms = {'freq0': freq0, | ||||
| 'freq1': freq1, | ||||
| 'freq2': freq2, | ||||
| 'freq3': freq3} | ||||
| self.message = "CGS parameters have been successfully read" | ||||
| return parms | ||||
| else: | ||||
| self.message = "Error reading CGS parameters" | ||||
| return None | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | def write_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | ip=self.device.ip_address | ||
| port=self.device.port_address | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | #---Frequencies from form | ||
|
Fiorella Quino
|
r247 | frequencies = self.parms_to_dict() | ||
| post_data = {} | ||||
| for data in frequencies: | ||||
| if data in ['freq0','freq1','freq2','freq3']: | ||||
| post_data[data] = frequencies[data] | ||||
| route = "http://" + str(ip) + ":" + str(port) + "/write/" | ||||
|
Juan C. Espinoza
|
r172 | |||
| try: | ||||
|
Fiorella Quino
|
r208 | r = requests.post(route, post_data, timeout=0.7) | ||
|
Fiorella Quino
|
r59 | except: | ||
| self.message = "Could not write CGS parameters" | ||||
|
Fiorella Quino
|
r247 | self.device.status = 0 | ||
| self.device.save() | ||||
|
Fiorella Quino
|
r207 | return False | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r247 | response = r.json() | ||
| self.message = response['message'] | ||||
| self.device.status = response['status'] | ||||
| self.device.save() | ||||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r247 | if self.device.status==1: | ||
| return False | ||||
|
Fiorella Quino
|
r248 | |||
|
Fiorella Quino
|
r247 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
|
Fiorella Quino
|
r59 | |||
|
Juan C. Espinoza
|
r6 | class Meta: | ||
| db_table = 'cgs_configurations' | ||||