File: checker.py

package info (click to toggle)
python-softlayer 6.2.5-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 7,508 kB
  • sloc: python: 57,195; makefile: 133; xml: 97; sh: 59
file content (98 lines) | stat: -rw-r--r-- 3,363 bytes parent folder | download
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
from SoftLayer.CLI import environment
from click.testing import CliRunner
from SoftLayer.CLI.core import cli
import itertools

runner = CliRunner()
env = environment.Environment()
env.load()
TopLevelCommands = env.list_commands()


def findHelp(Table):
    for row in Table:
        if '--help' in row:
            return True
    return False


def findCorrectFlagTable(arrayCommandByLine):
    cont = 0
    arrayTable = []
    for line in arrayCommandByLine:
        if '─' in line:
            cont += 1
        if cont == 2:
            if not findHelp(arrayTable):
                arrayTable = []
            cont = 0
        if '│' in line:
            arrayTable.append(line)
    return arrayTable


def getFlags(arrayCommandByLine):
    flags = []
    arrayFlagsTable = findCorrectFlagTable(arrayCommandByLine)

    for line in arrayFlagsTable:
        arrayFlags = line.split('│')
        if arrayFlags[1].strip() != '':
            flags.append(f'{arrayFlags[1].strip()},{arrayFlags[2].strip()}: {arrayFlags[3].strip()}')
        else:
            flags.append(f'{arrayFlags[2].strip()}: {arrayFlags[3].strip()}')
    return flags


def findCommandNested(arrayCommandByLine):
    commandNested = False
    commands = []
    for line in arrayCommandByLine:
        if commandNested:
            nestedCommand = line.split(' ')
            commands.append(nestedCommand[2])
        if 'Commands:' in line:
            commandNested = True
    return commands


def printCommandName(TopLevelCommand, command, nestedCommandName=''):
    if nestedCommandName != '':
        f.write(f'slcli {TopLevelCommand} {command} {nestedCommandName}\n')
    else:
        f.write(f'slcli {TopLevelCommand} {command}\n')


def printBody(commandArray):
    for flag in getFlags(commandArray):
        f.write(f'\tFlag: {flag}\n')
    f.write("\t--------------------------------\n")
    f.write(f'\tDescription: {commandArray[2].strip()}\n')
    f.write("\t--------------------------------\n")
    list_of_strings = "".join(commandArray[0])
    command_strings=list_of_strings.replace("Usage: cli","Usage: slcli")
    grouped_strings = ["".join(g) for k, g in itertools.groupby(command_strings, lambda x: x == " ") if not k]
    f.write(f'\t{" ".join(grouped_strings)}\n')
    f.write("==============================================================\n")


# Main Function
with open('output.txt', "w", encoding="utf-8") as f:
    f.write('SLCLI Command Directory\n')
    f.write("==============================================================\n")

    for TopLevelCommand in TopLevelCommands:
        commands = env.list_commands(TopLevelCommand)
        for command in commands:
            result = runner.invoke(cli, [TopLevelCommand, command, '--help'])
            commandArray = result.output.splitlines()
            nestedCommands = findCommandNested(commandArray)
            if len(nestedCommands) != 0:
                for nestedCommand in nestedCommands:
                    result = runner.invoke(cli, [TopLevelCommand, command, nestedCommand, '--help'])
                    commandArray = result.output.splitlines()
                    printCommandName(TopLevelCommand, command, nestedCommand)
                    printBody(commandArray)
            else:
                printCommandName(TopLevelCommand, command)
                printBody(commandArray)