Merge pull request #26 from asolidtime/main

don't specify dot character in available subtitle formats in the CLI

src/subsai/cli.py

@@ -124,7 +124,7 @@ def main():
                              "string)")
     parser.add_argument('-f', '--format', '--subtitles-format', default='srt',
                         help=f"Output subtitles format, available "
-                             f"formats {available_subs_formats()}")
+                             f"formats {available_subs_formats(include_extensions=False)}")
     parser.add_argument('-df', '--destination-folder', default=None,
                         help='The directory where the subtitles will be stored, default to the same folder where '
                              'the media file(s) is stored.')

src/subsai/utils.py

@@ -48,11 +48,20 @@ def available_translation_models() -> list:
     return models
 
 
-def available_subs_formats():
+def available_subs_formats(include_extensions=True):
     """
     Returns available subtitle formats
     from :attr:`pysubs2.FILE_EXTENSION_TO_FORMAT_IDENTIFIER`
 
+    :param include_extensions: include the dot separator in file extensions
+
     :return: list of subtitle formats
     """
-    return list(FILE_EXTENSION_TO_FORMAT_IDENTIFIER.keys())
+
+    extensions = list(FILE_EXTENSION_TO_FORMAT_IDENTIFIER.keys())
+
+    if include_extensions:
+        return extensions
+    else:
+        # remove the '.' separator from extension names
+        return [ext.split('.')[1] for ext in extensions]