feat: add backtest SQLAlchemy models

- Backtest, BacktestResult, BacktestEquityCurve
- BacktestHolding, BacktestTransaction

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
zephyrdark 2026-02-03 09:36:37 +09:00
parent 331ab2cc56
commit d9d9c8d772
2 changed files with 125 additions and 0 deletions

View File

@ -22,6 +22,15 @@ from app.models.stock import (
AssetClass, AssetClass,
JobLog, JobLog,
) )
from app.models.backtest import (
Backtest,
BacktestStatus,
RebalancePeriod,
BacktestResult,
BacktestEquityCurve,
BacktestHolding,
BacktestTransaction,
)
__all__ = [ __all__ = [
"User", "User",
@ -44,4 +53,11 @@ __all__ = [
"ETFPrice", "ETFPrice",
"AssetClass", "AssetClass",
"JobLog", "JobLog",
"Backtest",
"BacktestStatus",
"RebalancePeriod",
"BacktestResult",
"BacktestEquityCurve",
"BacktestHolding",
"BacktestTransaction",
] ]

View File

@ -0,0 +1,109 @@
"""
Backtest related models.
"""
from datetime import datetime
from sqlalchemy import (
Column, Integer, String, Numeric, DateTime, Date,
Text, ForeignKey, JSON, Enum as SQLEnum
)
from sqlalchemy.orm import relationship
import enum
from app.core.database import Base
class BacktestStatus(str, enum.Enum):
PENDING = "pending"
RUNNING = "running"
COMPLETED = "completed"
FAILED = "failed"
class RebalancePeriod(str, enum.Enum):
MONTHLY = "monthly"
QUARTERLY = "quarterly"
SEMI_ANNUAL = "semi_annual"
ANNUAL = "annual"
class Backtest(Base):
__tablename__ = "backtests"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
strategy_type = Column(String(50), nullable=False)
strategy_params = Column(JSON, default={})
start_date = Column(Date, nullable=False)
end_date = Column(Date, nullable=False)
rebalance_period = Column(SQLEnum(RebalancePeriod), default=RebalancePeriod.QUARTERLY)
initial_capital = Column(Numeric(20, 2), nullable=False)
commission_rate = Column(Numeric(10, 6), default=0.00015)
slippage_rate = Column(Numeric(10, 6), default=0.001)
benchmark = Column(String(20), default="KOSPI")
top_n = Column(Integer, default=30)
status = Column(SQLEnum(BacktestStatus), default=BacktestStatus.PENDING)
created_at = Column(DateTime, default=datetime.utcnow)
completed_at = Column(DateTime, nullable=True)
error_message = Column(Text, nullable=True)
# Relationships
result = relationship("BacktestResult", back_populates="backtest", uselist=False)
equity_curve = relationship("BacktestEquityCurve", back_populates="backtest")
holdings = relationship("BacktestHolding", back_populates="backtest")
transactions = relationship("BacktestTransaction", back_populates="backtest")
class BacktestResult(Base):
__tablename__ = "backtest_results"
backtest_id = Column(Integer, ForeignKey("backtests.id"), primary_key=True)
total_return = Column(Numeric(10, 4))
cagr = Column(Numeric(10, 4))
mdd = Column(Numeric(10, 4))
sharpe_ratio = Column(Numeric(10, 4))
volatility = Column(Numeric(10, 4))
benchmark_return = Column(Numeric(10, 4))
excess_return = Column(Numeric(10, 4))
backtest = relationship("Backtest", back_populates="result")
class BacktestEquityCurve(Base):
__tablename__ = "backtest_equity_curve"
backtest_id = Column(Integer, ForeignKey("backtests.id"), primary_key=True)
date = Column(Date, primary_key=True)
portfolio_value = Column(Numeric(20, 2))
benchmark_value = Column(Numeric(20, 2))
drawdown = Column(Numeric(10, 4))
backtest = relationship("Backtest", back_populates="equity_curve")
class BacktestHolding(Base):
__tablename__ = "backtest_holdings"
backtest_id = Column(Integer, ForeignKey("backtests.id"), primary_key=True)
rebalance_date = Column(Date, primary_key=True)
ticker = Column(String(20), primary_key=True)
name = Column(String(100))
weight = Column(Numeric(10, 4))
shares = Column(Integer)
price = Column(Numeric(12, 2))
backtest = relationship("Backtest", back_populates="holdings")
class BacktestTransaction(Base):
__tablename__ = "backtest_transactions"
id = Column(Integer, primary_key=True, index=True)
backtest_id = Column(Integer, ForeignKey("backtests.id"), nullable=False)
date = Column(Date, nullable=False)
ticker = Column(String(20), nullable=False)
action = Column(String(10), nullable=False) # buy/sell
shares = Column(Integer, nullable=False)
price = Column(Numeric(12, 2), nullable=False)
commission = Column(Numeric(12, 2), nullable=False)
backtest = relationship("Backtest", back_populates="transactions")