def __update_modules(self, message):
""" Update module information
:param message: Dictionary format module info
:type message: Dictionary
:return: None
"""
module_id = message['s']
module_uuid, module_version_info = unpack_data(message['b'], (6, 2))
# Handle new modules
if module_id not in (module.id for module in self._modules):
module_type = get_module_type_from_uuid(module_uuid)
self.__request_topology(module_id)
new_module = self.__add_new_module(
module_type, module_id, module_uuid, module_version_info
)
new_module.module_type = module_type
if module_type != 'network' and not new_module.is_up_to_date:
print(f"{str(new_module)} is not up to date. "
f"Please update the module by calling "
f"modi.update_module_firmware")
elif not self.__get_module_by_id(module_id).is_connected:
# Handle Reconnected modules
self.__get_module_by_id(module_id).is_connected = True
module_type = get_module_type_from_uuid(module_uuid)
print(f"{module_type} ({module_id}) has been reconnected!!")
def __update_health(self, message):
""" Update information by health message
:param message: Dictionary format message of the module
:type message: Dictionary
:return: None
"""
# Record battery information and user code state
module_id = message["s"]
_, _, _, battery_state, user_code_state \
= unpack_data(message['b'], (1, 1, 1, 1, 1))
curr_time = time.time()
# Checking starts only when module is registered
if module_id in (module.id for module in self._modules):
module = self.__get_module_by_id(module_id)
module.last_updated = curr_time
module.is_connected = True
# Update user code status
if module.user_code_status < 0:
module.user_code_status = user_code_state % 2
# Turn off pnp if pnp flag is on
if module.module_type != 'network' and user_code_state < 2:
self.__set_module_state(
module_id, Module.RUN, Module.PNP_OFF
)
# Reset disconnection alert status
if module.has_printed:
module.has_printed = False
# Disconnect module with no health message for more than 2 second
for module in self._modules:
if module.module_type != 'network' and \
curr_time - module.last_updated > 2:
# if not module.has_printed:
# print(
# f"{module.module_type.title()} ({module_id}) "
# f"has been disconnected!!"
# )
# module.has_printed = True
module.is_connected = False
module._last_set_message = None
def __update_property(self, message):
""" Update module property
:param message: Dictionary format message
:type message: Dictionary
:return: None
"""
# Do not update reserved property
property_number = message["d"]
if property_number == 0 or property_number == 1:
return
module = self.__get_module_by_id(message['s'])
if not module:
return
data = message['b']
if module.module_type == 'network':
module.update_property(property_number, unpack_data(data)[0])
else:
module.update_property(property_number, decode_data(data))
def __update_topology(self, message):
"""Update the topology of the connected modules
:param message: Dictionary format message of the module
:return: None
"""
# Setup prerequisites
src_id = message["s"]
byte_data = message["b"]
topology_by_id = {}
topology_ids = unpack_data(byte_data, (2, 2, 2, 2))
module = self.__get_module_by_id(src_id)
if module:
topology_by_id['type'] = get_module_type_from_uuid(module.uuid)
else:
topology_by_id['type'] = None
for idx, direction in enumerate(('r', 't', 'l', 'b')):
topology_by_id[direction] = topology_ids[idx] \
if topology_ids[idx] != 0xFFFF else None
# Handle battery module
if topology_ids[idx] == 0:
if 0 not in self._topology_data:
self._modules.append(Battery(0, -1, None))
battery_topology = {
'type': 'Battery',
'r': None,
't': None,
'l': None,
'b': src_id
}
self._topology_data[0] = battery_topology
elif src_id not in self._topology_data[0].values():
self._topology_data[0]['l'] = src_id
# Update topology data for the module
self._topology_data[src_id] = topology_by_id