fly316/PX4-Autopilot
7
0
Fork 0
mirror of https://gitee.com/mirrors_PX4/PX4-Autopilot.git synced 2026-11-03 04:01:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

uorb_rtps_classifier: improve way to check base type of alias

Browse Source
This commit is contained in:
TSC21
2019-08-25 23:04:09 +01:00
committed by Nuno Marques
parent ac6ee972d3
commit 062b693fea
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msg/tools/uorb_rtps_classifier.py
+14 -8
View File
@@ -144,19 +144,25 @@ class Classifier():
""" """
Check if alias message has correct base type Check if alias message has correct base type
""" """
rtps_registered_msgs = list( registered_alias_msgs = list(
dict['alias'] for dict in self.msg_id_map['rtps'] if 'alias' in dict.keys()) dict['alias'] for dict in self.msg_id_map['rtps'] if 'alias' in dict.keys())
uorb_msg = list(msg for msg in self.all_msgs_list)
incorrect_base_types = list(set(rtps_registered_msgs) - set(uorb_msg))
base_types = {} base_types = []
for dict in self.msg_id_map['rtps']:
if 'alias' not in dict.keys():
base_types.append(dict['msg'])
incorrect_base_types = list(
set(registered_alias_msgs) - set(base_types))
base_types_suggestion = {}
for incorrect in incorrect_base_types: for incorrect in incorrect_base_types:
base_types.update({incorrect: difflib.get_close_matches( base_types_suggestion.update({incorrect: difflib.get_close_matches(
incorrect, uorb_msg, n=1, cutoff=0.6)}) incorrect, base_types, n=1, cutoff=0.6)})
if len(base_types) > 0: if len(base_types_suggestion) > 0:
raise AssertionError( raise AssertionError(
('\n' + '\n'.join('\t- The multi-topic message base type {} does not exist. Did you mean \'{}\'?'.format(k, v[0]) for k, v in base_types.items()))) ('\n' + '\n'.join('\t- The multi-topic message base type {} does not exist.{}'.format(k, (' Did you mean \'' + v[0] + '\'?' if v else '')) for k, v in base_types_suggestion.items())))
@staticmethod @staticmethod
def parse_yaml_msg_id_file(yaml_file): def parse_yaml_msg_id_file(yaml_file):