diff --git a/goodchain/src/helpers/MenuHelper.py b/goodchain/src/helpers/MenuHelper.py index 8f11a4e..78c2c32 100644 --- a/goodchain/src/helpers/MenuHelper.py +++ b/goodchain/src/helpers/MenuHelper.py @@ -5,17 +5,6 @@ from helpers import BlockHelper as blockHelper from helpers import TaskHelper as taskHelper from helpers import UtilityHelper as utilityHelper -class bcolors: - HEADER = '\033[95m' - OKBLUE = '\033[94m' - OKCYAN = '\033[96m' - OKGREEN = '\033[92m' - WARNING = '\033[93m' - FAIL = '\033[91m' - ENDC = '\033[0m' - BOLD = '\033[1m' - UNDERLINE = '\033[4m' - class MenuHelper: def __init__(self, db): self.db = db diff --git a/goodchain/src/helpers/UtilityHelper.py b/goodchain/src/helpers/UtilityHelper.py index 76c3370..51ab871 100644 --- a/goodchain/src/helpers/UtilityHelper.py +++ b/goodchain/src/helpers/UtilityHelper.py @@ -3,6 +3,16 @@ import hashlib import os from helpers import BlockHelper as blockHelper +class bcolors: + HEADER = '\033[95m' + OKBLUE = '\033[94m' + OKCYAN = '\033[96m' + OKGREEN = '\033[92m' + WARNING = '\033[93m' + FAIL = '\033[91m' + ENDC = '\033[0m' + BOLD = '\033[1m' + UNDERLINE = '\033[4m' def computeHash(data): hash = hashlib.sha256() @@ -47,4 +57,27 @@ def loginStartup(self): blockHelper.validateMinedBlock(self) # TODO - check if there are any notifcations +def printError(text): + print(f"{bcolors.FAIL}{text}{bcolors.ENDC}") + +def printSuccess(text): + print(f"{bcolors.OKGREEN}{text}{bcolors.ENDC}") + +def printWarning(text): + print(f"{bcolors.WARNING}{text}{bcolors.ENDC}") + +def printInfo(text): + print(f"{bcolors.OKBLUE}{text}{bcolors.ENDC}") + +def printBold(text): + print(f"{bcolors.BOLD}{text}{bcolors.ENDC}") + +def printUnderline(text): + print(f"{bcolors.UNDERLINE}{text}{bcolors.ENDC}") + +def printHeader(text): + print(f"{bcolors.HEADER}{text}{bcolors.ENDC}") + + + \ No newline at end of file