Add local API key onboarding

This commit is contained in:
sinmb79
2026-04-03 09:13:05 +09:00
parent 544e4e0720
commit c8e1ec5d34
10 changed files with 328 additions and 0 deletions

View File

@@ -4,12 +4,28 @@ from functools import lru_cache
from pathlib import Path
import os
from dotenv import load_dotenv
from pydantic import BaseModel, Field
from civilplan_mcp.secure_store import default_key_store_path, load_api_keys
BASE_DIR = Path(__file__).resolve().parent.parent
def load_local_env() -> None:
env_path = BASE_DIR / ".env"
if env_path.exists():
load_dotenv(env_path, override=False)
def _load_secure_api_keys(path: Path) -> dict[str, str]:
try:
return load_api_keys(path=path)
except Exception:
return {}
class Settings(BaseModel):
app_name: str = "civilplan_mcp"
version: str = "1.0.0"
@@ -19,13 +35,22 @@ class Settings(BaseModel):
db_path: Path = Field(default_factory=lambda: BASE_DIR / "civilplan.db")
output_dir: Path = Field(default_factory=lambda: BASE_DIR / "output")
data_dir: Path = Field(default_factory=lambda: BASE_DIR / "civilplan_mcp" / "data")
key_store_path: Path = Field(default_factory=default_key_store_path)
data_go_kr_api_key: str = Field(default_factory=lambda: os.getenv("DATA_GO_KR_API_KEY", ""))
vworld_api_key: str = Field(default_factory=lambda: os.getenv("VWORLD_API_KEY", ""))
@lru_cache(maxsize=1)
def get_settings() -> Settings:
load_local_env()
settings = Settings()
secure_keys = _load_secure_api_keys(settings.key_store_path)
if not settings.data_go_kr_api_key:
settings.data_go_kr_api_key = secure_keys.get("DATA_GO_KR_API_KEY", "")
if not settings.vworld_api_key:
settings.vworld_api_key = secure_keys.get("VWORLD_API_KEY", "")
settings.output_dir.mkdir(parents=True, exist_ok=True)
return settings