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
|
from enum import Enum
from typing import Dict, List, Optional
import click
from pygments.lexers.shell import BashLexer
from prompt_toolkit import prompt, print_formatted_text, PromptSession
from prompt_toolkit.lexers import PygmentsLexer
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.shortcuts import ProgressBar
from prompt_toolkit.application import run_in_terminal
from . import color
class SwitchState(Enum):
Yes: int = 0
No: int = 1
OK: int = 2
Cancel: int = 3
confirmStrToSwitch: Dict[str, SwitchState] = {
"y": SwitchState.Yes,
"n": SwitchState.No,
"o": SwitchState.OK,
"c": SwitchState.Cancel
}
switchToConfirmStr: Dict[SwitchState, str] = {
v: k for k, v in confirmStrToSwitch.items()}
defaultInputCommandSession = PromptSession(
message="> ", lexer=PygmentsLexer(BashLexer), auto_suggest=AutoSuggestFromHistory())
class CLI:
def __init__(self, inputCommandSession: Optional[PromptSession] = None):
self.read = prompt
self.getProgressBar = ProgressBar
self.inputCommandSession: PromptSession = inputCommandSession if inputCommandSession \
else defaultInputCommandSession
self.inputCommand = self.inputCommandSession.prompt
self.edit = click.edit
self.clear = click.clear
def write(self, *values, **kwargs)->None: # pylint: disable=R0201
def func():
print_formatted_text(*values, **kwargs)
run_in_terminal(func)
def info(self, message, end: str = "\n")->None:
self.write(color.useCyan(message), end=end)
def warning(self, message, end: str = "\n")->None:
self.write(color.useYellow(message), end=end)
def error(self, message, end: str = "\n")->None:
self.write(color.useRed(message), end=end)
def ok(self, message, end: str = "\n")->None:
self.write(color.useGreen(message), end=end)
def confirm(self, message: str, choice: List[SwitchState])->SwitchState: # pragma: no cover
swstr = ','.join([switchToConfirmStr[x] for x in choice])
ret = self.read(f"{message} ({swstr}) ")
while ret not in confirmStrToSwitch or confirmStrToSwitch[ret] not in choice:
ret = self.read(
f"Not an acceptable value. Please input again ({swstr}): ")
return confirmStrToSwitch[ret]
|