Initial commit

backend/app/models/models.py

@@ -0,0 +1,141 @@
+from datetime import datetime
+from typing import Optional, List
+from sqlalchemy import Column, Integer, String, Numeric, Boolean, DateTime, Text, ARRAY
+from sqlalchemy.dialects.postgresql import JSONB
+from app.core.database import Base
+
+
+class TradingRule(Base):
+    __tablename__ = "trading_rules"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String(255), nullable=False)
+    description = Column(Text)
+    rule_type = Column(String(50), nullable=False)
+    conditions = Column(JSONB, nullable=False)
+    actions = Column(JSONB, nullable=False)
+    is_active = Column(Boolean, default=True)
+    priority = Column(Integer, default=0)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+
+class TradeHistory(Base):
+    __tablename__ = "trade_history"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    rule_id = Column(Integer)
+    stock_code = Column(String(20), nullable=False, index=True)
+    stock_name = Column(String(100))
+    order_type = Column(String(20), nullable=False)
+    order_price = Column(Numeric(15, 2))
+    order_quantity = Column(Integer, nullable=False)
+    executed_price = Column(Numeric(15, 2))
+    executed_quantity = Column(Integer)
+    status = Column(String(50), nullable=False)
+    order_id = Column(String(100))
+    message = Column(Text)
+    created_at = Column(DateTime, default=datetime.utcnow, index=True)
+    executed_at = Column(DateTime)
+
+
+class Portfolio(Base):
+    __tablename__ = "portfolio"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    stock_code = Column(String(20), nullable=False, unique=True)
+    stock_name = Column(String(100))
+    quantity = Column(Integer, nullable=False, default=0)
+    average_price = Column(Numeric(15, 2))
+    current_price = Column(Numeric(15, 2))
+    profit_loss = Column(Numeric(15, 2))
+    profit_loss_rate = Column(Numeric(10, 4))
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+
+class MarketData(Base):
+    __tablename__ = "market_data"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    stock_code = Column(String(20), nullable=False, index=True)
+    price = Column(Numeric(15, 2), nullable=False)
+    change_rate = Column(Numeric(10, 4))
+    volume = Column(Integer)
+    high_price = Column(Numeric(15, 2))
+    low_price = Column(Numeric(15, 2))
+    open_price = Column(Numeric(15, 2))
+    timestamp = Column(DateTime, default=datetime.utcnow, index=True)
+
+
+class News(Base):
+    __tablename__ = "news"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String(500), nullable=False)
+    content = Column(Text)
+    source = Column(String(100))
+    url = Column(Text)
+    stock_codes = Column(ARRAY(String))
+    sentiment = Column(String(20))
+    published_at = Column(DateTime, index=True)
+    created_at = Column(DateTime, default=datetime.utcnow)
+
+
+class ClaudeAnalysisLog(Base):
+    __tablename__ = "claude_analysis_log"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    rule_id = Column(Integer)
+    stock_code = Column(String(20))
+    analysis_type = Column(String(50))
+    input_data = Column(JSONB)
+    analysis_result = Column(JSONB)
+    decision = Column(String(50))
+    confidence = Column(Numeric(5, 4))
+    created_at = Column(DateTime, default=datetime.utcnow, index=True)
+
+
+class MCPToolCall(Base):
+    __tablename__ = "mcp_tool_calls"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    tool_name = Column(String(100), nullable=False)
+    parameters = Column(JSONB)
+    result = Column(JSONB)
+    status = Column(String(50))
+    execution_time_ms = Column(Integer)
+    created_at = Column(DateTime, default=datetime.utcnow, index=True)
+
+
+class StockAnalysis(Base):
+    """매일 배치로 저장되는 종목 분석 데이터"""
+    __tablename__ = "stock_analysis"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    stock_code = Column(String(20), nullable=False, index=True)
+    stock_name = Column(String(100), nullable=False)
+    analysis_date = Column(DateTime, nullable=False, index=True)
+    
+    # 가격 정보
+    current_price = Column(Integer, nullable=False)
+    change_rate = Column(Numeric(10, 4))
+    volume = Column(Integer)
+    
+    # 기술적 지표
+    rsi = Column(Numeric(10, 4))
+    ma5 = Column(Numeric(15, 2))
+    ma20 = Column(Numeric(15, 2))
+    ma60 = Column(Numeric(15, 2))
+    stochastic_k = Column(Numeric(10, 4))
+    stochastic_d = Column(Numeric(10, 4))
+    bollinger_upper = Column(Numeric(15, 2))
+    bollinger_lower = Column(Numeric(15, 2))
+    bollinger_middle = Column(Numeric(15, 2))
+    
+    # 분석 결과
+    signals = Column(JSONB)  # ["RSI 과매도", "골든크로스" 등]
+    score = Column(Integer, default=0)
+    
+    # 메타데이터
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)