refactor(pw): Improve handling of password input

2026-04-16 16:01:49 +02:00
parent 414c065df4
commit bb123c23a1
2 changed files with 82 additions and 18 deletions
@@ -1,7 +1,6 @@
 from typing import Optional
-import getpass
-import time
-import colorama
+
+from commands.util.password_manager import PasswordManager


 def choice(default: str, options: str, msg: str) -> str:
@@ -33,18 +32,13 @@ def confirm_overwrite(msg: Optional[str] = ""):
    )


-def password(msg: str = ""):
-    pw = ""
-    if msg != "":
-        pw = getpass.getpass(msg)
-    else:
-        pw = getpass.getpass()
-    if pw != "":
-        return pw
-    else:
-        time.sleep(1)
-        print(
-            colorama.Fore.RED + "Error:",
-            colorama.Style.RESET_ALL + "Password cannot be empty",
-        )
-        return password(msg)
+# Use class to store the password
+pw_manager = PasswordManager()
+
+
+def password():
+    return pw_manager.get()
+
+
+def unlock_sudo():
+    return pw_manager.validate()