gahusb
b690900cfc
config.py: kis_env_type (virtual/real) + KIS_REAL_*/KIS_VIRTUAL_* env variables (V1 호환). kis_app_key/kis_app_secret/kis_account properties auto-select based on env type. main.py: KIS not-configured warning uses descriptive message including env type + expected var name. test_main.py: monkeypatch load_dotenv to no-op + setenv empty string (instead of delenv) — defeats .env re-read on importlib.reload. Pre-existing test_startup_warns_if_webai_api_key_missing also fixed. 33/33 tests pass (was 31/33). Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
"""Signal V2 환경변수 로딩."""
|
|
import os
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv(Path(__file__).parent.parent / ".env")
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Settings:
|
|
stock_api_url: str = field(
|
|
default_factory=lambda: os.getenv("STOCK_API_URL", "").rstrip("/")
|
|
)
|
|
webai_api_key: str = field(
|
|
default_factory=lambda: os.getenv("WEBAI_API_KEY", "").strip()
|
|
)
|
|
port: int = field(default_factory=lambda: int(os.getenv("SIGNAL_V2_PORT", "8001")))
|
|
db_path: Path = field(
|
|
default_factory=lambda: Path(__file__).parent / "data" / "signal_v2.db"
|
|
)
|
|
# KIS — V1 호환 패턴 (KIS_ENV_TYPE virtual/real)
|
|
kis_env_type: str = field(default_factory=lambda: os.getenv("KIS_ENV_TYPE", "virtual").lower())
|
|
kis_real_app_key: str = field(default_factory=lambda: os.getenv("KIS_REAL_APP_KEY", "").strip())
|
|
kis_real_app_secret: str = field(default_factory=lambda: os.getenv("KIS_REAL_APP_SECRET", "").strip())
|
|
kis_real_account: str = field(default_factory=lambda: os.getenv("KIS_REAL_ACCOUNT", "").strip())
|
|
kis_virtual_app_key: str = field(default_factory=lambda: os.getenv("KIS_VIRTUAL_APP_KEY", "").strip())
|
|
kis_virtual_app_secret: str = field(default_factory=lambda: os.getenv("KIS_VIRTUAL_APP_SECRET", "").strip())
|
|
kis_virtual_account: str = field(default_factory=lambda: os.getenv("KIS_VIRTUAL_ACCOUNT", "").strip())
|
|
v1_token_path: Path = field(
|
|
default_factory=lambda: Path(
|
|
os.getenv("V1_TOKEN_PATH",
|
|
str(Path(__file__).parent.parent / "signal_v1" / "data" / "kis_token.json"))
|
|
)
|
|
)
|
|
|
|
@property
|
|
def kis_is_virtual(self) -> bool:
|
|
return self.kis_env_type != "real"
|
|
|
|
@property
|
|
def kis_app_key(self) -> str:
|
|
return self.kis_real_app_key if self.kis_env_type == "real" else self.kis_virtual_app_key
|
|
|
|
@property
|
|
def kis_app_secret(self) -> str:
|
|
return self.kis_real_app_secret if self.kis_env_type == "real" else self.kis_virtual_app_secret
|
|
|
|
@property
|
|
def kis_account(self) -> str:
|
|
return self.kis_real_account if self.kis_env_type == "real" else self.kis_virtual_account
|
|
|
|
|
|
def get_settings() -> Settings:
|
|
return Settings()
|