63 lines
1.4 KiB
Python
63 lines
1.4 KiB
Python
# Copyright 2020-2021 Ternaris.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
"""CLI tool for rosbag conversion."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING
|
|
|
|
from .converter import ConverterError, convert
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Callable
|
|
|
|
|
|
def pathtype(exists: bool = True) -> Callable:
|
|
"""Path argument for argparse.
|
|
|
|
Args:
|
|
exists: Path should exists in filesystem.
|
|
|
|
Returns:
|
|
Argparse type function.
|
|
|
|
"""
|
|
|
|
def topath(pathname: str) -> Path:
|
|
path = Path(pathname)
|
|
if exists != path.exists():
|
|
raise argparse.ArgumentTypeError(
|
|
f'{path} should {"exist" if exists else "not exist"}.',
|
|
)
|
|
return path
|
|
|
|
return topath
|
|
|
|
|
|
def main() -> None:
|
|
"""Parse cli arguments and run conversion."""
|
|
parser = argparse.ArgumentParser(description='Convert rosbag1 to rosbag2.')
|
|
parser.add_argument(
|
|
'src',
|
|
type=pathtype(),
|
|
help='source path to read rosbag1 from',
|
|
)
|
|
parser.add_argument(
|
|
'--dst',
|
|
type=pathtype(exists=False),
|
|
help='destination path for rosbag2',
|
|
)
|
|
args = parser.parse_args()
|
|
try:
|
|
convert(args.src, args.dst)
|
|
except ConverterError as err:
|
|
print(f'ERROR: {err}') # noqa: T001
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|