Initial pve cloud backup app
This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from .db import db, utc_now
|
||||
from .migrations import DEFAULT_SETTINGS
|
||||
|
||||
|
||||
INT_KEYS = {"default_retention_value", "max_concurrent_backups"}
|
||||
BOOL_KEYS = {"setup_complete"}
|
||||
|
||||
|
||||
def parse_value(key: str, value: str):
|
||||
if key in INT_KEYS:
|
||||
return int(value)
|
||||
if key in BOOL_KEYS:
|
||||
return value.lower() == "true"
|
||||
if key == "allowed_cors_origins":
|
||||
return [item.strip() for item in value.split(",") if item.strip()]
|
||||
return value
|
||||
|
||||
|
||||
def serialize_value(key: str, value) -> str:
|
||||
if key in BOOL_KEYS:
|
||||
return "true" if bool(value) else "false"
|
||||
if key == "allowed_cors_origins":
|
||||
if isinstance(value, list):
|
||||
return ",".join(str(item).strip() for item in value if str(item).strip())
|
||||
return str(value)
|
||||
|
||||
|
||||
def get_settings() -> dict:
|
||||
with db() as conn:
|
||||
rows = conn.execute("SELECT key, value FROM settings").fetchall()
|
||||
raw = {row["key"]: row["value"] for row in rows}
|
||||
for key, value in DEFAULT_SETTINGS.items():
|
||||
raw.setdefault(key, value)
|
||||
return {key: parse_value(key, value) for key, value in raw.items()}
|
||||
|
||||
|
||||
def set_settings(values: dict) -> dict:
|
||||
now = utc_now()
|
||||
valid_keys = set(DEFAULT_SETTINGS)
|
||||
unknown = set(values) - valid_keys
|
||||
if unknown:
|
||||
raise ValueError(f"Unknown settings: {', '.join(sorted(unknown))}")
|
||||
with db() as conn:
|
||||
for key, value in values.items():
|
||||
conn.execute(
|
||||
"""
|
||||
INSERT INTO settings(key, value, updated_at)
|
||||
VALUES (?, ?, ?)
|
||||
ON CONFLICT(key) DO UPDATE SET value = excluded.value, updated_at = excluded.updated_at
|
||||
""",
|
||||
(key, serialize_value(key, value), now),
|
||||
)
|
||||
return get_settings()
|
||||
|
||||
|
||||
def get_setting(key: str, default: str = "") -> str:
|
||||
with db() as conn:
|
||||
row = conn.execute("SELECT value FROM settings WHERE key = ?", (key,)).fetchone()
|
||||
if row:
|
||||
return row["value"]
|
||||
return default
|
||||
Reference in New Issue
Block a user