from datetime import datetime

from sqlalchemy import (
    Column, Integer, String, BigInteger, Boolean, Numeric, DateTime, Date,
    ForeignKey,
)
from sqlalchemy.orm import relationship

from app.core.database import Base


class ScreeningSignal(Base):
    __tablename__ = "screening_signals"

    id = Column(Integer, primary_key=True, index=True)
    screen_date = Column(Date, nullable=False, index=True)
    ticker = Column(String(20), nullable=False, index=True)
    name = Column(String(100))
    sector = Column(String(50))
    market_cap = Column(BigInteger)
    trading_value = Column(BigInteger)
    is_limit_up = Column(Boolean, default=False)
    daily_return = Column(Numeric(8, 4))
    trigger_low = Column(Numeric(12, 2))
    market_state = Column(String(20))
    status = Column(String(20), default="pending", index=True)
    entry_date = Column(Date)
    entry_price = Column(Numeric(12, 2))
    exit_date = Column(Date)
    exit_price = Column(Numeric(12, 2))
    created_at = Column(DateTime, default=datetime.utcnow)

    orders = relationship("AutoOrder", back_populates="screening_signal")


class AutoOrder(Base):
    __tablename__ = "auto_orders"

    id = Column(Integer, primary_key=True, index=True)
    order_date = Column(DateTime, nullable=False, index=True)
    ticker = Column(String(20), nullable=False, index=True)
    order_type = Column(String(10))
    qty = Column(Integer)
    price = Column(Numeric(12, 2))
    order_no = Column(String(50))
    status = Column(String(20), default="pending")
    screening_signal_id = Column(Integer, ForeignKey("screening_signals.id"), nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)

    screening_signal = relationship("ScreeningSignal", back_populates="orders")