crafty-4/app/classes/shared/console.py

77 lines
2.2 KiB
Python

import datetime
import logging
import sys
logger = logging.getLogger(__name__)
try:
from colorama import init
from termcolor import colored
except ModuleNotFoundError as e:
logging.critical("Import Error: Unable to load {} module".format(e, e.name))
print("Import Error: Unable to load {} module".format(e, e.name))
from app.classes.shared.installer import installer
installer.do_install()
sys.exit(1)
class Console:
def __init__(self):
if 'colorama' in sys.modules:
init()
@staticmethod
def do_print(message, color):
if 'termcolor' in sys.modules or 'colorama' in sys.modules:
print(colored(message, color))
else:
print(message)
def magenta(self, message):
self.do_print(message, "magenta")
def cyan(self, message):
self.do_print(message, "cyan")
def yellow(self, message):
self.do_print(message, "yellow")
def red(self, message):
self.do_print(message, "red")
def green(self, message):
self.do_print(message, "green")
def white(self, message):
self.do_print(message, "white")
def debug(self, message):
dt = datetime.datetime.now().strftime("%Y-%m-%d %I:%M:%S %p")
self.magenta("[+] Crafty: {} - DEBUG:\t{}".format(dt, message))
def info(self, message):
dt = datetime.datetime.now().strftime("%Y-%m-%d %I:%M:%S %p")
self.white("[+] Crafty: {} - INFO:\t{}".format(dt, message))
def warning(self, message):
dt = datetime.datetime.now().strftime("%Y-%m-%d %I:%M:%S %p")
self.cyan("[+] Crafty: {} - WARNING:\t{}".format(dt, message))
def error(self, message):
dt = datetime.datetime.now().strftime("%Y-%m-%d %I:%M:%S %p")
self.yellow("[+] Crafty: {} - ERROR:\t{}".format(dt, message))
def critical(self, message):
dt = datetime.datetime.now().strftime("%Y-%m-%d %I:%M:%S %p")
self.red("[+] Crafty: {} - CRITICAL:\t{}".format(dt, message))
def help(self, message):
dt = datetime.datetime.now().strftime("%Y-%m-%d %I:%M:%S %p")
self.green("[+] Crafty: {} - HELP:\t{}".format(dt, message))
console = Console()