Update colors.py

0xInfection · Nov 18, 2018 · 320c312 · 320c312
1 parent cbe2861
commit 320c312
Showing 1 changed file with 26 additions and 21 deletions.
diff --git a/core/colors.py b/core/colors.py
@@ -9,25 +9,30 @@
 # This module requires XSRFProbe
 # https://github.com/0xInfection/XSRFProbe
 
-###############################
-class color:
-    END  = '\033[0m'                # normal
-    BOLD  = '\033[1m'               # bold
-    RED  = '\033[1;91m'             # red
-    GREEN  = '\033[1;92m'           # green
-    ORANGE  = '\033[1;93m'          # orange
-    BLUE  = '\033[1;94m'            # blue
-    PURPLE  = '\033[1;95m'          # purple
-    UNDERLINE = '\033[4m'           # underline
-    CYAN  = '\033[1;96m'            # cyan
-    GREY = '\033[1;97m'             # gray
-    BR = '\033[1;97;41m'            # background red
-    BG = '\033[1;97;42m'            # background green
-    BY = '\033[1;97;43m'            # background yellow
+import os
+if os.name != 'nt' and os.name != 'mac':
+    class color:
+        END  = '\033[0m'                # normal
+        BOLD  = '\033[1m'               # bold
+        RED  = '\033[1;91m'             # red
+        GREEN  = '\033[1;92m'           # green
+        ORANGE  = '\033[1;93m'          # orange
+        BLUE  = '\033[1;94m'            # blue
+        PURPLE  = '\033[1;95m'          # purple
+        UNDERLINE = '\033[4m'           # underline
+        CYAN  = '\033[1;96m'            # cyan
+        GREY = '\033[1;97m'             # gray
+        BR = '\033[1;97;41m'            # background red
+        BG = '\033[1;97;42m'            # background green
+        BY = '\033[1;97;43m'            # background yellow
 
-O  = '\033[1m \033[93m[!]\033[0m '  # information
-R  = '\033[1m \033[91m[-]\033[0m '  # something's not right
-GR = '\033[1m \033[97m[*]\033[0m '  # processing
-G  = '\033[1m \033[92m[+]\033[0m '  # yay!
-C  = '\033[1m \033[96m[+]\033[0m '  # crawling...
-###############################
+    O  = '\033[1m \033[93m[!]\033[0m '  # information
+    R  = '\033[1m \033[91m[-]\033[0m '  # something's not right
+    GR = '\033[1m \033[97m[*]\033[0m '  # processing
+    G  = '\033[1m \033[92m[+]\033[0m '  # yay!
+    C  = '\033[1m \033[96m[+]\033[0m '  # crawling...
+
+else:
+    class color:
+        END=BOLD=RED=GREEN=ORANGE=BLUE=PURPLE=UNDERLINE=CYAN=GREY=BR=BG=BY=''  # no escape sequences 
+    O = R = GR = G = C = ''  # no color values