ros2#1043 Writer choices set to list (ros2#1044)

* Set to list: writer_choices: record.py Signed-off-by: EsipovPA <esipov.p@mail.ru> * Set to list: storage_choices: reindex.py Signed-off-by: EsipovPA <esipov.p@mail.ru>
DensoADAS · Jul 26, 2022 · 9291bcb · 9291bcb
1 parent 5dc9e7f
commit 9291bcb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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()

diff --git 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(