Fix argument handling for --log-format option

pixee · Oct 18, 2023 · 6e7d936 · 6e7d936
1 parent 826377f
commit 6e7d936
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/codemodder/cli.py b/src/codemodder/cli.py
@@ -139,7 +139,7 @@ def parse_args(argv, codemod_registry):
         "--log-format",
         type=OutputFormat,
         default=OutputFormat.HUMAN,
-        choices=[str(x).split(".")[-1].lower() for x in list(OutputFormat)],
+        choices=[OutputFormat.HUMAN, OutputFormat.JSON],
         help="the format for the log output",
     )
     parser.add_argument(

diff --git a/src/codemodder/logging.py b/src/codemodder/logging.py
@@ -13,6 +13,10 @@ class OutputFormat(Enum):
     HUMAN = "human"
     JSON = "json"
 
+    def __str__(self):
+        """For rendering properly in argparse help."""
+        return self.value.lower()
+
 
 def log_section(section_name: str):
     """