pylib/tcp_socket
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Channel logging implemented

Browse Source
This commit is contained in:
Dirk Alders 2020-12-25 16:52:02 +01:00
parent 576494b200
commit 7146692712
1 changed files with 38 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
__init__.py
View File

@ -54,7 +54,8 @@ class tcp_base(object):
.. note:: This class is not designed for direct usage.
"""
LOG_PREFIX = 'TCP_IP:'
DEFAULT_CHANNEL_NAME = 'all_others'
RX_LENGTH = 0xff
COM_TIMEOUT = 0.5
IS_CLIENT = False
@ -62,6 +63,7 @@ class tcp_base(object):
def __init__(self, host, port, rx_log_lvl=logging.INFO):
self.host = host
self.port = port
self.init_channel_name()
self.__socket__ = None
self.__data_available_callback__ = None
self.__supress_data_available_callback__ = False
@ -75,6 +77,12 @@ class tcp_base(object):
self.__queue__.enqueue(5, self.__receive_task__, rx_log_lvl)
self.__queue__.run()
def init_channel_name(self, channel_name=None):
if channel_name is None:
self.logger = logging.getLogger(ROOT_LOGGER_NAME).getChild(__name__ + '.' + self.DEFAULT_CHANNEL_NAME)
else:
self.logger = logging.getLogger(ROOT_LOGGER_NAME).getChild(__name__ + '.' + channel_name)
def is_connected(self):
return self.__connection__ is not None
@ -105,7 +113,7 @@ class tcp_base(object):
if self.__connection__ is None:
return None
if time.time() > tm + timeout:
logger.warning('%s TIMEOUT (%ss): Not enough data in buffer. Requested %s and buffer size is %d.', self.LOG_PREFIX, repr(timeout), repr(num or 'all'), len(self.__receive_buffer__))
self.logger.warning('TIMEOUT (%ss): Not enough data in buffer. Requested %s and buffer size is %d.', repr(timeout), repr(num or 'all'), len(self.__receive_buffer__))
return None
time.sleep(0.05)
if num is None:
@ -129,16 +137,23 @@ class tcp_base(object):
This method sends data via the initiated communication channel.
"""
cnt = 0
while self.__connection__ is None and cnt < int(10 * timeout):
time.sleep(.1) # give some time to establish the connection
cnt += 1
tm = time.time()
while time.time() - tm < timeout:
if self.__connection__ is not None:
try:
self.__connection__.sendall(data)
logger.log(log_lvl, '%s TX -> "%s"', self.LOG_PREFIX, stringtools.hexlify(data))
except BlockingIOError:
time.sleep(.1) # try again till timeout exceeds
except BrokenPipeError:
self.logger.exception('Exception while sending data')
self.__connection_lost__()
return False
else:
self.logger.log(log_lvl, 'TX -> "%s"', stringtools.hexlify(data))
return True
else:
logger.warning('%s Cound NOT send -> "%s"', self.LOG_PREFIX, stringtools.hexlify(data))
time.sleep(.1) # give some time to establish the connection
self.logger.warning('Cound NOT send -> "%s"', stringtools.hexlify(data))
return False
def register_callback(self, callback):
@ -184,7 +199,7 @@ class tcp_base(object):
time.sleep(.05)
else:
if len(data) > 0:
logger.log(rx_log_lvl, '%s RX <- "%s"', self.LOG_PREFIX, stringtools.hexlify(data))
self.logger.log(rx_log_lvl, 'RX <- "%s"', stringtools.hexlify(data))
self.__receive_buffer__ += data
else:
self.__connection_lost__()
@ -204,12 +219,12 @@ class tcp_base(object):
self.__connection__.close()
self.__connection__ = None
self.__client_address__ = None
logger.info('%s Connection lost...', self.LOG_PREFIX)
self.logger.info('Connection lost...')
if self.__disconnect_callback__ is not None:
self.__disconnect_callback__()
def __clean_receive_buffer__(self):
logger.debug("%s Cleaning up receive-buffer", self.LOG_PREFIX)
self.logger.debug("Cleaning up receive-buffer")
self.__receive_buffer__ = ""
def close(self):
@ -258,7 +273,7 @@ class tcp_server(tcp_base):
self.__socket__.settimeout(self.COM_TIMEOUT)
self.__socket__.setblocking(False)
if not self.__listening_message_displayed__:
logger.info('%s Server listening to %s:%d', self.LOG_PREFIX, self.host, self.port)
self.logger.info('Server listening to %s:%d', self.host, self.port)
self.__listening_message_displayed__ = True
try:
self.__connection__, self.__client_address__ = self.__socket__.accept()
@ -268,7 +283,7 @@ class tcp_server(tcp_base):
else:
time.sleep(.05)
else:
logger.info('%s Connection established... (from %s)', self.LOG_PREFIX, self.client_address())
self.logger.info('Connection established... (from %s)', self.client_address())
self.__clean_receive_buffer__()
self.__connection__.setblocking(False)
if self.__connect_callback__ is not None:
@ -313,7 +328,7 @@ class tcp_client(tcp_base):
self.__connection__ = None
time.sleep(.05)
else:
logger.info('%s Connection to %s:%s established', self.LOG_PREFIX, self.host, self.port)
self.logger.info('Connection to %s:%s established', self.host, self.port)
self.__clean_receive_buffer__()
self.__connection__ = self.__socket__
if self.__connect_callback__ is not None:
@ -346,7 +361,7 @@ class tcp_base_stp(tcp_base):
self.__stp__ = stringtools.stp.stp()
def __clean_receive_buffer__(self):
logger.debug("%s Cleaning up receive-buffer", self.LOG_PREFIX)
self.logger.debug("Cleaning up receive-buffer")
self.__receive_buffer__ = []
def receive(self, timeout=1):
@ -377,7 +392,7 @@ class tcp_base_stp(tcp_base):
This method sends one message via the initiated communication channel.
"""
if tcp_base.send(self, stringtools.stp.build_frame(data), timeout=timeout, log_lvl=logging.DEBUG):
logger.log(log_lvl, '%s TX -> "%s"', self.LOG_PREFIX, stringtools.hexlify(data))
self.logger.log(log_lvl, 'TX -> "%s"', stringtools.hexlify(data))
return True
else:
return False
@ -395,10 +410,10 @@ class tcp_base_stp(tcp_base):
time.sleep(.05)
else:
if len(data) > 0:
logger.debug('%s -- <- "%s"', self.LOG_PREFIX, stringtools.hexlify(data))
self.logger.debug('-- <- "%s"', stringtools.hexlify(data))
content = self.__stp__.process(data)
for msg in content:
logger.log(rx_log_lvl, '%s RX <- "%s"', self.LOG_PREFIX, stringtools.hexlify(msg))
self.logger.log(rx_log_lvl, 'RX <- "%s"', stringtools.hexlify(msg))
self.__receive_buffer__.append(msg)
else:
self.__connection_lost__()