""" 설정값 영속 저장. SQLite key-value 테이블 한 개로 단순화. - 텔레그램 토큰 / chat_id - 모니터링 심볼 - 알림 시간축 목록 - 쿨다운 / STOP_LOSS_PCT 등 운영 파라미터 Streamlit rerun 안전: 모듈 최상단의 connection 캐시는 모듈 캐싱 (sys.modules) 으로 process lifetime 동안 보존. 멀티 스레드 (alert thread) 도 같은 DB 파일을 읽으므로 SQLite 의 thread-safe 모드 (`check_same_thread=False`) 로 연다. """ import os import sqlite3 import threading from typing import Any, Optional DB_PATH = os.environ.get("SETTINGS_DB_PATH", "/app/data/settings.db") _lock = threading.RLock() _conn: Optional[sqlite3.Connection] = None def _get_conn() -> sqlite3.Connection: global _conn if _conn is None: os.makedirs(os.path.dirname(DB_PATH), exist_ok=True) _conn = sqlite3.connect(DB_PATH, check_same_thread=False, isolation_level=None) _conn.execute("PRAGMA journal_mode=WAL") _conn.execute( """ CREATE TABLE IF NOT EXISTS settings ( key TEXT PRIMARY KEY, value TEXT NOT NULL, updated_at TEXT NOT NULL DEFAULT (datetime('now')) ) """ ) return _conn DEFAULTS = { # ── 알림 / 모니터링 ── "telegram_token": "", "telegram_chat_id": "", "alert_symbol": "BTCUSDT", "alert_timeframes": "5m,15m,30m,1h", "alert_cooldown_sec": "600", "stop_loss_pct": "0.0075", "alert_enabled": "1", "daily_report_enabled": "1", "polling_interval_sec": "30", # ── 신호 임계값 (RSI / body) ── "long_rsi_max": "75", "short_rsi_min": "25", "strong_long_rsi_max": "65", "strong_short_rsi_min": "35", "body_pct_min": "0.002", # 일반 신호 캔들 body 최소 (양/음봉) "reversal_body_pct": "0.003", # 추세 꺾임 body 최소 "reversal_vol_mult": "1.3", # 추세 꺾임 거래량 동반 배수 # ── 거래량 / 펀딩비 ── "vol_exhaustion_mult": "3.0", # exhaustion 판정 (vol > avg × N) "vol_net_mult": "2.0", # vol_long/short_signal: net > avg × N "oi_active_pct": "0.001", # OI 활성도 임계 (변동률 절대값) "fr_long_overheat": "0.005", # 롱 과열 (배너) "fr_short_caution": "-0.005", # 숏스퀴즈 경보 "fr_short_extreme": "-0.007", # 숏 주의 신호 임계 # ── 차트 / UI ── "candle_limit_desktop": "53", "candle_limit_mobile": "14", "forming_stable_polls": "2", # forming candle 안정성 — 연속 N polls True 요구 } def init_db_with_env_defaults(): """최초 기동 시 .env 값을 DB 기본값으로 복사. 이미 존재하는 키는 건드리지 않음.""" with _lock: conn = _get_conn() for k, default in DEFAULTS.items(): cur = conn.execute("SELECT value FROM settings WHERE key=?", (k,)) if cur.fetchone() is not None: continue seed = default env_map = { "telegram_token": "TELEGRAM_TOKEN", "telegram_chat_id": "TELEGRAM_CHAT_ID", } if k in env_map: seed = os.environ.get(env_map[k], default) or default conn.execute( "INSERT INTO settings(key, value) VALUES (?, ?)", (k, seed), ) def get(key: str, default: Any = None) -> str: with _lock: conn = _get_conn() cur = conn.execute("SELECT value FROM settings WHERE key=?", (key,)) row = cur.fetchone() if row is None: return DEFAULTS.get(key, default) if default is None else default return row[0] def get_int(key: str, default: int = 0) -> int: try: return int(get(key, default)) except (TypeError, ValueError): return default def get_float(key: str, default: float = 0.0) -> float: try: return float(get(key, default)) except (TypeError, ValueError): return default def get_bool(key: str, default: bool = False) -> bool: v = get(key, "1" if default else "0") return str(v).strip().lower() in ("1", "true", "yes", "on") def get_list(key: str, default=None, sep: str = ",") -> list: v = get(key, "") if not v: return list(default or []) return [s.strip() for s in v.split(sep) if s.strip()] def set_value(key: str, value: Any): with _lock: conn = _get_conn() conn.execute( """ INSERT INTO settings(key, value, updated_at) VALUES(?, ?, datetime('now')) ON CONFLICT(key) DO UPDATE SET value=excluded.value, updated_at=excluded.updated_at """, (key, str(value)), ) def all_settings() -> dict: with _lock: conn = _get_conn() cur = conn.execute("SELECT key, value FROM settings ORDER BY key") return dict(cur.fetchall())