kms_tool/formatText.py at master · Cheain/kms_tool · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
def byterize(obj):
    objdict = obj.__dict__['fields']

    def do_encode(dictio, key):
        if isinstance(dictio[key], str) and len(dictio[key]) > 0 and key not in ['SecondaryAddr']:
            dictio[key] = dictio[key].encode('latin-1')
        elif hasattr(dictio[key], '__dict__'):
            subdictio = dictio[key].__dict__['fields']
            for subkey in subdictio:
                do_encode(subdictio, subkey)

    for field in objdict:
        do_encode(objdict, field)

    return obj


def justify(astring, indent=35, break_every=100):
    str_indent = ('\n' + ' ' * indent)
    splitted = astring.split('\n')
    longests = [(n, s) for n, s in enumerate(splitted) if len(s) >= break_every]

    for longest in longests:
        lines = []
        for i in range(0, len(longest[1]), break_every):
            lines.append(longest[1][i: i + break_every])
        splitted[longest[0]] = str_indent.join(lines)

    if len(splitted) > 1:
        justy = str_indent.join(splitted)
    else:
        justy = str_indent + str_indent.join(splitted)

    return justy


class ShellStyle(object):
    def style(self, s, style):
        return style + s + '\033[0m'

    def green(self, s):
        return self.style(s, '\033[92m')

    def blue(self, s):
        return self.style(s, '\033[94m')

    def yellow(self, s):
        return self.style(s, '\033[93m')

    def red(self, s):
        return self.style(s, '\033[91m')

    def magenta(self, s):
        return self.style(s, '\033[95m')

    def cyan(self, s):
        return self.style(s, '\033[96m')

    def white(self, s):
        return self.style(s, '\033[97m')

    def bold(self, s):
        return self.style(s, '\033[1m')

    def underline(self, s):
        return self.style(s, '\033[4m')


def shell_message(nshell):
    shelldict = {0: ShellStyle().yellow("Client generating RPC Bind Request..."),
                 1: ShellStyle().yellow("Client sending RPC Bind Request...") + ShellStyle().red(
                     "\t\t\t\t===============>"),
                 2: ShellStyle().red("===============>\t\t") + ShellStyle().yellow(
                     "Server received RPC Bind Request !!!"),
                 3: ShellStyle().yellow("\t\t\t\tServer parsing RPC Bind Request..."),
                 4: ShellStyle().yellow("\t\t\t\tServer generating RPC Bind Response..."),
                 5: ShellStyle().red("<===============\t\t") + ShellStyle().yellow(
                     "Server sending RPC Bind Response..."),
                 6: ShellStyle().green("\t\t\t\tRPC Bind acknowledged !!!\n"),
                 7: ShellStyle().yellow("Client received RPC Bind Response !!!") + ShellStyle().red(
                     "\t\t\t\t<==============="),
                 8: ShellStyle().green("RPC Bind acknowledged !!!\n"),
                 9: ShellStyle().blue("Client generating Activation Request dictionary..."),
                 10: ShellStyle().blue("Client generating Activation Request data..."),
                 11: ShellStyle().blue("Client generating RPC Activation Request..."),
                 12: ShellStyle().blue("Client sending RPC Activation Request...") + ShellStyle().red(
                     "\t\t\t===============>"),
                 13: ShellStyle().red("===============>\t\t") + ShellStyle().blue(
                     "Server received RPC Activation Request !!!"),
                 14: ShellStyle().blue("\t\t\t\tServer parsing RPC Activation Request..."),
                 15: ShellStyle().blue("\t\t\t\tServer processing KMS Activation Request..."),
                 16: ShellStyle().blue("\t\t\t\tServer processing KMS Activation Response..."),
                 17: ShellStyle().blue("\t\t\t\tServer generating RPC Activation Response..."),
                 18: ShellStyle().red("<===============\t\t") + ShellStyle().blue(
                     "Server sending RPC Activation Response..."),
                 19: ShellStyle().green("\t\t\t\tServer responded, now in Stand by...\n"),
                 20: ShellStyle().blue("Client received Response !!!") + ShellStyle().red("\t\t\t\t\t<==============="),
                 21: ShellStyle().green("Activation Done !!!"),
                 -1: ShellStyle().red("\t\t\t\t\t\t\t\tServer receiving"),
                 -2: ShellStyle().red("Client sending"),
                 -3: ShellStyle().red("Client receiving"),
                 -4: ShellStyle().red("\t\t\t\t\t\t\t\tServer sending")
                 }

    if isinstance(nshell, list):
        for n in nshell:
            print(shelldict[n])
    else:
        print(shelldict[nshell])