6332976c4a
* hotfix: don't send two MODBUS commands together * fix unit tests * remove read loop * optional sleep between msg read and sending rsp * wait after read 0.5s before sending a response * add pending state * fix state definitions * determine the connection timeout by the conn state * avoid sending MODBUS cmds in the inverter's reporting phase * update changelog
123 lines
3.2 KiB
Python
123 lines
3.2 KiB
Python
import logging
|
|
import weakref
|
|
from typing import Callable, Generator
|
|
from enum import Enum
|
|
|
|
|
|
if __name__ == "app.src.messages":
|
|
from app.src.infos import Infos
|
|
from app.src.modbus import Modbus
|
|
else: # pragma: no cover
|
|
from infos import Infos
|
|
from modbus import Modbus
|
|
|
|
logger = logging.getLogger('msg')
|
|
|
|
|
|
def hex_dump_memory(level, info, data, num):
|
|
n = 0
|
|
lines = []
|
|
lines.append(info)
|
|
tracer = logging.getLogger('tracer')
|
|
if not tracer.isEnabledFor(level):
|
|
return
|
|
|
|
for i in range(0, num, 16):
|
|
line = ' '
|
|
line += '%04x | ' % (i)
|
|
n += 16
|
|
|
|
for j in range(n-16, n):
|
|
if j >= len(data):
|
|
break
|
|
line += '%02x ' % abs(data[j])
|
|
|
|
line += ' ' * (3 * 16 + 9 - len(line)) + ' | '
|
|
|
|
for j in range(n-16, n):
|
|
if j >= len(data):
|
|
break
|
|
c = data[j] if not (data[j] < 0x20 or data[j] > 0x7e) else '.'
|
|
line += '%c' % c
|
|
|
|
lines.append(line)
|
|
|
|
tracer.log(level, '\n'.join(lines))
|
|
|
|
|
|
class IterRegistry(type):
|
|
def __iter__(cls) -> Generator['Message', None, None]:
|
|
for ref in cls._registry:
|
|
obj = ref()
|
|
if obj is not None:
|
|
yield obj
|
|
|
|
|
|
class State(Enum):
|
|
'''state of the logical connection'''
|
|
init = 0
|
|
'''just created'''
|
|
received = 1
|
|
'''at least one packet received'''
|
|
up = 2
|
|
'''at least one cmd-rsp transaction'''
|
|
pend = 3
|
|
'''inverter transaction pending, don't send MODBUS cmds'''
|
|
closed = 4
|
|
'''connection closed'''
|
|
|
|
|
|
class Message(metaclass=IterRegistry):
|
|
_registry = []
|
|
|
|
def __init__(self, server_side: bool, send_modbus_cb:
|
|
Callable[[bytes, int, str], None], mb_timeout: int):
|
|
self._registry.append(weakref.ref(self))
|
|
|
|
self.server_side = server_side
|
|
if server_side:
|
|
self.mb = Modbus(send_modbus_cb, mb_timeout)
|
|
else:
|
|
self.mb = None
|
|
|
|
self.header_valid = False
|
|
self.header_len = 0
|
|
self.data_len = 0
|
|
self.unique_id = 0
|
|
self.node_id = '' # will be overwritten in the child class's __init__
|
|
self.sug_area = ''
|
|
self._recv_buffer = bytearray(0)
|
|
self._send_buffer = bytearray(0)
|
|
self._forward_buffer = bytearray(0)
|
|
self.new_data = {}
|
|
self.state = State.init
|
|
|
|
'''
|
|
Empty methods, that have to be implemented in any child class which
|
|
don't use asyncio
|
|
'''
|
|
def _read(self) -> None: # read data bytes from socket and copy them
|
|
# to our _recv_buffer
|
|
return # pragma: no cover
|
|
|
|
def _update_header(self, _forward_buffer):
|
|
'''callback for updating the header of the forward buffer'''
|
|
return # pragma: no cover
|
|
|
|
'''
|
|
Our puplic methods
|
|
'''
|
|
def close(self) -> None:
|
|
if self.mb:
|
|
self.mb.close()
|
|
self.mb = None
|
|
pass # pragma: no cover
|
|
|
|
def inc_counter(self, counter: str) -> None:
|
|
self.db.inc_counter(counter)
|
|
Infos.new_stat_data['proxy'] = True
|
|
|
|
def dec_counter(self, counter: str) -> None:
|
|
self.db.dec_counter(counter)
|
|
Infos.new_stat_data['proxy'] = True
|