octosuite/lib/colors.py

import os
import sys
import platform
from datetime import datetime

colors = True
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 = red_bg = ""

else:
    date_time = datetime.now()
    sys_info = [("Processor",platform.processor),
                        ("Node", platform.node),
                        ("Release", platform.release),
                        ("Architecture", platform.architecture),
                        ("Version", platform.version)]
                        
    banner = f"""
                 OCTOSUITE © 2022 Richard Mwewa
               {date_time.strftime('%A %d %B %Y, %H:%M:%S%p')}
            
            
        {platform.system()}"""
                        
    print(banner)
    for key, value in sys_info:
        print(f"\t├─ {key}: {value()}")
    print("\n")
    while True:
    	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"
    			red = "\033[91m"
    			reset = "\033[0m"
    			green = "\033[92m"
    			red_bg = "\033[41;37m"
    			break
    		elif color_chooser.lower() == "n":
    		    red = white = green = red_bg = reset = ""
    		    break
    		else:
    			print(f"\n[ ! ] Your response ({color_chooser}) is invalid (expected y or n)")
    			
    	except KeyboardInterrupt:
    		exit(f"[ ! ] Process interrupted with (Ctrl+C)")