Fix colored output of tools on Windows (doldecomp#1133)

BR- · Jan 27, 2024 · 37ea54b · 37ea54b
1 parent 1afb3e7
commit 37ea54b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tools/decomp.py b/tools/decomp.py
@@ -136,6 +136,11 @@ def main():
             pyperclip.copy(output)
         if args.print:
             if args.color:
+                try:
+                    import colorama
+                    colorama.just_fix_windows_console()
+                except ModuleNotFoundError:
+                    pass
                 from pygments import highlight
                 from pygments.formatters import TerminalFormatter
                 from pygments.lexers import CLexer

diff --git a/tools/m2ctx/m2ctx.py b/tools/m2ctx/m2ctx.py
@@ -194,6 +194,11 @@ def main():
 
     if not args.quiet:
         if args.colorize:
+            try:
+                import colorama
+                colorama.just_fix_windows_console()
+            except ModuleNotFoundError:
+                pass
             from pygments import highlight
             from pygments.formatters import TerminalFormatter
             from pygments.lexers import CLexer