🐛 Make sure rich_markup_mode=None disables Rich formatting

tests/test_others.py

@@ -208,27 +208,6 @@ def main(name: str = typer.Option(..., autocompletion=name_callback)):
     assert exc_info.value.message == "Invalid autocompletion callback parameters: val2"
 
 
-def test_forward_references():
-    app = typer.Typer()
-
-    @app.command()
-    def main(arg1, arg2: int, arg3: "int", arg4: bool = False, arg5: "bool" = False):
-        print(f"arg1: {type(arg1)} {arg1}")
-        print(f"arg2: {type(arg2)} {arg2}")
-        print(f"arg3: {type(arg3)} {arg3}")
-        print(f"arg4: {type(arg4)} {arg4}")
-        print(f"arg5: {type(arg5)} {arg5}")
-
-    result = runner.invoke(app, ["Hello", "2", "invalid"])
-
-    assert "Invalid value for 'ARG3': 'invalid' is not a valid integer" in result.stdout
-    result = runner.invoke(app, ["Hello", "2", "3", "--arg4", "--arg5"])
-    assert (
-        "arg1: <class 'str'> Hello\narg2: <class 'int'> 2\narg3: <class 'int'> 3\narg4: <class 'bool'> True\narg5: <class 'bool'> True\n"
-        in result.stdout
-    )
-
-
 def test_context_settings_inheritance_single_command():
     app = typer.Typer(context_settings={"help_option_names": ["-h", "--help"]})
 

tests/test_rich_markup_mode.py

@@ -0,0 +1,116 @@
+import typer
+import typer.completion
+from typer.testing import CliRunner
+
+runner = CliRunner()
+rounded = ["╭", "─", "┬", "╮", "│", "├", "┼", "┤", "╰", "┴", "╯"]
+
+
+def test_rich_markup_mode_default():
+    app = typer.Typer()
+
+    @app.command()
+    def main(arg: str):
+        """Main function"""
+        print(f"Hello {arg}")
+
+    assert app.rich_markup_mode is None
+
+    result = runner.invoke(app, ["World"])
+    assert "Hello World" in result.stdout
+
+    result = runner.invoke(app, ["--help"])
+    assert all(c not in result.stdout for c in rounded)
+
+
+def test_rich_markup_mode_rich():
+    app = typer.Typer(rich_markup_mode="rich")
+
+    @app.command()
+    def main(arg: str):
+        """Main function"""
+        print(f"Hello {arg}")
+
+    assert app.rich_markup_mode == "rich"
+
+    result = runner.invoke(app, ["World"])
+    assert "Hello World" in result.stdout
+
+    result = runner.invoke(app, ["--help"])
+    assert any(c in result.stdout for c in rounded)
+
+
+# Test mostly for coverage
+def test_clickexception_rich():
+    app = typer.Typer(rich_markup_mode="rich")
+
+    @app.command()
+    def main(arg1, arg2: int, arg3: "int", arg4: bool = False, arg5: "bool" = False):
+        print(f"arg1: {type(arg1)} {arg1}")
+        print(f"arg2: {type(arg2)} {arg2}")
+        print(f"arg3: {type(arg3)} {arg3}")
+        print(f"arg4: {type(arg4)} {arg4}")
+        print(f"arg5: {type(arg5)} {arg5}")
+
+    result = runner.invoke(app, ["Hello", "2", "invalid"])
+
+    assert "Invalid value for 'ARG3': 'invalid' is not a valid integer" in result.stdout
+    print(result.stdout)
+    result = runner.invoke(app, ["Hello", "2", "3", "--arg4", "--arg5"])
+    assert (
+        "arg1: <class 'str'> Hello\narg2: <class 'int'> 2\narg3: <class 'int'> 3\narg4: <class 'bool'> True\narg5: <class 'bool'> True\n"
+        in result.stdout
+    )
+    print(result.stdout)
+
+
+# Test mostly for coverage
+def test_clickexception_no_rich():
+    app = typer.Typer(rich_markup_mode=None)
+
+    @app.command()
+    def main(arg1, arg2: int, arg3: "int", arg4: bool = False, arg5: "bool" = False):
+        print(f"arg1: {type(arg1)} {arg1}")
+        print(f"arg2: {type(arg2)} {arg2}")
+        print(f"arg3: {type(arg3)} {arg3}")
+        print(f"arg4: {type(arg4)} {arg4}")
+        print(f"arg5: {type(arg5)} {arg5}")
+
+    result = runner.invoke(app, ["Hello", "2", "invalid"])
+
+    assert "Invalid value for 'ARG3': 'invalid' is not a valid integer" in result.stdout
+    print(result.stdout)
+    result = runner.invoke(app, ["Hello", "2", "3", "--arg4", "--arg5"])
+    assert (
+        "arg1: <class 'str'> Hello\narg2: <class 'int'> 2\narg3: <class 'int'> 3\narg4: <class 'bool'> True\narg5: <class 'bool'> True\n"
+        in result.stdout
+    )
+    print(result.stdout)
+
+
+# Test mostly for coverage
+def test_aborted_rich():
+    from docs_src.multiple_values.options_with_multiple_values import tutorial001 as mod
+
+    runner = CliRunner()
+    app = typer.Typer(rich_markup_mode="rich")
+    app.command()(mod.main)
+
+    result = runner.invoke(app)
+    assert result.exit_code != 0
+    assert "No user provided" in result.output
+    assert "Aborted" in result.output
+
+
+# Test mostly for coverage
+def test_aborted_no_rich():
+    from docs_src.multiple_values.options_with_multiple_values import tutorial001 as mod
+
+    runner = CliRunner()
+    app = typer.Typer(rich_markup_mode=None)
+    app.command()(mod.main)
+
+    result = runner.invoke(app)
+    assert result.exit_code != 0
+    assert "No user provided" in result.output
+    assert "Aborted" in result.output

