Fix setup_service to mark setup_complete only after successful runtime DB init
This commit is contained in:
@@ -7,19 +7,27 @@ enforcing the rule that setup can only run once.
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
from app.utils.async_utils import run_blocking
|
||||
from pathlib import Path
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
import bcrypt
|
||||
import structlog
|
||||
|
||||
if TYPE_CHECKING:
|
||||
import aiosqlite
|
||||
|
||||
from app.db import init_db, open_db
|
||||
from app.repositories import settings_repo
|
||||
from app.utils.async_utils import run_blocking
|
||||
from app.utils.setup_utils import (
|
||||
get_map_color_thresholds as util_get_map_color_thresholds,
|
||||
)
|
||||
from app.utils.setup_utils import (
|
||||
get_password_hash as util_get_password_hash,
|
||||
)
|
||||
from app.utils.setup_utils import (
|
||||
set_map_color_thresholds as util_set_map_color_thresholds,
|
||||
)
|
||||
|
||||
if TYPE_CHECKING:
|
||||
import aiosqlite
|
||||
|
||||
log: structlog.stdlib.BoundLogger = structlog.get_logger()
|
||||
|
||||
@@ -89,22 +97,28 @@ async def run_setup(
|
||||
await settings_repo.set_setting(db, _KEY_DATABASE_PATH, database_path)
|
||||
|
||||
runtime_initialized = await _ensure_database_initialized(database_path)
|
||||
if not runtime_initialized:
|
||||
log.error(
|
||||
"bangui_setup_failed",
|
||||
reason="runtime_database_initialization_failed",
|
||||
database_path=database_path,
|
||||
)
|
||||
raise RuntimeError("Runtime database could not be initialized.")
|
||||
|
||||
runtime_db: aiosqlite.Connection | None = None
|
||||
try:
|
||||
if runtime_initialized:
|
||||
runtime_db = await open_db(database_path)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_PASSWORD_HASH, hashed)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_DATABASE_PATH, database_path)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_FAIL2BAN_SOCKET, fail2ban_socket)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_TIMEZONE, timezone)
|
||||
await settings_repo.set_setting(
|
||||
runtime_db, _KEY_SESSION_DURATION, str(session_duration_minutes)
|
||||
)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_MAP_COLOR_THRESHOLD_HIGH, "100")
|
||||
await settings_repo.set_setting(runtime_db, _KEY_MAP_COLOR_THRESHOLD_MEDIUM, "50")
|
||||
await settings_repo.set_setting(runtime_db, _KEY_MAP_COLOR_THRESHOLD_LOW, "20")
|
||||
await settings_repo.set_setting(runtime_db, _KEY_SETUP_DONE, "1")
|
||||
runtime_db = await open_db(database_path)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_PASSWORD_HASH, hashed)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_DATABASE_PATH, database_path)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_FAIL2BAN_SOCKET, fail2ban_socket)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_TIMEZONE, timezone)
|
||||
await settings_repo.set_setting(
|
||||
runtime_db, _KEY_SESSION_DURATION, str(session_duration_minutes)
|
||||
)
|
||||
await settings_repo.set_setting(runtime_db, _KEY_MAP_COLOR_THRESHOLD_HIGH, "100")
|
||||
await settings_repo.set_setting(runtime_db, _KEY_MAP_COLOR_THRESHOLD_MEDIUM, "50")
|
||||
await settings_repo.set_setting(runtime_db, _KEY_MAP_COLOR_THRESHOLD_LOW, "20")
|
||||
await settings_repo.set_setting(runtime_db, _KEY_SETUP_DONE, "1")
|
||||
finally:
|
||||
if runtime_db is not None:
|
||||
await runtime_db.close()
|
||||
@@ -115,13 +129,6 @@ async def run_setup(
|
||||
log.info("bangui_setup_completed")
|
||||
|
||||
|
||||
from app.utils.setup_utils import (
|
||||
get_map_color_thresholds as util_get_map_color_thresholds,
|
||||
get_password_hash as util_get_password_hash,
|
||||
set_map_color_thresholds as util_set_map_color_thresholds,
|
||||
)
|
||||
|
||||
|
||||
async def get_password_hash(db: aiosqlite.Connection) -> str | None:
|
||||
"""Return the stored bcrypt password hash, or ``None`` if not set."""
|
||||
return await util_get_password_hash(db)
|
||||
|
||||
Reference in New Issue
Block a user