Keepass outline
This commit is contained in:
parent
b18488a6d3
commit
162c0adf13
9 changed files with 115 additions and 3 deletions
|
|
@ -1,18 +1,26 @@
|
|||
from functools import lru_cache
|
||||
|
||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||
from pydantic import BaseModel
|
||||
from pathlib import Path
|
||||
|
||||
import yaml
|
||||
from pydantic import BaseModel
|
||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||
|
||||
|
||||
class AppConfig(BaseModel):
|
||||
host: str = "127.0.0.1"
|
||||
port: int = 8000
|
||||
reload: bool = True
|
||||
|
||||
|
||||
class KeePassConfig(BaseModel):
|
||||
file: str = "database.kdbx"
|
||||
secret: Path | str = "/run/secrets/kp_secret"
|
||||
|
||||
|
||||
class Settings(BaseSettings):
|
||||
model_config = SettingsConfigDict(env_prefix="KARL_", env_nested_delimiter="__")
|
||||
app: AppConfig = AppConfig()
|
||||
kp: KeePassConfig = KeePassConfig()
|
||||
|
||||
@classmethod
|
||||
def from_yaml(cls, path: Path | str = "config/config.yaml") -> "Settings":
|
||||
|
|
@ -23,6 +31,7 @@ class Settings(BaseSettings):
|
|||
data = yaml.safe_load(fh) or {}
|
||||
return cls(**data)
|
||||
|
||||
|
||||
@lru_cache
|
||||
def get_settings() -> Settings:
|
||||
return Settings.from_yaml()
|
||||
|
|
|
|||
49
app/model/passwords.py
Normal file
49
app/model/passwords.py
Normal file
|
|
@ -0,0 +1,49 @@
|
|||
from dataclasses import dataclass, field
|
||||
from typing import Type
|
||||
|
||||
# TODO: unnecessary?
|
||||
|
||||
@dataclass
|
||||
class PathItem:
|
||||
name: str
|
||||
t: Type
|
||||
|
||||
@dataclass
|
||||
class Path:
|
||||
path: list[PathItem] = field(default_factory=list)
|
||||
|
||||
def append(self, name, t):
|
||||
self.path.append(PathItem(name, t))
|
||||
|
||||
def __str__(self):
|
||||
return "/".join([i.name for i in self.path])
|
||||
|
||||
|
||||
@dataclass
|
||||
class Group:
|
||||
name: str
|
||||
passwords: list["Password"] = field(default_factory=list)
|
||||
parent: "Group|None" = None
|
||||
|
||||
@property
|
||||
def path(self):
|
||||
if self.parent is None:
|
||||
new_path = Path()
|
||||
new_path.append(self.name, type(self))
|
||||
return new_path
|
||||
return self.parent.path.append(self.name, type(self))
|
||||
|
||||
|
||||
@dataclass
|
||||
class Password:
|
||||
name: str
|
||||
group: Group
|
||||
|
||||
@property
|
||||
def path(self):
|
||||
return self.group.path.append(self.name, type(self))
|
||||
|
||||
class UnencryptedPassword(Password):
|
||||
def __init__(self, name: str, value: str, group: Group):
|
||||
super().__init__(name, group)
|
||||
self.value = value
|
||||
38
app/services/passwords.py
Normal file
38
app/services/passwords.py
Normal file
|
|
@ -0,0 +1,38 @@
|
|||
import os.path
|
||||
|
||||
from pykeepass import PyKeePass, create_database, Group
|
||||
|
||||
|
||||
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()
|
||||
|
||||
self._kp_org = self.__get_or_create_store(settings.kp.file, secret)
|
||||
self._kp = self.__get_lock(settings.kp.file, secret)
|
||||
|
||||
@staticmethod
|
||||
def __get_or_create_store(path, passwd) -> PyKeePass:
|
||||
if os.path.exists(path):
|
||||
return PyKeePass(
|
||||
path,
|
||||
password=passwd,
|
||||
)
|
||||
return create_database(path, passwd)
|
||||
|
||||
@staticmethod
|
||||
def __get_lock(path, passwd) -> PyKeePass:
|
||||
lock_path = path + ".lock"
|
||||
import shutil
|
||||
shutil.copyfile(path, lock_path)
|
||||
return Passwords.__get_or_create_store(lock_path, passwd)
|
||||
|
||||
@property
|
||||
def store(self):
|
||||
return self._kp.root_group
|
||||
|
||||
def save(self, group: Group):
|
||||
pass
|
||||
1
app/services/system.py
Normal file
1
app/services/system.py
Normal file
|
|
@ -0,0 +1 @@
|
|||
|
||||
0
app/util/__init__.py
Normal file
0
app/util/__init__.py
Normal file
11
app/util/dicts.py
Normal file
11
app/util/dicts.py
Normal file
|
|
@ -0,0 +1,11 @@
|
|||
from types import SimpleNamespace
|
||||
|
||||
|
||||
class NestedNamespace(SimpleNamespace):
|
||||
def __init__(self, dictionary, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
for key, value in dictionary.items():
|
||||
if isinstance(value, dict):
|
||||
self.__setattr__(key, NestedNamespace(value))
|
||||
else:
|
||||
self.__setattr__(key, value)
|
||||
|
|
@ -2,3 +2,6 @@ app:
|
|||
host: "127.0.0.1"
|
||||
port: 8000
|
||||
reload: true
|
||||
kp:
|
||||
file: "config/kp.kdbx"
|
||||
secret: "config/secret.txt"
|
||||
|
|
|
|||
|
|
@ -11,6 +11,7 @@ dependencies = [
|
|||
"jinja2>=3.1.4",
|
||||
"pydantic-settings>=2.4.0",
|
||||
"pyyaml>=6.0.2",
|
||||
"pykeepass>=4.1.1.post1"
|
||||
]
|
||||
|
||||
[project.optional-dependencies]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue