You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.3 KiB
77 lines
2.3 KiB
#!/usr/bin/env python3
|
|
from rich.console import Console
|
|
from rich.markdown import Markdown
|
|
from rich.padding import Padding
|
|
from rich.panel import Panel
|
|
from rich.text import Text
|
|
from rich.columns import Columns
|
|
import re
|
|
|
|
console = Console()
|
|
|
|
|
|
def print_markdown(text):
|
|
"""Prints a rich info message. Support Markdown syntax."""
|
|
|
|
md = Padding(Markdown(text), 2)
|
|
console.print(md)
|
|
|
|
|
|
def print_step(text):
|
|
"""Prints a rich info message."""
|
|
|
|
panel = Panel(Text(text, justify="left"))
|
|
console.print(panel)
|
|
|
|
|
|
def print_substep(text, style=""):
|
|
"""Prints a rich info message without the panelling."""
|
|
console.print(text, style=style)
|
|
|
|
|
|
def print_table(items):
|
|
"""Prints items in a table."""
|
|
|
|
console.print(Columns([Panel(f"[yellow]{item}", expand=True) for item in items]))
|
|
|
|
|
|
def handle_input(
|
|
message: str = "",
|
|
check_type=False,
|
|
match: str = "",
|
|
err_message: str = "",
|
|
nmin=None,
|
|
nmax=None,
|
|
oob_error="",
|
|
extra_info="",
|
|
):
|
|
match = re.compile(match + "$")
|
|
console.print(extra_info, no_wrap=True)
|
|
while True:
|
|
console.print(message, end="")
|
|
user_input = input("").strip()
|
|
if re.match(match, user_input) is not None:
|
|
if check_type is not False:
|
|
try:
|
|
user_input = check_type(user_input)
|
|
if nmin is not None and user_input < nmin:
|
|
console.print("[red]" + oob_error) # Input too low failstate
|
|
continue
|
|
if nmax is not None and user_input > nmax:
|
|
console.print("[red]" + oob_error) # Input too high
|
|
continue
|
|
break # Successful type conversion and number in bounds
|
|
except ValueError:
|
|
console.print("[red]" + err_message) # Type conversion failed
|
|
continue
|
|
if nmin is not None and len(user_input) < nmin: # Check if string is long enough
|
|
console.print("[red]" + oob_error)
|
|
continue
|
|
if nmax is not None and len(user_input) > nmax: # Check if string is not too long
|
|
console.print("[red]" + oob_error)
|
|
continue
|
|
break
|
|
console.print("[red]" + err_message)
|
|
|
|
return user_input
|