added --color flag to force color even if stdout is a tty

ihabunek · Dec 6, 2023 · b92012d · b92012d
1 parent 1c5abb8
commit b92012d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/toot/console.py b/toot/console.py
@@ -164,6 +164,11 @@ def editor(value):
         "action": 'store_true',
         "default": False,
     }),
+    (["--color"], {
+        "help": "always use ANSI colors in output",
+        "action": 'store_true',
+        "default": False,
+    }),
     (["--quiet"], {
         "help": "don't write to stdout on success",
         "action": 'store_true',

diff --git a/toot/output.py b/toot/output.py
@@ -110,6 +110,11 @@ def use_ansi_color():
     if sys.platform == 'win32' and 'ANSICON' not in os.environ:
         return False
 
+    # With the --color flag, force color, even if stdout is a tty
+    # and even if --no-color is also specified (!)
+    if "--color" in sys.argv:
+        return True
+
     # Don't show color if stdout is not a tty, e.g. if output is piped on
     if not sys.stdout.isatty():
         return False