From 3de197c765405a3e4f6c24c643e806d6cdb7bb09 Mon Sep 17 00:00:00 2001 From: EsipovPA Date: Mon, 18 Jul 2022 01:52:34 +0300 Subject: [PATCH 1/2] Set to list: writer_choices: record.py Signed-off-by: EsipovPA --- ros2bag/ros2bag/verb/record.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ros2bag/ros2bag/verb/record.py b/ros2bag/ros2bag/verb/record.py index 021cd5ab9..26bf94f1a 100644 --- a/ros2bag/ros2bag/verb/record.py +++ b/ros2bag/ros2bag/verb/record.py @@ -35,7 +35,8 @@ class RecordVerb(VerbExtension): def add_arguments(self, parser, cli_name): # noqa: D102 writer_choices = get_registered_writers() - default_writer = 'sqlite3' if 'sqlite3' in writer_choices else writer_choices[0] + default_writer = 'sqlite3' if 'sqlite3' in writer_choices else \ + next(iter(writer_choices)) compression_format_choices = get_registered_compressors() serialization_choices = get_registered_serializers() From 1f0259cb5156fac35a77d862494ede2b0e7101ee Mon Sep 17 00:00:00 2001 From: EsipovPA Date: Mon, 18 Jul 2022 01:52:58 +0300 Subject: [PATCH 2/2] Set to list: storage_choices: reindex.py Signed-off-by: EsipovPA --- ros2bag/ros2bag/verb/reindex.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ros2bag/ros2bag/verb/reindex.py b/ros2bag/ros2bag/verb/reindex.py index 83ccca4c5..7a1ca6fc7 100644 --- a/ros2bag/ros2bag/verb/reindex.py +++ b/ros2bag/ros2bag/verb/reindex.py @@ -33,7 +33,9 @@ class ReindexVerb(VerbExtension): def add_arguments(self, parser, cli_name): storage_choices = get_registered_readers() - default_storage = 'sqlite3' if 'sqlite3' in storage_choices else storage_choices[0] + default_storage = 'sqlite3' if 'sqlite3' in storage_choices else \ + next(iter(storage_choices)) + parser.add_argument( 'bag_directory', type=check_path_exists, help='bag to reindex') parser.add_argument(