no color for windows shell (#22)

cle-b · Sep 21, 2024 · 1d2ec45 · 1d2ec45
1 parent 5343601
commit 1d2ec45
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/lymbo/color.py b/lymbo/color.py
@@ -1,4 +1,12 @@
-RED = "\033[31m"
-GREEN = "\033[32m"
-YELLOW = "\033[33m"
-RESET = "\033[0m"
+import os
+
+if os.name == "nt":
+    RED = ""
+    GREEN = ""
+    YELLOW = ""
+    RESET = ""
+else:
+    RED = "\033[31m"
+    GREEN = "\033[32m"
+    YELLOW = "\033[33m"
+    RESET = "\033[0m"