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

Add typename guessing to examples

Browse Source
This commit is contained in:
Marko Durkovic 2022-05-25 15:39:47 +02:00
parent abc1acfecb
commit 0bf94a6238
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
docs/examples/register_types_files.py
View File

@ -4,21 +4,32 @@ from pathlib import Path
from rosbags.typesys import get_types_from_msg, register_types
def guess_msgtype(path: Path) -> str:
"""Guess message type name from path."""
name = path.relative_to(path.parents[2]).with_suffix('')
if 'msg' not in name.parts:
name = name.parent / 'msg' / name.name
return str(name)
add_types = {}
msgdef = Path('/path/to/custom_msgs/msg/Speed.msg').read_text(encoding='utf-8')
add_types.update(get_types_from_msg(msgdef, 'custom_msgs/msg/Speed.msg'))
msgdef = Path('/path/to/custom_msgs/msg/Accel.msg').read_text(encoding='utf-8')
add_types.update(get_types_from_msg(msgdef, 'custom_msgs/msg/Accel.msg'))
for pathstr in [
'/path/to/custom_msgs/msg/Speed.msg',
'/path/to/custom_msgs/msg/Accel.msg',
]:
msgpath = Path(pathstr)
msgdef = msgpath.read_text(encoding='utf-8')
add_types.update(get_types_from_msg(msgdef, guess_msgtype(msgpath)))
register_types(add_types)
# Type import works only after the register_types call,
# the classname is derived from the msgtype names above
# the classname is derived from the msgtype names above.
# pylint: disable=no-name-in-module,wrong-import-position
from rosbags.typesys.types import custom_msgs__msg__Speed as Speed # type: ignore # noqa
from rosbags.typesys.types import custom_msgs__msg__Accel as Accel # type: ignore # noqa
from rosbags.typesys.types import custom_msgs__msg__Speed as Speed # type: ignore # noqa
# pylint: enable=no-name-in-module,wrong-import-position