models.py
296 lines
| 9.6 KiB
| text/x-python
|
PythonLexer
Juan C. Espinoza
|
r45 | import ast | ||
|
Juan C. Espinoza
|
r25 | import json | ||
|
Juan C. Espinoza
|
r175 | import requests | ||
| r345 | import base64 | |||
| r346 | import struct | |||
|
Juan C. Espinoza
|
r185 | from struct import pack | ||
| r346 | import time | |||
| from django.contrib import messages | ||||
|
Juan C. Espinoza
|
r0 | from django.db import models | ||
| r335 | from django.urls import reverse | |||
|
Juan C. Espinoza
|
r23 | from django.core.validators import MinValueValidator, MaxValueValidator | ||
|
Juan C. Espinoza
|
r6 | from apps.main.models import Configuration | ||
|
Juan C. Espinoza
|
r79 | |||
| r345 | AXIS_VALUE = ( | |||
| r360 | ('azimuth', 'PPI'), | |||
| ('elevation', 'RHI') | ||||
| r345 | ) | |||
|
Juan C. Espinoza
|
r6 | |||
| r346 | class PedestalConfiguration(Configuration): | |||
|
Juan C. Espinoza
|
r172 | |||
| r345 | axis = models.CharField( | |||
| verbose_name='Axis', | ||||
| r360 | max_length=10, | |||
| r345 | choices=AXIS_VALUE, | |||
| null=False, | ||||
| blank=False | ||||
| ) | ||||
| speed = models.FloatField( | ||||
| verbose_name='Speed', | ||||
| validators=[MinValueValidator(-20), MaxValueValidator(20)], | ||||
| blank=False, | ||||
| null=False | ||||
| ) | ||||
| table = models.CharField( | ||||
| verbose_name="Table", | ||||
| max_length=100, | ||||
| default='', | ||||
| blank=False, | ||||
| r352 | null=False, | |||
| help_text="Please separate the values with commas" | ||||
| r345 | ) | |||
|
Juan C. Espinoza
|
r6 | |||
| class Meta: | ||||
| r345 | db_table = 'pedestal_configurations' | |||
| def __str__(self): | ||||
| r349 | return str(self.label) | |||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r79 | def get_absolute_url_plot(self): | ||
| r345 | return reverse('url_plot_pedestal_pulses', args=[str(self.id)]) | |||
|
Juan C. Espinoza
|
r172 | |||
Fiorella Quino
|
r264 | def request(self, cmd, method='get', **kwargs): | ||
| req = getattr(requests, method)(self.device.url(cmd), **kwargs) | ||||
| payload = req.json() | ||||
| return payload | ||||
|
Juan C. Espinoza
|
r175 | def status_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r222 | try: | ||
| r350 | #self.device.status = 0 | |||
| #payload = self.request('status') | ||||
| payload = requests.get(self.device.url()) | ||||
| print(payload) | ||||
| if payload: | ||||
| self.device.status = 1 | ||||
|
Fiorella Quino
|
r264 | elif payload['status']=='disable': | ||
|
Juan C. Espinoza
|
r222 | self.device.status = 2 | ||
|
Juan C. Espinoza
|
r185 | else: | ||
|
Fiorella Quino
|
r243 | self.device.status = 1 | ||
|
Juan C. Espinoza
|
r222 | self.device.save() | ||
| r346 | self.message = 'Pedestal status: {}'.format(payload['status']) | |||
|
Juan C. Espinoza
|
r222 | return False | ||
|
Juan C. Espinoza
|
r185 | except Exception as e: | ||
|
Juan C. Espinoza
|
r222 | if 'No route to host' not in str(e): | ||
| self.device.status = 4 | ||||
| self.device.save() | ||||
| r346 | self.message = 'Pedestal status: {}'.format(str(e)) | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r243 | |||
| self.device.save() | ||||
| return True | ||||
|
Juan C. Espinoza
|
r175 | |||
| def reset_device(self): | ||||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r185 | try: | ||
|
Fiorella Quino
|
r264 | payload = self.request('reset', 'post') | ||
|
Juan C. Espinoza
|
r222 | if payload['reset']=='ok': | ||
| r346 | self.message = 'Pedestal restarted OK' | |||
|
Fiorella Quino
|
r264 | self.device.status = 2 | ||
| self.device.save() | ||||
|
Juan C. Espinoza
|
r185 | else: | ||
| r346 | self.message = 'Pedestal restart fail' | |||
|
Juan C. Espinoza
|
r185 | self.device.status = 4 | ||
| self.device.save() | ||||
| except Exception as e: | ||||
| r346 | self.message = 'Pedestal reset: {}'.format(str(e)) | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r185 | return True | ||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r175 | def stop_device(self): | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r185 | try: | ||
| r346 | command = self.device.url() + "stop" | |||
| r = requests.get(command) | ||||
| if r: | ||||
| self.device.status = 4 | ||||
|
Juan C. Espinoza
|
r185 | self.device.save() | ||
| r346 | self.message = 'Pedestal stopped' | |||
|
Juan C. Espinoza
|
r185 | else: | ||
| self.device.status = 4 | ||||
| self.device.save() | ||||
| return False | ||||
| except Exception as e: | ||||
|
Juan C. Espinoza
|
r222 | if 'No route to host' not in str(e): | ||
| self.device.status = 4 | ||||
| else: | ||||
| self.device.status = 0 | ||||
| r350 | #self.message = 'Pedestal stop: {}'.format(str(e)) | |||
| self.message = "Pedestal can't start, please check network/device connection or IP address/port configuration" | ||||
|
Juan C. Espinoza
|
r222 | self.device.save() | ||
|
Fiorella Quino
|
r243 | return False | ||
|
Juan C. Espinoza
|
r185 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | def start_device(self): | ||
| r350 | ||||
|
Juan C. Espinoza
|
r185 | try: | ||
| r346 | pedestal = PedestalConfiguration.objects.get(pk=self) | |||
| r348 | print(pedestal) | |||
| r360 | #pedestal_axis = pedestal.get_axis_display() | |||
| r345 | print(pedestal) | |||
| r360 | #print(pedestal_axis) | |||
| r345 | table = pedestal.table | |||
| print(table) | ||||
| r352 | li = list(table.split(",")) | |||
| r346 | print(li) | |||
| list_of_floats = [] | ||||
| for item in li: | ||||
| list_of_floats.append(float(item)) | ||||
| print(list_of_floats) | ||||
| byte_table = [] | ||||
| for x in list_of_floats: | ||||
| temp = bytearray(struct.pack("f", x)) | ||||
| byte_table.append(temp[3]) | ||||
| byte_table.append(temp[2]) | ||||
| byte_table.append(temp[1]) | ||||
| byte_table.append(temp[0]) | ||||
| print(byte_table) | ||||
| r357 | coded_table = base64.standard_b64encode(bytes(byte_table)) | |||
| r346 | coded_table_ascii = coded_table.decode('ascii') | |||
| print(coded_table_ascii) | ||||
| r360 | data = {'axis': pedestal.axis, 'speed': pedestal.speed, 'table': coded_table_ascii} | |||
| r345 | print(data) | |||
| r346 | json_data = json.dumps(data) | |||
| print(json_data) | ||||
| first_position = table[0] | ||||
| r360 | if pedestal.axis=='PPI': | |||
| r346 | json_az = json.dumps({"axis": 'azimuth', "position": 0.0}) | |||
| json_el = json.dumps({"axis": 'elevation', "position": first_position}) | ||||
| else: | ||||
| json_az = json.dumps({"axis": 'azimuth', "position": first_position}) | ||||
| json_el = json.dumps({"axis": 'elevation', "position": 0.0}) | ||||
| r357 | base64_table = base64.standard_b64encode(json_data.encode('ascii')) | |||
| base64_az = base64.standard_b64encode(json_az.encode('ascii')) | ||||
| base64_el = base64.standard_b64encode(json_el.encode('ascii')) | ||||
| r346 | ||||
| table_url = self.device.url() + "table?params=" | ||||
| az_url = self.device.url() + "position?params=" | ||||
| el_url = self.device.url() + "position?params=" | ||||
| complete_url = table_url + base64_table.decode('ascii') | ||||
| az_url = az_url + base64_az.decode('ascii') | ||||
| el_url = el_url + base64_el.decode('ascii') | ||||
| print(complete_url) | ||||
| print(az_url) | ||||
| print(el_url) | ||||
| r = requests.get(az_url) | ||||
| r = requests.get(el_url) | ||||
| #time.sleep(10) | ||||
| r = requests.get(complete_url) | ||||
| if r: | ||||
|
Juan C. Espinoza
|
r185 | self.device.status = 3 | ||
| self.device.save() | ||||
| r346 | self.message = 'Pedestal configured and started' | |||
|
Juan C. Espinoza
|
r185 | else: | ||
| return False | ||||
| except Exception as e: | ||||
|
Juan C. Espinoza
|
r222 | if 'No route to host' not in str(e): | ||
| self.device.status = 4 | ||||
| else: | ||||
| self.device.status = 0 | ||||
| r350 | #self.message = 'Pedestal start: {}'.format(str(e)) | |||
| self.message = "Pedestal can't start, please check network/device connection or IP address/port configuration" | ||||
|
Juan C. Espinoza
|
r222 | self.device.save() | ||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r185 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
| r346 | #def write_device(self, raw=False): | |||
| r299 | ||||
|
Juan C. Espinoza
|
r328 | if not raw: | ||
| clock = RCClock.objects.get(rc_configuration=self) | ||||
| r345 | print(clock) | |||
|
Juan C. Espinoza
|
r328 | if clock.mode: | ||
| data = {'default': clock.frequency} | ||||
| else: | ||||
| data = {'manual': [clock.multiplier, clock.divisor, clock.reference]} | ||||
| payload = self.request('setfreq', 'post', data=json.dumps(data)) | ||||
| r345 | print(payload) | |||
| r335 | if payload['command'] != 'ok': | |||
| r346 | self.message = 'Pedestal write: {}'.format(payload['command']) | |||
|
Juan C. Espinoza
|
r328 | else: | ||
| self.message = payload['programming'] | ||||
| if payload['programming'] == 'fail': | ||||
| r346 | self.message = 'Pedestal write: error programming CGS chip' | |||
|
Juan C. Espinoza
|
r328 | |||
| r299 | values = [] | |||
| for pulse, delay in zip(self.get_pulses(), self.get_delays()): | ||||
| r302 | while delay>65536: | |||
| r299 | values.append((pulse, 65535)) | |||
| r302 | delay -= 65536 | |||
| r299 | values.append((pulse, delay-1)) | |||
|
Juan C. Espinoza
|
r236 | data = bytearray() | ||
|
Fiorella Quino
|
r243 | #reset | ||
| data.extend((128, 0)) | ||||
| #disable | ||||
|
Juan C. Espinoza
|
r236 | data.extend((129, 0)) | ||
| r304 | #SW switch | |||
| if self.control_sw: | ||||
| data.extend((130, 2)) | ||||
| else: | ||||
| data.extend((130, 0)) | ||||
|
Fiorella Quino
|
r243 | #divider | ||
| data.extend((131, self.clock_divider-1)) | ||||
|
Juan C. Espinoza
|
r236 | #enable writing | ||
| data.extend((139, 62)) | ||||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r236 | last = 0 | ||
|
Fiorella Quino
|
r243 | for tup in values: | ||
| vals = pack('<HH', last^tup[0], tup[1]) | ||||
|
Juan C. Espinoza
|
r236 | last = tup[0] | ||
|
Fiorella Quino
|
r243 | data.extend((133, vals[1], 132, vals[0], 133, vals[3], 132, vals[2])) | ||
|
Juan C. Espinoza
|
r236 | #enable | ||
| data.extend((129, 1)) | ||||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r316 | if raw: | ||
| return b64encode(data) | ||||
|
Juan C. Espinoza
|
r185 | try: | ||
| r299 | payload = self.request('stop', 'post') | |||
| payload = self.request('reset', 'post') | ||||
| #payload = self.request('divider', 'post', data={'divider': self.clock_divider-1}) | ||||
| #payload = self.request('write', 'post', data=b64encode(bytearray((139, 62))), timeout=20) | ||||
| n = len(data) | ||||
| x = 0 | ||||
| #while x < n: | ||||
|
Fiorella Quino
|
r264 | payload = self.request('write', 'post', data=b64encode(data)) | ||
| r299 | # x += 1024 | |||
|
Juan C. Espinoza
|
r222 | if payload['write']=='ok': | ||
|
Fiorella Quino
|
r264 | self.device.status = 3 | ||
|
Juan C. Espinoza
|
r185 | self.device.save() | ||
| r346 | self.message = 'Pedestal configured and started' | |||
|
Juan C. Espinoza
|
r185 | else: | ||
|
Juan C. Espinoza
|
r236 | self.device.status = 1 | ||
|
Juan C. Espinoza
|
r185 | self.device.save() | ||
| r346 | self.message = 'Pedestal write: {}'.format(payload['write']) | |||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r243 | |||
| r299 | #payload = self.request('start', 'post') | |||
|
Juan C. Espinoza
|
r185 | except Exception as e: | ||
|
Juan C. Espinoza
|
r222 | if 'No route to host' not in str(e): | ||
| self.device.status = 4 | ||||
| else: | ||||
| self.device.status = 0 | ||||
| r346 | self.message = 'Pedestal write: {}'.format(str(e)) | |||
|
Juan C. Espinoza
|
r222 | self.device.save() | ||
|
Juan C. Espinoza
|
r185 | return False | ||
|
Fiorella Quino
|
r243 | |||
|
Juan C. Espinoza
|
r185 | return True | ||
|
Juan C. Espinoza
|
r172 | |||
|
Juan C. Espinoza
|
r107 | |||
|
Fiorella Quino
|
r264 | def get_absolute_url_import(self): | ||
| r349 | return reverse('url_import_pedestal_conf', args=[str(self.id)]) | |||
