From 007e7076c7404c86180ff3aa6ea3fe15024a87c3 Mon Sep 17 00:00:00 2001 From: chqn Date: Wed, 3 Jan 2024 13:12:48 -0600 Subject: Refactored function signatures and added some TODOs --- src/user_interface.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'src/user_interface.py') diff --git a/src/user_interface.py b/src/user_interface.py index 32abf60..99eb524 100644 --- a/src/user_interface.py +++ b/src/user_interface.py @@ -5,7 +5,7 @@ import color_engine from rich import print -def colorPickerUI(img_path: str): +def colorPickerUI(img_path: str) -> tuple: #display the selected color scheme and ask user if they like it or want to generate a new color scheme confirmed = False while not confirmed: @@ -36,10 +36,12 @@ def colorPickerUI(img_path: str): confirmed = True return hex_colors, hex_compliments -def pickRandomWallpaper(): +def pickRandomWallpaper() -> str: confirmed = False while not confirmed: wallpaper = Paths['wallpapers'] + random.choice(os.listdir(Paths['wallpapers'])) + # TODO: Replace image preview with something native to python + # (Currently borrowing viu module from Rust) os.system(f'viu {wallpaper}') print(f'picked wallpaper: {wallpaper}') print('[bold](a)ccept (r)etry') -- cgit v1.2.3