52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
import os.path
|
|
from contextlib import contextmanager
|
|
from typing import Any, Generator
|
|
|
|
import keyring
|
|
from injectable import injectable
|
|
from pykeepass import PyKeePass, create_database
|
|
|
|
|
|
@injectable(singleton=True)
|
|
class Passwords:
|
|
def __init__(self):
|
|
from app.config import get_settings
|
|
settings = get_settings()
|
|
|
|
with open(settings.kp.secret, "r") as fh:
|
|
keyring.set_password("karl", "kp", fh.read().splitlines()[0])
|
|
self._path = settings.kp.file
|
|
|
|
@contextmanager
|
|
def open(self, mode: str = "r") -> Generator[PyKeePass | Any, Any, None]:
|
|
kp = PyKeePass(self._path, password=keyring.get_password("karl", "kp")) \
|
|
if os.path.exists(self._path) else create_database(self._path, password=keyring.get_password("karl", "kp"))
|
|
yield kp
|
|
if mode == "rw":
|
|
kp.save()
|
|
|
|
def get_values(self, keys: list[str]) -> dict[str, str]:
|
|
output = {}
|
|
for k in keys:
|
|
key_parts = k.split(".")
|
|
path = key_parts[:-1] if len(key_parts) > 2 else None
|
|
entry_name = key_parts[-2]
|
|
field_name = key_parts[-1]
|
|
with self.open() as kp:
|
|
kp_entry = kp.find_entries(path=path, first=True, title=entry_name)
|
|
output[k] = self._get_field_value(kp_entry, field_name)
|
|
return output
|
|
|
|
@staticmethod
|
|
def _get_field_value(kp_entry, field_name):
|
|
if kp_entry is None:
|
|
return None
|
|
match field_name:
|
|
case "username":
|
|
return kp_entry.username
|
|
case "password":
|
|
return kp_entry.password
|
|
case "url":
|
|
return kp_entry.url
|
|
case _:
|
|
return kp_entry.get_custom_property(field_name)
|