diff --git a/src/password_manager/manager.py b/src/password_manager/manager.py index eca2087..8b8ca8a 100644 --- a/src/password_manager/manager.py +++ b/src/password_manager/manager.py @@ -367,11 +367,15 @@ class PasswordManager: """ try: choice = input( - "Do you want to (1) Enter an existing seed or (2) Generate a new seed? (1/2): " + "Do you want to (1) Paste in an existing seed in full " + "(2) Enter an existing seed one word at a time or " + "(3) Generate a new seed? (1/2/3): " ).strip() if choice == "1": - fingerprint = self.setup_existing_seed() + fingerprint = self.setup_existing_seed(method="paste") elif choice == "2": + fingerprint = self.setup_existing_seed(method="words") + elif choice == "3": fingerprint = self.generate_new_seed() else: print(colored("Invalid choice. Exiting.", "red")) diff --git a/src/tests/test_profile_management.py b/src/tests/test_profile_management.py index bbb52de..7665b87 100644 --- a/src/tests/test_profile_management.py +++ b/src/tests/test_profile_management.py @@ -41,7 +41,7 @@ def test_add_and_delete_entry(monkeypatch): manager_module.PasswordManager, "generate_bip85_seed", lambda self: seed ) monkeypatch.setattr(manager_module, "confirm_action", lambda *_a, **_k: True) - monkeypatch.setattr("builtins.input", lambda *_a, **_k: "2") + monkeypatch.setattr("builtins.input", lambda *_a, **_k: "3") pm.add_new_fingerprint()