diff --git a/bin/welcome.py b/bin/welcome.py index 774375a..8dde2be 100755 --- a/bin/welcome.py +++ b/bin/welcome.py @@ -48,6 +48,14 @@ 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(): """ @@ -72,9 +80,9 @@ def wave_w_dots(): def main(): - prompts = {"wave": wave, "dots": dots, "wave_w_dots": wave_w_dots} + prompts = {"wave": wave, "dots": dots,'double-dots':double_dots, "wave-dots": wave_w_dots} parser = argparse.ArgumentParser( - description="generate ansi color coded login prompts'" + description="generate ansi color coded login prompts" ) parser.add_argument("--welcome", type=str, help="a welcome screen to print") @@ -87,6 +95,7 @@ def main(): elif args.welcome not in prompts.keys(): console.print(f"{args.welcome} is not a registered prompt") else: + console.print() prompts[args.welcome]()