tests/test_tutorial/test_arguments/test_help/test_tutorial006.py

@@ -1,3 +1,4 @@
+import os
 import subprocess
 import sys
 
@@ -32,5 +33,6 @@ def test_script():
         [sys.executable, "-m", "coverage", "run", mod.__file__, "--help"],
         capture_output=True,
         encoding="utf-8",
+        env={**os.environ, "PYTHONIOENCODING": "utf-8"},
     )
     assert "Usage" in result.stdout

tests/test_tutorial/test_arguments/test_help/test_tutorial006_an.py

@@ -1,3 +1,4 @@
+import os
 import subprocess
 import sys
 
@@ -32,5 +33,6 @@ def test_script():
         [sys.executable, "-m", "coverage", "run", mod.__file__, "--help"],
         capture_output=True,
         encoding="utf-8",
+        env={**os.environ, "PYTHONIOENCODING": "utf-8"},
     )
     assert "Usage" in result.stdout

tests/test_tutorial/test_arguments/test_help/test_tutorial007.py

@@ -9,7 +9,7 @@
 
 runner = CliRunner()
 
-app = typer.Typer()
+app = typer.Typer(rich_markup_mode="rich")
 app.command()(mod.main)
 
 

tests/test_tutorial/test_arguments/test_help/test_tutorial007_an.py

@@ -9,7 +9,7 @@
 
 runner = CliRunner()
 
-app = typer.Typer()
+app = typer.Typer(rich_markup_mode="rich")
 app.command()(mod.main)
 
 

tests/test_tutorial/test_commands/test_help/test_tutorial003.py

@@ -6,6 +6,7 @@
 from docs_src.commands.help import tutorial003 as mod
 
 app = mod.app
+app.rich_markup_mode = "rich"
 
 runner = CliRunner()
 

tests/test_tutorial/test_options/test_help/test_tutorial002.py

@@ -8,7 +8,7 @@
 
 runner = CliRunner()
 
-app = typer.Typer()
+app = typer.Typer(rich_markup_mode="rich")
 app.command()(mod.main)
 
 

tests/test_tutorial/test_options/test_help/test_tutorial002_an.py

@@ -8,7 +8,7 @@
 
 runner = CliRunner()
 
-app = typer.Typer()
+app = typer.Typer(rich_markup_mode="rich")
 app.command()(mod.main)
 
 

typer/core.py

@@ -167,6 +167,7 @@ def _main(
     complete_var: Optional[str] = None,
     standalone_mode: bool = True,
     windows_expand_args: bool = True,
+    rich_markup_mode: MarkupMode = None,
     **extra: Any,
 ) -> Any:
     # Typer override, duplicated from click.main() to handle custom rich exceptions
@@ -208,7 +209,7 @@ def _main(
             if not standalone_mode:
                 raise
             # Typer override
-            if rich:
+            if rich and rich_markup_mode is not None:
                 rich_utils.rich_format_error(e)
             else:
                 e.show()
@@ -238,7 +239,7 @@ def _main(
         if not standalone_mode:
             raise
         # Typer override
-        if rich:
+        if rich and rich_markup_mode is not None:
             rich_utils.rich_abort_error()
         else:
             click.echo(_("Aborted!"), file=sys.stderr)
@@ -656,6 +657,7 @@ def main(
         complete_var: Optional[str] = None,
         standalone_mode: bool = True,
         windows_expand_args: bool = True,
+        rich_markup_mode: MarkupMode = None,
         **extra: Any,
     ) -> Any:
         return _main(
@@ -665,11 +667,12 @@ def main(
             complete_var=complete_var,
             standalone_mode=standalone_mode,
             windows_expand_args=windows_expand_args,
+            rich_markup_mode=rich_markup_mode,
             **extra,
         )
 
     def format_help(self, ctx: click.Context, formatter: click.HelpFormatter) -> None:
-        if not rich:
+        if not rich or self.rich_markup_mode is None:
             return super().format_help(ctx, formatter)
         return rich_utils.rich_format_help(
             obj=self,
@@ -718,6 +721,7 @@ def main(
         complete_var: Optional[str] = None,
         standalone_mode: bool = True,
         windows_expand_args: bool = True,
+        rich_markup_mode: MarkupMode = None,
         **extra: Any,
     ) -> Any:
         return _main(
@@ -727,11 +731,12 @@ def main(
             complete_var=complete_var,
             standalone_mode=standalone_mode,
             windows_expand_args=windows_expand_args,
+            rich_markup_mode=rich_markup_mode,
             **extra,
         )
 
     def format_help(self, ctx: click.Context, formatter: click.HelpFormatter) -> None:
-        if not rich:
+        if not rich or self.rich_markup_mode is None:
             return super().format_help(ctx, formatter)
         return rich_utils.rich_format_help(
             obj=self,