feat: add backtest module with DB cache and scenario comparison
Backtest improvements: - Add backtest.py with Oracle DB-backed OHLCV cache (no repeated API calls) - Add backtest_trades table to cache simulation results by params hash (same params -> instant load, skip re-simulation) - Add walk-forward scenario comparison (--walkforward-cmp) - Add trend ceiling filter (--trend-cmp, max gain threshold) - Add ticker win-rate filter (--ticker-cmp, SQL-based instant analysis) - Precompute daily_features once per data load (not per scenario) Live bot fixes: - monitor: add hard stop-loss from buy price (in addition to trailing) - strategy: fix re-entry condition to require +1% above last sell price - price_collector: add 48h backfill on startup for trend calculation - main: call backfill_prices() at startup Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
1556
backtest.py
Normal file
1556
backtest.py
Normal file
File diff suppressed because it is too large
Load Diff
@@ -19,7 +19,7 @@ TIME_STOP_MIN_GAIN_PCT = float(os.getenv("TIME_STOP_MIN_GAIN_PCT", "3"))
|
|||||||
|
|
||||||
|
|
||||||
def _check_trailing_stop(ticker: str, pos: dict, current: float) -> bool:
|
def _check_trailing_stop(ticker: str, pos: dict, current: float) -> bool:
|
||||||
"""트레일링 스탑 체크. 매도 시 True 반환."""
|
"""트레일링 스탑(최고가 기준) + 고정 스탑(매수가 기준) 체크. 매도 시 True 반환."""
|
||||||
trader.update_peak(ticker, current)
|
trader.update_peak(ticker, current)
|
||||||
|
|
||||||
pos = trader.get_positions().get(ticker)
|
pos = trader.get_positions().get(ticker)
|
||||||
@@ -27,15 +27,24 @@ def _check_trailing_stop(ticker: str, pos: dict, current: float) -> bool:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
peak = pos["peak_price"]
|
peak = pos["peak_price"]
|
||||||
|
buy_price = pos["buy_price"]
|
||||||
drop_from_peak = (peak - current) / peak
|
drop_from_peak = (peak - current) / peak
|
||||||
|
drop_from_buy = (buy_price - current) / buy_price # 구매가 대비 하락률
|
||||||
|
|
||||||
if drop_from_peak >= STOP_LOSS_PCT:
|
if drop_from_peak >= STOP_LOSS_PCT:
|
||||||
reason = (
|
reason = (
|
||||||
f"트레일링스탑 | 최고가={peak:,.0f}원 → "
|
f"트레일링스탑 | 최고가={peak:,.0f}원 → "
|
||||||
f"현재={current:,.0f}원 ({drop_from_peak:.1%} 하락)"
|
f"현재={current:,.0f}원 ({drop_from_peak:.1%} 하락)"
|
||||||
)
|
)
|
||||||
trader.sell(ticker, reason=reason)
|
return trader.sell(ticker, reason=reason)
|
||||||
return True
|
|
||||||
|
if drop_from_buy >= STOP_LOSS_PCT:
|
||||||
|
reason = (
|
||||||
|
f"스탑로스 | 매수가={buy_price:,.0f}원 → "
|
||||||
|
f"현재={current:,.0f}원 ({drop_from_buy:.1%} 하락)"
|
||||||
|
)
|
||||||
|
return trader.sell(ticker, reason=reason)
|
||||||
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
@@ -70,15 +79,17 @@ def _check_position(ticker: str, pos: dict) -> None:
|
|||||||
if current is None:
|
if current is None:
|
||||||
return
|
return
|
||||||
|
|
||||||
pnl = (current - pos["buy_price"]) / pos["buy_price"] * 100
|
buy_price = pos["buy_price"]
|
||||||
|
pnl = (current - buy_price) / buy_price * 100
|
||||||
peak = pos["peak_price"]
|
peak = pos["peak_price"]
|
||||||
drop_from_peak = (peak - current) / peak
|
drop_from_peak = (peak - current) / peak
|
||||||
|
drop_from_buy = (buy_price - current) / buy_price
|
||||||
entry_time = pos.get("entry_time", datetime.now())
|
entry_time = pos.get("entry_time", datetime.now())
|
||||||
elapsed_hours = (datetime.now() - entry_time).total_seconds() / 3600
|
elapsed_hours = (datetime.now() - entry_time).total_seconds() / 3600
|
||||||
|
|
||||||
logger.info(
|
logger.info(
|
||||||
f"[감시] {ticker} 현재={current:,.0f} | 최고={peak:,.0f} | "
|
f"[감시] {ticker} 현재={current:,.0f} | 매수가={buy_price:,.0f} | 최고={peak:,.0f} | "
|
||||||
f"하락={drop_from_peak:.1%} | 수익률={pnl:+.1f}% | "
|
f"수익률={pnl:+.1f}% | peak하락={drop_from_peak:.1%} | buy하락={drop_from_buy:.1%} | "
|
||||||
f"보유={elapsed_hours:.1f}h"
|
f"보유={elapsed_hours:.1f}h"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -94,7 +105,7 @@ def run_monitor(interval: int = CHECK_INTERVAL) -> None:
|
|||||||
"""전체 포지션 감시 루프."""
|
"""전체 포지션 감시 루프."""
|
||||||
logger.info(
|
logger.info(
|
||||||
f"모니터 시작 | 체크={interval}초 | "
|
f"모니터 시작 | 체크={interval}초 | "
|
||||||
f"트레일링스탑={STOP_LOSS_PCT:.0%} | "
|
f"트레일링스탑={STOP_LOSS_PCT:.1%} | "
|
||||||
f"타임스탑={TIME_STOP_HOURS:.0f}h/{TIME_STOP_MIN_GAIN_PCT:+.0f}%"
|
f"타임스탑={TIME_STOP_HOURS:.0f}h/{TIME_STOP_MIN_GAIN_PCT:+.0f}%"
|
||||||
)
|
)
|
||||||
while True:
|
while True:
|
||||||
|
|||||||
@@ -5,10 +5,11 @@ from __future__ import annotations
|
|||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
import pyupbit
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
from .market import get_top_tickers
|
from .market import get_top_tickers
|
||||||
from .price_db import cleanup_old_prices, insert_prices
|
from .price_db import cleanup_old_prices, insert_prices, insert_prices_with_time
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -16,6 +17,38 @@ COLLECT_INTERVAL = 600 # 10분 (초)
|
|||||||
CLEANUP_EVERY = 6 # 1시간(10분 × 6)마다 오래된 데이터 정리
|
CLEANUP_EVERY = 6 # 1시간(10분 × 6)마다 오래된 데이터 정리
|
||||||
|
|
||||||
|
|
||||||
|
def backfill_prices(hours: int = 48) -> None:
|
||||||
|
"""시작 시 과거 N시간치 1시간봉 종가를 DB에 백필.
|
||||||
|
|
||||||
|
price_history에 데이터가 없으면 추세 판단이 불가능하므로
|
||||||
|
봇 시작 직후 한 번 호출해 과거 데이터를 채운다.
|
||||||
|
"""
|
||||||
|
tickers = get_top_tickers()
|
||||||
|
if not tickers:
|
||||||
|
logger.warning("[백필] 종목 목록 없음, 스킵")
|
||||||
|
return
|
||||||
|
|
||||||
|
count = hours + 2 # 여유 있게 요청
|
||||||
|
total_rows = 0
|
||||||
|
|
||||||
|
for ticker in tickers:
|
||||||
|
try:
|
||||||
|
df = pyupbit.get_ohlcv(ticker, interval="minute60", count=count)
|
||||||
|
if df is None or df.empty:
|
||||||
|
continue
|
||||||
|
rows = [
|
||||||
|
(ticker, float(row["close"]), ts.to_pydatetime())
|
||||||
|
for ts, row in df.iterrows()
|
||||||
|
]
|
||||||
|
insert_prices_with_time(rows)
|
||||||
|
total_rows += len(rows)
|
||||||
|
time.sleep(0.1)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[백필] {ticker} 오류: {e}")
|
||||||
|
|
||||||
|
logger.info(f"[백필] 완료 — {len(tickers)}개 종목 / {total_rows}개 레코드 저장")
|
||||||
|
|
||||||
|
|
||||||
def run_collector(interval: int = COLLECT_INTERVAL) -> None:
|
def run_collector(interval: int = COLLECT_INTERVAL) -> None:
|
||||||
"""가격 수집 루프."""
|
"""가격 수집 루프."""
|
||||||
logger.info(f"가격 수집기 시작 (주기={interval//60}분)")
|
logger.info(f"가격 수집기 시작 (주기={interval//60}분)")
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
"""Strategy C: 실시간 상승 추세(DB) AND 거래량 모멘텀 동시 충족 시 매수 신호."""
|
"""Strategy C: 현재 기준 N시간 전 대비 상승 추세(DB) AND 거래량 모멘텀 동시 충족 시 매수 신호."""
|
||||||
|
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
@@ -10,8 +10,8 @@ from .price_db import get_price_n_hours_ago
|
|||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
# 추세 판단: N시간 전 대비 +M% 이상이면 상승 중
|
# 추세 판단: 현재 기준 N시간 전 DB 가격 대비 +M% 이상이면 상승 중
|
||||||
TREND_HOURS = float(os.getenv("TREND_HOURS", "1"))
|
TREND_HOURS = float(os.getenv("TREND_HOURS", "12"))
|
||||||
TREND_MIN_GAIN_PCT = float(os.getenv("TREND_MIN_GAIN_PCT", "3"))
|
TREND_MIN_GAIN_PCT = float(os.getenv("TREND_MIN_GAIN_PCT", "3"))
|
||||||
|
|
||||||
# 모멘텀: MA 기간, 거래량 급증 배수
|
# 모멘텀: MA 기간, 거래량 급증 배수
|
||||||
@@ -20,10 +20,10 @@ VOLUME_MULTIPLIER = 2.0
|
|||||||
|
|
||||||
|
|
||||||
def check_trend(ticker: str) -> bool:
|
def check_trend(ticker: str) -> bool:
|
||||||
"""상승 추세 조건: 현재가가 N시간 전 대비 +M% 이상."""
|
"""상승 추세 조건: 현재가가 DB에 저장된 N시간 전 가격 대비 +M% 이상."""
|
||||||
past_price = get_price_n_hours_ago(ticker, TREND_HOURS)
|
past_price = get_price_n_hours_ago(ticker, TREND_HOURS)
|
||||||
if past_price is None:
|
if past_price is None:
|
||||||
logger.debug(f"[추세] {ticker} 과거 가격 없음 (데이터 수집 중)")
|
logger.debug(f"[추세] {ticker} {TREND_HOURS:.0f}h 전 가격 없음 (수집 중)")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
current = get_current_price(ticker)
|
current = get_current_price(ticker)
|
||||||
@@ -35,8 +35,8 @@ def check_trend(ticker: str) -> bool:
|
|||||||
|
|
||||||
if result:
|
if result:
|
||||||
logger.info(
|
logger.info(
|
||||||
f"[추세↑] {ticker} {TREND_HOURS:.0f}h 전={past_price:,.0f} "
|
f"[추세↑] {ticker} {TREND_HOURS:.0f}h 전={past_price:,.2f} "
|
||||||
f"현재={current:,.0f} (+{gain_pct:.1f}%)"
|
f"현재={current:,.2f} (+{gain_pct:.1f}%)"
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
logger.debug(
|
logger.debug(
|
||||||
|
|||||||
8
main.py
8
main.py
@@ -19,7 +19,7 @@ logging.basicConfig(
|
|||||||
|
|
||||||
from core.monitor import run_monitor
|
from core.monitor import run_monitor
|
||||||
from core.notify import notify_error, notify_status
|
from core.notify import notify_error, notify_status
|
||||||
from core.price_collector import run_collector
|
from core.price_collector import backfill_prices, run_collector
|
||||||
from core.trader import get_positions, restore_positions
|
from core.trader import get_positions, restore_positions
|
||||||
from daemon.runner import run_scanner
|
from daemon.runner import run_scanner
|
||||||
|
|
||||||
@@ -45,6 +45,10 @@ def main() -> None:
|
|||||||
# 재시작 시 기존 잔고 복원 (이중 매수 방지)
|
# 재시작 시 기존 잔고 복원 (이중 매수 방지)
|
||||||
restore_positions()
|
restore_positions()
|
||||||
|
|
||||||
|
# 과거 가격 백필 (추세 판단용 DB 데이터가 없는 경우 채움)
|
||||||
|
logger.info("과거 가격 백필 시작 (48시간)...")
|
||||||
|
backfill_prices(hours=48)
|
||||||
|
|
||||||
# 트레일링 스탑 감시 스레드 (10초 주기)
|
# 트레일링 스탑 감시 스레드 (10초 주기)
|
||||||
monitor_thread = threading.Thread(
|
monitor_thread = threading.Thread(
|
||||||
target=run_monitor, args=(10,), daemon=True, name="monitor"
|
target=run_monitor, args=(10,), daemon=True, name="monitor"
|
||||||
@@ -57,7 +61,7 @@ def main() -> None:
|
|||||||
)
|
)
|
||||||
status_thread.start()
|
status_thread.start()
|
||||||
|
|
||||||
# 10분 주기 가격 수집 스레드 (추세 판단용 DB 저장)
|
# 가격 수집 스레드 (10분 주기 → Oracle DB price_history 저장, 추세 판단용)
|
||||||
collector_thread = threading.Thread(
|
collector_thread = threading.Thread(
|
||||||
target=run_collector, daemon=True, name="collector"
|
target=run_collector, daemon=True, name="collector"
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user