net/ldaptool
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Catch exception if no password found

Browse Source
This commit is contained in:
Daniel Dizdarevic 2023-05-08 17:47:21 +02:00
parent 21069e892e
commit c42bef6f02
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/ldaptool/search/config.py
View File

@ -148,7 +148,9 @@ class Keepass(PasswordManager):
password = getpass.getpass(f"KeePass password for database {self.database}: ")
kp = pykeepass.PyKeePass(self.database, password=password)
entry = kp.find_entries(username=password_name, first=True)
return entry.password # type: ignore
if entry:
return entry.password # type: ignore
@dataclasses.dataclass
@ -221,6 +223,11 @@ class Config:
if realm.account is None:
raise RuntimeError("Can't get password without acccount - should use kerberos instead")
if self.password_manager:
return self.password_manager.get_password(realm.password_name)
password = self.password_manager.get_password(realm.password_name)
else:
password = getpass.getpass(f"Enter password for {realm.password_name}: ")
return getpass.getpass(f"Enter password for {realm.password_name}: ")
if password:
return password
else:
raise SystemExit(f"No password found for {realm.password_name}")