111 lines
2.8 KiB
Python
Executable File
111 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from rich.console import Console
|
|
from rich.text import Text
|
|
import argparse
|
|
|
|
console = Console(force_terminal=True)
|
|
|
|
colors = [
|
|
"black",
|
|
"red",
|
|
"green",
|
|
"yellow",
|
|
"blue",
|
|
"magenta",
|
|
"cyan",
|
|
"white",
|
|
"bright_black",
|
|
"bright_red",
|
|
"bright_green",
|
|
"bright_yellow",
|
|
"bright_blue",
|
|
"bright_magenta",
|
|
"bright_cyan",
|
|
"bright_white",
|
|
]
|
|
|
|
|
|
def wave():
|
|
"""
|
|
╮╭╭╮╮╭╭╮╮╭╭╮╮╭╭╮
|
|
╰╯╯╰╰╯╯╰╰╯╯╰╰╯╯╰
|
|
"""
|
|
|
|
lines = ["╮╭╭╮╮╭╭╮╮╭╭╮╮╭╭╮", "╰╯╯╰╰╯╯╰╰╯╯╰╰╯╯╰"]
|
|
for line in lines:
|
|
styled_line = [
|
|
Text(line[i * 2 : i * 2 + 2], style=color)
|
|
for i, color in enumerate(colors[:8])
|
|
]
|
|
console.print(Text("").join(styled_line))
|
|
|
|
|
|
def dots():
|
|
"""
|
|
● ● ● ● ● ● ● ●
|
|
"""
|
|
|
|
console.print(Text(" ").join([Text("●", style=color) for color in colors[:8]]))
|
|
|
|
|
|
def double_dots():
|
|
"""
|
|
● ● ● ● ● ● ● ●
|
|
● ● ● ● ● ● ● ●
|
|
"""
|
|
|
|
console.print(Text(" ").join([Text("●", style=color) for color in colors[:8]]))
|
|
console.print(Text(" ").join([Text("●", style=color) for color in colors[8:]]))
|
|
|
|
|
|
def wave_w_dots():
|
|
"""
|
|
Alternating colors from red to white with
|
|
black separators
|
|
|
|
╮╮●╭╭╭─╮╮╮●╭╭╭─╮╮╮●╭╭╭─╮╮╮●╭╭
|
|
╰╰─╯╯╯●╰╰╰─╯╯╯●╰╰╰─╯╯╯●╰╰╰─╯╯
|
|
"""
|
|
|
|
lines = ["╮╮●╭╭╭─╮╮╮●╭╭╭─╮╮╮●╭╭╭─╮╮╮●╭╭", "╰╰─╯╯╯●╰╰╰─╯╯╯●╰╰╰─╯╯╯●╰╰╰─╯╯"]
|
|
|
|
for line in lines:
|
|
styled_line = []
|
|
for i, x in enumerate(line):
|
|
if i % 4:
|
|
styled_line.append(Text(x, style=colors[1 + int(i / 4)]))
|
|
else:
|
|
styled_line.append(Text(x, style="black"))
|
|
|
|
console.print(Text("").join(styled_line))
|
|
|
|
|
|
def main():
|
|
prompts = {
|
|
"wave": wave,
|
|
"dots": dots,
|
|
"double-dots": double_dots,
|
|
"wave-dots": wave_w_dots,
|
|
}
|
|
parser = argparse.ArgumentParser(
|
|
description="generate ansi color coded login prompts"
|
|
)
|
|
parser.add_argument("--welcome", type=str, help="a welcome screen to print")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not args.welcome:
|
|
for name, prompt in prompts.items():
|
|
console.print(f"name: {name}")
|
|
prompt()
|
|
elif args.welcome not in prompts.keys():
|
|
console.print(f"{args.welcome} is not a registered prompt")
|
|
else:
|
|
console.print()
|
|
prompts[args.welcome]()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|