60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import os.path
|
|
import shutil
|
|
|
|
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:
|
|
secret = fh.read().splitlines()[0]
|
|
self._path = settings.kp.file
|
|
self._kp_org = self._open_or_create(self._path, secret)
|
|
self._kp = self._open_lock(self._path, secret)
|
|
|
|
@staticmethod
|
|
def _open_or_create(path, password) -> PyKeePass:
|
|
if os.path.exists(path):
|
|
return PyKeePass(path, password=password)
|
|
return create_database(path, password)
|
|
|
|
@staticmethod
|
|
def _open_lock(path, password) -> PyKeePass:
|
|
lock_path = path + ".lock"
|
|
shutil.copyfile(path, lock_path)
|
|
return Passwords._open_or_create(lock_path, password)
|
|
|
|
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]
|
|
kp_entry = self._kp_org.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)
|
|
|
|
def save(self):
|
|
# nadpisz plik źródłowy zmianami z lock
|
|
self._kp.save()
|
|
shutil.copyfile(self._path + ".lock", self._path)
|