from app.main.data_management.data_base import *


class FEPortfolio(Schema.base):
    __tablename__ = 'fe_portfolio'

    id = Column(INTEGER(11), primary_key=True, nullable=False)
    start_date = Column(DateTime)
    end_date = Column(DateTime)
    user_id = Column(ForeignKey('user.id'), nullable=False, index=True)
    game_session_id = Column(ForeignKey('game_session.id'), nullable=False, index=True)
    game_session = relationship('GameSession')

    user = relationship('User')
    orders = relationship("FEOrder", back_populates="portfolio")
    positions = relationship("FEPosition", back_populates="portfolio")
    portfolio_indicators = relationship("FEPortfolioIndicator", back_populates="portfolio")

    def __init__(self, user_id, game_session_id, start_date, end_date):
        self.user_id = user_id
        self.game_session_id = game_session_id
        self.start_date = start_date
        self.end_date = end_date