--no-colors option

Author: wvdongen

README.md

@@ -70,6 +70,8 @@ optional arguments:
   --capitalize          Transform 'word' into 'Word'.
   --strip-extension     Strip word extension: word.ext into word
   --alpha               Filter non alphanumeric words from wordlist
+  --no-progress         Don't show tested words and progress. (For dumb terminals)
+  --no-colors           Don't use output colors to keep output clean, e.g. when redirecting output to file
 
 License, requests, etc: https://github.com/deibit/cansina
 ```
@@ -84,7 +86,7 @@ Screenshot
 Installing dependencies
 =======================
 
-pip install --user requests[security] or pip install -r requeriments.txt
+pip install --user requests[security] or pip install -r requirements.txt
 
 (try removing --user and install with sudo in case of errors)
 

cansina.py

@@ -186,6 +186,8 @@ def _make_cookie_jar(_cookies):
         help="Filter non alphanumeric words from wordlist", default=False, action="store_true")
 parser.add_argument('--no-progress', dest='no_progress',
         help="Don't show tested words and progress. (For dumb terminals)", default=False, action="store_true")
+parser.add_argument('--no-colors', dest='no_colors',
+        help="Disable coloring. (For dumb terminals)", default=False, action="store_true")
 
 args = parser.parse_args()
 
@@ -401,6 +403,7 @@ def _make_cookie_jar(_cookies):
 Console.show_content_type = show_content_type
 Console.header()
 Console.set_show_progress(False if args.no_progress else True)
+Console.set_show_colors(False if args.no_colors else True)
 
 #
 # Create the thread_pool and start the daemonized threads

core/printer.py

@@ -88,11 +88,15 @@ class Console:
     number_of_requests = 0
     number_of_threads = 0
     show_progress = True
+    show_colors = True
 
     @staticmethod
     def set_show_progress(show_progress):
         Console.show_progress = show_progress
 
+    def set_show_colors(show_colors):
+        Console.show_colors = show_colors
+
     @staticmethod
     def start_eta_queue(size):
         Console.eta_queue = ETAQueue(size, Console.number_of_requests, Console.number_of_threads)
@@ -122,22 +126,26 @@ def body(task):
         elif os.name == 'nt':
             return
 
+        to_format = "{1: ^3} | {2: >10} | {3: >6} | {4: >4} | {7} [{0: >2}%] - {5: ^9} - {6}"
+        to_format_without_progress = "{0: ^3} | {1: >10} | {2: >6} | {3: >4} | {5:^} {4}"
+
         color = ""
-        if task.response_code == "200":
-            color = GREEN
-        if task.response_code == "401" or task.response_code == "403":
-            color = RED
-        if task.response_code == "404" and (not task.response_code in task.banned_response_codes):
-            color = GRAY
-        if task.response_code == "301" or task.response_code == "302":
-            color = LBLUE
-        if task.response_code.startswith('5') or task.response_code == '400':
-            color = YELLOW
-        if task.content_detected:
-            color = MAGENTA
-
-        to_format = color + "{1: ^3} | {2: >10} | {3: >6} | {4: >4} | {7} [{0: >2}%] - {5: ^9} - {6}" + ENDC
-        to_format_without_progress = color + "{0: ^3} | {1: >10} | {2: >6} | {3: >4} | {5:^} {4}" + ENDC
+        if Console.show_colors:
+            if task.response_code == "200":
+                color = GREEN
+            if task.response_code == "401" or task.response_code == "403":
+                color = RED
+            if task.response_code == "404" and (not task.response_code in task.banned_response_codes):
+                color = GRAY
+            if task.response_code == "301" or task.response_code == "302":
+                color = LBLUE
+            if task.response_code.startswith('5') or task.response_code == '400':
+                color = YELLOW
+            if task.content_detected:
+                color = MAGENTA
+
+            to_format = color + to_format + ENDC
+            to_format_without_progress = color + to_format_without_progress + ENDC
 
         # User wants to see full path
         if Console.show_full_path:
@@ -176,7 +184,7 @@ def body(task):
             t_encode = t_encode[:abs(COLUMNS - Console.MIN_COLUMN_SIZE)]
 
         # if an entry is about to be log, remove percentage and eta time
-        if color and not task.response_code in task.banned_response_codes:
+        if color is not None and not task.response_code in task.banned_response_codes:
             to_console = to_format_without_progress.format(task.response_code,
                                                     task.response_size,
                                                     task.number,
@@ -193,9 +201,9 @@ def body(task):
                                           t_encode, content_type)
 
             sys.stdout.write(to_console[:COLUMNS-2])
+            sys.stdout.write('\r')
 
         sys.stdout.flush()
-        sys.stdout.write('\r')
 
-        if not os.name == 'nt':
+        if not os.name == 'nt' and Console.show_progress:
             sys.stdout.write("\x1b[0K")