bellingcat/octosuite
1
0
Fork 0
mirror of https://github.com/bellingcat/octosuite.git synced 2026-06-13 05:48:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update colors.py

Browse Source
This commit is contained in:
Richard Mwewa
2022-04-14 11:11:41 +02:00
committed by GitHub
parent b4f009c7d4
commit 5718c5f229
1 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lib/colors.py
View File

@@ -1,16 +1,28 @@
import os
import sys
# Colors will be unavailable on non-linux machines
colors = True
machine = sys.platform
machine = sys.platform
# Colors will be unavailable on non-linux machines
if machine.lower().startswith(("os", "win", "darwin","ios")):
colors = False
if not colors:
reset = red = white = green = ""
reset = red = white = green = green_bg = white_bg = red_bg = ""
else:
white = "\033[97m"
red = "\033[91m"
reset = "\033[0m"
green = "\033[92m"
else:
try:
color_chooser = input(f"[ ? ] Welcome {os.getlogin()}, would you like to enable colors for this session? [Y/n] ")
if color_chooser.lower() == "y":
white = "\033[97m"
white_bg = "\033[47;30m"
red = "\033[91m"
reset = "\033[0m"
green = "\033[92m"
green_bg = "\033[42;37m"
red_bg = "\033[41;37m"
else:
red = white = green = green_bg = white_bg = red_bg = reset = ""
except KeyboardInterrupt:
exit(f"[ ! ] Process interrupted with Ctrl+C")