galaxis-po/backend/app/models/screening.py

51 lines
1.7 KiB
Python
Raw Normal View History

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")