apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add string constant support to msg parser

Browse Source
This commit is contained in:
Marko Durkovic 2021-10-03 06:28:02 +02:00
parent 203131b777
commit 1333cf1168
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/rosbags/typesys/msg.py
View File

@ -44,7 +44,8 @@ comment
= r'#[^\n]*'
const_dcl
= type_spec identifier '=' integer_literal
= 'string' identifier '=' r'[^\n]+'
/ type_spec identifier '=' integer_literal
field_dcl
= type_spec identifier
@ -172,7 +173,12 @@ class VisitorMSG(Visitor):
def visit_const_dcl(self, children: Any) -> Any:
"""Process const declaration, suppress output."""
return Nodetype.CONST, (children[0][1], children[1][1], children[3])
typ = children[0][1]
if typ == 'string':
value = children[3].strip()
else:
value = children[3]
return Nodetype.CONST, (typ, children[1][1], value)
def visit_specification(self, children: Any) -> Typesdict:
"""Process start symbol."""

3
tests/test_parse.py
View File

@ -18,6 +18,7 @@ MSG = """
# comment
int32 global=42
string str= foo bar\t
std_msgs/Header header
std_msgs/msg/Bool bool
@ -115,7 +116,7 @@ def test_parse_msg():
ret = get_types_from_msg(MSG, 'test_msgs/msg/Foo')
assert 'test_msgs/msg/Foo' in ret
consts, fields = ret['test_msgs/msg/Foo']
assert consts == [('global', 'int32', 42)]
assert consts == [('global', 'int32', 42), ('str', 'string', 'foo bar')]
assert fields[0][0] == 'header'
assert fields[0][1][1] == 'std_msgs/msg/Header'
assert fields[1][0] == 'bool'