Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
thespeedx
GitHub Repository: thespeedx/tbomb
 Path: blob/master/utils/decorators.py
3476 views
1
from colorama import Fore, Style

2


3


4
class IconicDecorator(object):

5
    def __init__(self):

6
        self.PASS = Style.BRIGHT + Fore.GREEN + "[ ✔ ]" + Style.RESET_ALL

7
        self.FAIL = Style.BRIGHT + Fore.RED + "[ ✘ ]" + Style.RESET_ALL

8
        self.WARN = Style.BRIGHT + Fore.YELLOW + "[ ! ]" + Style.RESET_ALL

9
        self.HEAD = Style.BRIGHT + Fore.CYAN + "[ # ]" + Style.RESET_ALL

10
        self.CMDL = Style.BRIGHT + Fore.BLUE + "[ → ]" + Style.RESET_ALL

11
        self.STDS = "     "

12


13


14
class StatusDecorator(object):

15
    def __init__(self):

16
        self.PASS = Style.BRIGHT + Fore.GREEN + "[ SUCCESS ]" + Style.RESET_ALL

17
        self.FAIL = Style.BRIGHT + Fore.RED + "[ FAILURE ]" + Style.RESET_ALL

18
        self.WARN = Style.BRIGHT + Fore.YELLOW + "[ WARNING ]"\

19
            + Style.RESET_ALL

20
        self.HEAD = Style.BRIGHT + Fore.CYAN + "[ SECTION ]" + Style.RESET_ALL

21
        self.CMDL = Style.BRIGHT + Fore.BLUE + "[ COMMAND ]" + Style.RESET_ALL

22
        self.STDS = "           "

23


24


25
class MessageDecorator(object):

26
    def __init__(self, attr):

27
        ICON = IconicDecorator()

28
        STAT = StatusDecorator()

29
        if attr == "icon":

30
            self.PASS = ICON.PASS

31
            self.FAIL = ICON.FAIL

32
            self.WARN = ICON.WARN

33
            self.HEAD = ICON.HEAD

34
            self.CMDL = ICON.CMDL

35
            self.STDS = ICON.STDS

36
        elif attr == "stat":

37
            self.PASS = STAT.PASS

38
            self.FAIL = STAT.FAIL

39
            self.WARN = STAT.WARN

40
            self.HEAD = STAT.HEAD

41
            self.CMDL = STAT.CMDL

42
            self.STDS = STAT.STDS

43


44
    def SuccessMessage(self, RequestMessage):

45
        print(self.PASS + " " + Style.RESET_ALL + RequestMessage)

46


47
    def FailureMessage(self, RequestMessage):

48
        print(self.FAIL + " " + Style.RESET_ALL + RequestMessage)

49


50
    def WarningMessage(self, RequestMessage):

51
        print(self.WARN + " " + Style.RESET_ALL + RequestMessage)

52


53
    def SectionMessage(self, RequestMessage):

54
        print(self.HEAD + " " + Fore.CYAN + Style.BRIGHT

55
              + RequestMessage + Style.RESET_ALL)

56


57
    def CommandMessage(self, RequestMessage):

58
        return self.CMDL + " " + Style.RESET_ALL + RequestMessage

59


60
    def GeneralMessage(self, RequestMessage):

61
        print(self.STDS + " " + Style.RESET_ALL + RequestMessage)

62


63


64